|
|
@@ -2,6 +2,8 @@ import sqlite3
|
|
|
import json
|
|
|
import os
|
|
|
import paho.mqtt.client as mqtt
|
|
|
+import logging
|
|
|
+
|
|
|
|
|
|
DB_FILE = os.getenv('DB_FILE', 'agent-summary.db3')
|
|
|
MQTT_HOST = os.getenv('MQTT_HOST', 'iris.dgtlu.net')
|
|
|
@@ -9,6 +11,15 @@ MQTT_TOPIC = os.getenv('MQTT_TOPIC','agent/summary')
|
|
|
MQTT_USER = os.getenv('MQTT_USER','agent')
|
|
|
MQTT_PASS = os.getenv('MQTT_PASS','agent')
|
|
|
|
|
|
+logger = logging.get("__name__")
|
|
|
+debug = os.getenv("DEBUG", "")
|
|
|
+if debug:
|
|
|
+ logger.setLevel(logging.DEBUG)
|
|
|
+else
|
|
|
+ logger.setLevel(logging.INFO)
|
|
|
+logger.addHandler(logging.StreamHandler())
|
|
|
+
|
|
|
+
|
|
|
def db_init():
|
|
|
conn = sqlite3.connect(DB_FILE)
|
|
|
c = conn.cursor()
|
|
|
@@ -35,9 +46,9 @@ def db_save_input(conn: sqlite3.Connection, json_string):
|
|
|
def mqtt_init() -> mqtt:
|
|
|
def on_connect(mqt_client, userdata, flags, rc, properties):
|
|
|
if rc == 0:
|
|
|
- print("Connected to MQTT Broker!")
|
|
|
+ logger.info("Connected to MQTT Broker!")
|
|
|
else:
|
|
|
- print("Failed to connect, return code %d\n", rc)
|
|
|
+ logger.error("Failed to connect, return code %d\n", rc)
|
|
|
|
|
|
client = mqtt.Client(mqtt.CallbackAPIVersion.VERSION2)
|
|
|
client.on_connect = on_connect
|
|
|
@@ -48,7 +59,7 @@ def mqtt_init() -> mqtt:
|
|
|
|
|
|
def mqtt_subscribe(client: mqtt, db_conn: sqlite3.Connection):
|
|
|
def on_message(mqt_client, userdata, msg):
|
|
|
- print(f"Received `{msg.payload.decode()}` from `{msg.topic}` topic")
|
|
|
+ logger.debug(f"Received `{msg.payload.decode()}` from `{msg.topic}` topic")
|
|
|
db_save_input(db_conn, msg.payload.decode())
|
|
|
|
|
|
client.subscribe(MQTT_TOPIC)
|