/
testmqtt.py
45 lines (35 loc) · 1.33 KB
/
testmqtt.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
from libs.SysUtil import SysUtil
import paho.mqtt.client as client
import time
from zlib import crc32
def on_message(client, userdata, msg):
payload = msg.payload.decode("utf-8").strip()
print("topic: {} payload: {}".format(msg.topic, payload))
def on_connect(client, *args):
print(args)
print("Subscribing to rpi/{}/operation".format(SysUtil.get_machineid()))
client.subscribe("rpi/{}/operation".format(SysUtil.get_machineid()), qos=1)
iden = bytes(SysUtil.get_hostname() + "-Updater", "utf-8")
def setupmqtt():
mqtt = client.Client(client_id=str(crc32(iden)),
clean_session=True,
protocol=client.MQTTv311,
transport="tcp")
mqtt.on_connect = on_connect
mqtt.on_message = on_message
try:
with open("mqttpassword") as f:
mqtt.username_pw_set(username=SysUtil.get_hostname(), password=f.read().strip())
except:
print("invalid password")
mqtt.username_pw_set(username=SysUtil.get_hostname(), password="INVALIDPASSWORD")
mqtt.connect("10.8.0.1", port=1883)
mqtt.loop_forever()
# mqtt.subscribe("rpi/{}/operation".format(SysUtil.get_machineid()), qos=1)
def main():
setupmqtt()
# while True:
# time.sleep(10)
# print("Sleeping")
if __name__ == '__main__':
main()