master

utils_v2/queue/async_kafka.py

@@ -473,7 +473,7 @@ if __name__ == "__main__":
 
     import asyncio
     import time
-    from data_models.kafka_message import KafkaMessage
+    # from data_models.kafka_message import KafkaMessage
 
     ssl_ctx = get_ssl_context(
         ca_file = r"/home/developer/PycharmProjects/utils/cred/kafka/cert_authority.pem",
@@ -497,13 +497,13 @@ if __name__ == "__main__":
 
         while True:
             messages = await consumer.consume(count = 1)
-            if len(messages) > 0: print("MESSAGE:", json.to_string(messages[0], default=str))
+            if len(messages) > 0: print("MESSAGE:", json.to_string(messages[0], default = str))
             await asyncio.sleep(1.0)
 
     async def producer_test():
 
         producer = ProducerKafka(
-            topic = "kft_file_upload",
+            topic = "tickers",
             bootstrap_servers = "del.ditscentre.in:9092",
             security_protocol = "SSL",
             ssl_context = ssl_ctx
@@ -512,43 +512,11 @@ if __name__ == "__main__":
         print("READY!")
 
         while True:
-            my_msg = KafkaMessage(
-                data = {
-                    "accepted": False,
-                    "reason": "low resolution"
-                },
-                media = {
-                    "name": "pikachu_poster.jpg",
-                    "ext": "jpg",
-                    "url": "https://nexcom.ditscentre.in/utils/files/small/download/66ded1c1c1c05139a618b5ff",
-                    "attr": {
-                        "user": "SarangKabir",
-                        "project": "ACE-PGP",
-                        "id": 173,
-                        "campaignActivityId": "25",
-                        "idCampaign": 49,
-                        "phoneNo": "7977821877"
-                    }
-                },
-                appId = "aceWockhardt",
-                proc = {
-                    "name": "_assessImg",
-                    "attr": {
-                        "blurThreshold": 0.25,
-                        "clarityThreshold": 0.65,
-                        "nsfwThreshold": 0.25,
-                        "minWidth": 512,
-                        "minHeight": 512
-                    }
-                },
-                ack = None
-            )
-            success = await producer.produce(my_msg.model_dump())
-            print("produced...")
-            time.sleep(1.0)
-            break
+            success = await producer.produce({"name": "Bhopli", "color": "orange"})
+            print("PRODUCED:", success)
+            await asyncio.sleep(0.25)
 
         await producer.close()
 
 
-    asyncio.run(consumer_test())
+    asyncio.run(producer_test())