add: reddit comment tracking

ClashKingInc · Jul 14, 2024 · c4dc6f6 · c4dc6f6
1 parent efe7816
commit c4dc6f6
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 15 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/bot/reddit/track.py b/bot/reddit/track.py
@@ -4,6 +4,7 @@
 import re
 import orjson
 import pendulum as pend
+from loguru import logger
 
 from kafka import KafkaProducer
 from os import getenv
@@ -26,6 +27,7 @@
 
 
 async def post_stream():
+    logger.info("Started Post Stream")
     while True:
         try:
             count = 0
@@ -47,9 +49,11 @@ async def post_stream():
                                            "tags" : tags}}
                     producer.send(topic="reddit", value=orjson.dumps(json_data), timestamp_ms=int(pend.now(tz=pend.UTC).timestamp()) * 1000)
         except Exception as e:
+            logger.error(str(e))
             continue
 
 async def comment_stream():
+    logger.info("Started Comment Stream")
     while True:
         try:
             count = 0
@@ -69,6 +73,7 @@ async def comment_stream():
                                        }}
                 producer.send(topic="reddit", value=orjson.dumps(json_data), timestamp_ms=int(pend.now(tz=pend.UTC).timestamp()) * 1000)
         except Exception as e:
+            logger.error(str(e))
             continue