phase 4

tests/detect/manual/push_logs.py

@@ -10,12 +10,16 @@ Opens: http://mpr.local.ar/detection/?job=<JOB_ID>
 
 import argparse
 import json
+import logging
 import random
 import time
 from datetime import datetime, timezone
 
 import redis
 
+logging.basicConfig(level=logging.INFO, format="%(levelname)-7s %(name)s — %(message)s")
+logger = logging.getLogger(__name__)
+
 STAGES = ["FrameExtractor", "SceneFilter", "YOLODetector", "OCRStage", "BrandResolver", "VLMLocal", "Aggregator"]
 LEVELS = ["INFO", "INFO", "INFO", "INFO", "WARNING", "DEBUG", "ERROR"]  # weighted toward INFO
 MESSAGES = {
@@ -70,9 +74,9 @@ def main():
     r = redis.Redis(port=args.port, decode_responses=True)
     key = f"detect_events:{args.job}"
 
-    print(f"Pushing {args.count} log events to {key} (redis port {args.port})")
-    print(f"Open: http://mpr.local.ar/detection/?job={args.job}")
-    print()
+    logger.info("Pushing %d log events to %s (redis port %d)", args.count, key, args.port)
+    logger.info("Open: http://mpr.local.ar/detection/?job=%s", args.job)
+    input("\nPress Enter to start...")
 
     for i in range(args.count):
         stage = random.choice(STAGES)
@@ -88,10 +92,10 @@ def main():
         }
 
         r.rpush(key, json.dumps(event))
-        print(f"  {level:7s} {stage:16s} {msg[:60]}")
+        logger.log(getattr(logging, level, logging.INFO), "[%s] %s", stage, msg)
         time.sleep(args.delay)
 
-    print(f"\nDone. {args.count} events pushed.")
+    logger.info("Done. %d events pushed.", args.count)
 
 
 if __name__ == "__main__":