Add basic logging for facts db task

humitifier/infra/__init__.py

@@ -1,22 +1,30 @@
 import aiosqlite
 import asyncio
 import time
+from humitifier.logging import logging
 from humitifier.infra.models.host import Host
 from humitifier.infra.models.hostfacts import HostFacts
 from humitifier.infra.facts import query_inventory_outputs
 from humitifier.db import queries
 from humitifier.config import CONFIG
 
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
+
 
 async def update_fact_db():
     hosts = list(Host.load_inventory())
     ts = int(time.time())
+    logger.info("Querying inventory for %s hosts", len(hosts))
     outputs = await query_inventory_outputs(hosts)
+    logger.info("Parsing %s outputs", len(outputs))
     parse_tasks = [HostFacts.from_output(out, ts) for out in outputs]
     hostfacts = await asyncio.gather(*parse_tasks)
     rows = [await hf.sql_row for hf in hostfacts]
     async with aiosqlite.connect(CONFIG.db) as db:
+        logger.info("Running schema create script on: %s", CONFIG.db)
         await queries.create_facts_table(db)
+        logger.info("Inserting %s rows into %s", len(rows), CONFIG.db)
         await queries.bulk_insert_host_facts(db, rows)
         await db.commit()
 

humitifier/logging.py

@@ -0,0 +1,6 @@
+import logging
+import sys
+
+
+log_format = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+logging.basicConfig(stream=sys.stdout, level=logging.WARNING, format=log_format)

humitifier/tasks.py

@@ -8,8 +8,6 @@ def create_scheduler():
 
     @task_app.task(CONFIG.tasks["infra_update"])
     async def infra_update():
-        print("Updating facts")
         await update_fact_db()
-        print("Fact Update Complete")
 
     return task_app