|
|
|
|
import os
|
|
|
|
|
|
|
|
|
|
import paho.mqtt.client as mqtt
|
|
|
|
|
import time
|
|
|
|
|
|
|
|
|
|
broker = '121.40.129.71'
|
|
|
|
|
port = 48835
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
with open("edu_coder/user.txt", "r") as f:
|
|
|
|
|
user_id = f.read().strip()
|
|
|
|
|
user_id = user_id.split("=")[-1]
|
|
|
|
|
|
|
|
|
|
# topic = "request/{}".format(user_id)
|
|
|
|
|
|
|
|
|
|
# 回调函数，当客户端收到连接响应时被调用
|
|
|
|
|
def on_connect(client, userdata, flags, rc, properties=None):
|
|
|
|
|
print("连接结果: " + str(rc))
|
|
|
|
|
# 订阅请求主题
|
|
|
|
|
client.subscribe("response/{}".format(user_id), qos=2)
|
|
|
|
|
|
|
|
|
|
client.subscribe("image/{}".format(user_id), qos=2)
|
|
|
|
|
|
|
|
|
|
# 回调函数，当客户端收到消息时被调用
|
|
|
|
|
def on_message(client, userdata, msg):
|
|
|
|
|
# print("收到消息: " + msg.topic + " " + str(msg.payload))
|
|
|
|
|
|
|
|
|
|
if msg.topic == "response/{}".format(user_id):
|
|
|
|
|
print("收到响应: "+str(msg.topic) + str(msg.payload))
|
|
|
|
|
|
|
|
|
|
if msg.topic == "image/{}".format(user_id):
|
|
|
|
|
with open("image.jpg", "wb") as f:
|
|
|
|
|
f.write(msg.payload)
|
|
|
|
|
print("图片已保存。")
|
|
|
|
|
|
|
|
|
|
global response_received
|
|
|
|
|
response_received = True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
client = mqtt.Client()
|
|
|
|
|
except:
|
|
|
|
|
client = mqtt.Client(mqtt.CallbackAPIVersion.VERSION2,)
|
|
|
|
|
|
|
|
|
|
# 设置回调函数
|
|
|
|
|
client.on_connect = on_connect
|
|
|
|
|
client.on_message = on_message
|
|
|
|
|
|
|
|
|
|
# 连接到MQTT代理
|
|
|
|
|
client.connect(broker, port, 60)
|
|
|
|
|
|
|
|
|
|
# 启动一个后台线程来处理网络流量和回调
|
|
|
|
|
client.loop_start()
|
|
|
|
|
|
|
|
|
|
# 发布消息
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 等待响应
|
|
|
|
|
response_received = False
|
|
|
|
|
while not response_received:
|
|
|
|
|
print("等待响应...")
|
|
|
|
|
time.sleep(1)
|
|
|
|
|
client.publish("request/{}".format(user_id), "image produce", qos=2)
|
|
|
|
|
|
|
|
|
|
# 停止网络循环并断开连接
|
|
|
|
|
client.loop_stop()
|
|
|
|
|
client.disconnect()
|
|
|
|
|
|
|
|
|
|
print("收到响应，客户端退出。")
|
