Quellcode durchsuchen

add proper logger to summary

adri vor 3 Wochen
Ursprung
Commit
a29da0eb49
2 geänderte Dateien mit 31 neuen und 3 gelöschten Zeilen
  1. 17 0
      .gitignore
  2. 14 3
      summary/main.py

+ 17 - 0
.gitignore

@@ -0,0 +1,17 @@
+.env
+Commands/
+*.bak
+Masks/
+Media/
+Plugins/
+XML.bak/
+XML/
+backup/
+builder/
+*.disabled
+*.old
+old/
+sounds/
+summary/*.csv
+summary/db/
+

+ 14 - 3
summary/main.py

@@ -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)