import os

import paho.mqtt.client as mqtt
import time

broker = '121.40.129.71'
port = 48835



with open("../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.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


# 创建客户端实例
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("收到响应，客户端退出。")