lot_manager/test.py

69 lines
1.7 KiB
Python
Raw Normal View History

2023-12-21 09:13:42 +08:00
import json
import time
import uvicorn
from fastapi import FastAPI
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:
def __init__(self, broker, port, topic, username, password):
self.broker = broker
self.port = port
self.topic = topic
self.username = username
self.password = password
# 千万不要指定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),
qos=0)
def close(self):
self.client.publish("lihai_lot_walnutpi_dev_4", 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():
try:
MQTT.start()
MQTT.push()
except Exception as e:
2023-12-21 09:14:35 +08:00
# print(e)
2023-12-21 09:13:42 +08:00
pass
@app.get("/close_stream")
def close_stream():
try:
MQTT.start()
MQTT.close()
except Exception as e:
2023-12-21 09:14:35 +08:00
# print(e)
2023-12-21 09:13:42 +08:00
pass
if __name__ == '__main__':
MQTT = MQTTClient(broker, port, subscribe_topic, username, password)
uvicorn.run(app, host="127.0.0.1", port=8000)