|
|
#!/usr/bin/env python
|
|
|
|
|
|
import os, sys
|
|
|
import signal
|
|
|
import time
|
|
|
import json
|
|
|
import paho.mqtt.client as paho
|
|
|
from watchdog.events import FileSystemEventHandler
|
|
|
from watchdog.observers.polling import PollingObserver
|
|
|
import platform
|
|
|
|
|
|
MQTTHOST = os.getenv('BROKER_URL', '192.168.1.130')
|
|
|
PROCSITE = os.getenv('PROC_SITE', 'sophy-proc')
|
|
|
MQTTPORT = int(os.getenv('MQTTPORT', 1883))
|
|
|
MQTTUSERNAME = os.getenv('MQTTUSERNAME', None)
|
|
|
MQTTPASSWORD = os.getenv('MQTTPASSWORD', None)
|
|
|
MQTTWATCHDIR = '/data'
|
|
|
MQTTQOS = int(os.getenv('MQTTQOS', 0))
|
|
|
MQTTRETAIN = int(os.getenv('MQTTRETAIN', 0))
|
|
|
|
|
|
MQTTFIXEDTOPIC = os.getenv('MQTTFIXEDTOPIC', 'sophy/acq')
|
|
|
|
|
|
WATCHDEBUG = os.getenv('WATCHDEBUG', 1)
|
|
|
|
|
|
# Publish with retain (True or False)
|
|
|
if MQTTRETAIN == 1:
|
|
|
MQTTRETAIN=True
|
|
|
else:
|
|
|
MQTTRETAIN=False
|
|
|
|
|
|
OS = platform.system()
|
|
|
|
|
|
clientid = 'sophy-acq-%s' % os.getpid()
|
|
|
mqtt = paho.Client(clientid, clean_session=True)
|
|
|
|
|
|
if MQTTUSERNAME is not None or MQTTPASSWORD is not None:
|
|
|
mqtt.username_pw_set(MQTTUSERNAME, MQTTPASSWORD)
|
|
|
|
|
|
def on_publish(mosq, userdata, mid):
|
|
|
pass
|
|
|
|
|
|
def on_disconnect(mosq, userdata, rc):
|
|
|
print ("disconnected")
|
|
|
time.sleep(1)
|
|
|
|
|
|
def on_message(client, userdata, msg):
|
|
|
payload = msg.payload.decode()
|
|
|
print(f"Received `{payload}` from `{msg.topic}` topic", flush=True)
|
|
|
|
|
|
def signal_handler(signal, frame):
|
|
|
""" Bail out at the top level """
|
|
|
|
|
|
mqtt.loop_stop()
|
|
|
mqtt.disconnect()
|
|
|
|
|
|
sys.exit(0)
|
|
|
|
|
|
def main():
|
|
|
print('Sophy acquisition monitor will publish messages to {}'.format(MQTTFIXEDTOPIC), flush=True)
|
|
|
mqtt.on_disconnect = on_disconnect
|
|
|
mqtt.on_publish = on_publish
|
|
|
mqtt.connect(MQTTHOST, MQTTPORT)
|
|
|
mqtt.subscribe('sophy/control')
|
|
|
mqtt.on_message = on_message
|
|
|
|
|
|
mqtt.loop_start()
|
|
|
|
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
|
|
|
|
test = json.dumps({'file': '', 'size': ''})
|
|
|
|
|
|
while 1:
|
|
|
mqtt.publish(MQTTFIXEDTOPIC, test, qos=MQTTQOS, retain=MQTTRETAIN)
|
|
|
time.sleep(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
main()
|
