From 75d69e8d2208413250355fd0305cb784f026a9a4 Mon Sep 17 00:00:00 2001 From: esensar Date: Thu, 26 Apr 2018 15:03:38 +0200 Subject: [PATCH] Reformat files according to PEP8 --- .gitignore | 1 + app/__init__.py | 3 +++ app/mod_devices/__init__.py | 6 ++++++ 3 files changed, 10 insertions(+) diff --git a/.gitignore b/.gitignore index 3f38f3b..0b54df1 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,4 @@ +.idea/ *.py[co] *.DS_* diff --git a/app/__init__.py b/app/__init__.py index ba0a9cc..9d86e35 100644 --- a/app/__init__.py +++ b/app/__init__.py @@ -8,13 +8,16 @@ app.config.from_object('config') app.config.from_pyfile('config.py') db = SQLAlchemy(app) + def on_stop(): print('Application stopping') tear_down_mqtt() + setup_mqtt(app) atexit.register(on_stop) + @app.route("/") def hello(): return "Hello World!" diff --git a/app/mod_devices/__init__.py b/app/mod_devices/__init__.py index cf5851e..73a8456 100644 --- a/app/mod_devices/__init__.py +++ b/app/mod_devices/__init__.py @@ -2,30 +2,36 @@ from flask_mqtt import Mqtt mqtt = Mqtt() + def setup_mqtt(app): mqtt.init_app(app) mqtt.client.on_message = handle_mqtt_message mqtt.client.on_subscribe = handle_subscribe print('MQTT client initialized') + def tear_down_mqtt(): mqtt.unsubscribe_all() if hasattr(mqtt, 'client') and mqtt.client is not None: mqtt.client.disconnect() print('MQTT client destroyed') + @mqtt.on_connect() def handle_connect(client, userdata, flags, rc): print('MQTT client connected') mqtt.subscribe('topic/state') + @mqtt.on_disconnect() def handle_disconnect(): print('MQTT client disconnected') + def handle_subscribe(client, userdata, mid, granted_qos): print('MQTT client subscribed') + def handle_mqtt_message(client, userdata, message): data = dict( topic=message.topic,