From a0e4dcac5b70c1caf5acdea3a669286c6e2d3217 Mon Sep 17 00:00:00 2001
From: xyj <10908227994@qq.com>
Date: Thu, 28 Dec 2023 09:37:37 +0800
Subject: [PATCH] update

---
 test.py | 29 +++++++++++------------------
 1 file changed, 11 insertions(+), 18 deletions(-)

diff --git a/test.py b/test.py
index 447cec0..4868eb2 100644
--- a/test.py
+++ b/test.py
@@ -7,11 +7,7 @@ import paho.mqtt.client as mqtt
 
 app = FastAPI()
 
-broker = 'mqtt.lihaink.cn'
-port = 1883
-subscribe_topic = 'info_dev_4'
-username = 'lihai_lot_land_1'
-password = 'lihai_lot_land_1'
+
 
 
 class MQTTClient:
@@ -24,26 +20,22 @@ class MQTTClient:
         # 千万不要指定client_id 不然死翘翘
         self.client = mqtt.Client()
         self.client.username_pw_set(self.username, self.password)
-        self.client.on_connect = self.on_connect
-
-    def on_connect(self, client, userdata, flags, rc):
-        if rc == 0:
-            self.client.subscribe(self.topic)
 
     def push(self):
-        self.client.publish("lihai_lot_walnutpi_dev_4", payload=json.dumps({"msg": "push_stream"}, ensure_ascii=False),
+        self.client.publish(self.topic, payload=json.dumps({"msg": "push_stream"}, ensure_ascii=False),
                             qos=0)
 
     def close(self):
-        self.client.publish("lihai_lot_walnutpi_dev_4", payload=json.dumps({"msg": "close_stream"}, ensure_ascii=False),
+        self.client.publish(self.topic, payload=json.dumps({"msg": "close_stream"}, ensure_ascii=False),
                             qos=0)
 
     def start(self):
         self.client.connect(self.broker, self.port)
 
 
-@app.get("/push_stream")
-def push_stream():
+@app.post("/push_stream")
+def push_stream(username, device):
+    MQTT = MQTTClient(broker, port, device, username, username)
     try:
         MQTT.start()
         MQTT.push()
@@ -52,9 +44,10 @@ def push_stream():
         pass
 
 
-@app.get("/close_stream")
-def close_stream():
+@app.post("/close_stream")
+def close_stream(username, device):
     try:
+        MQTT = MQTTClient(broker, port, device, username, username)
         MQTT.start()
         MQTT.close()
     except Exception as e:
@@ -63,6 +56,6 @@ def close_stream():
 
 
 if __name__ == '__main__':
-    MQTT = MQTTClient(broker, port, subscribe_topic, username, password)
-
+    broker = 'mqtt.lihaink.cn'
+    port = 1883
     uvicorn.run(app, host="127.0.0.1", port=8000)