(20241228) Payments module revamped!

playground/socketio/kafka_passthrough.py

@@ -0,0 +1,154 @@
+"""
+
+    AUTHOR:
+
+        Khushal P Soonderji
+
+    DATE:
+
+        Saturday, 28th Dec., 2024
+
+    OBJECTIVE:
+
+        To provide a quick way to test out passthrough messages over Kafka.
+
+    REFERENCES:
+
+        N/A
+
+    DOWNLOADS:
+
+        N/A
+
+"""
+
+
+#   *****************************************************************************************************************
+#  *****                                                                                                          ****
+# ***                                                   IMPORT                                                      ***
+#  *****                                                                                                          ****
+#   *****************************************************************************************************************
+
+
+# To make sibling directories accessible for imports:
+import sys
+sys.path.append(".")
+sys.path.append("..")
+
+# System-level:
+import os
+
+# Utils:
+from utils_v2.string import json
+from utils_v2.system import files
+from utils_v2.date_time import date_time
+from utils_v2.queue.async_kafka import ProducerKafka, ConsumerKafka, get_ssl_context
+
+# For async activities:
+import asyncio
+
+# Common:
+from shared import constants
+
+# For random choices:
+import random
+
+
+#   *****************************************************************************************************************
+#  *****                                                                                                          ****
+# ***                                           MACROS / ONE-TIME INIT                                              ***
+#  *****                                                                                                          ****
+#   *****************************************************************************************************************
+
+
+# --- Nothing Yet
+
+
+#   *****************************************************************************************************************
+#  *****                                                                                                          ****
+# ***                                                VARIABLES                                                      ***
+#  *****                                                                                                          ****
+#   *****************************************************************************************************************
+
+
+# --- Nothing Yet
+
+
+#   *****************************************************************************************************************
+#  *****                                                                                                          ****
+# ***                                                  FUNCTIONS                                                    ***
+#  *****                                                                                                          ****
+#   *****************************************************************************************************************
+
+
+# --- Nothing Yet
+
+
+#   *****************************************************************************************************************
+#  *****                                                                                                          ****
+# ***                                               MAIN PROGRAM                                                    ***
+#  *****                                                                                                          ****
+#   *****************************************************************************************************************
+
+
+if __name__ == "__main__":
+
+    # Define the test params:
+    TOPIC = "socket-io-bcast"
+    BOOTSTRAP_SERVERS = "del.ditscentre.in:9092"
+    cwd = files.get_cwd()
+    pdir = files.get_parent_directory(cwd, depth = 2)
+    SSL_CONTEXT = ssl_context = get_ssl_context(
+        ca_file = os.path.join(pdir, "creds", "kafka", "cert_authority.pem"),
+        cert_file = os.path.join(pdir, "creds", "kafka", "fullchain.pem"),
+        key_file = os.path.join(pdir, "creds", "kafka", "privkey.pem")
+    )
+
+    async def keep_producing():
+
+        # Create the producer:
+        my_producer = ProducerKafka(
+            topic = TOPIC,
+            bootstrap_servers = BOOTSTRAP_SERVERS,
+            security_protocol = "SSL",
+            ssl_context = SSL_CONTEXT
+        )
+
+        # Create a message for the producer to produce:
+        strategy_message = {
+            "to": "FC9O3N75Ax7B1v4NAAAD",
+            "event": "Strategy",
+            "namespace": "/finstitutions/trading",
+            "data": {
+                "message": "Your strategy ABC says buy XYZ.",
+                "playSound": True
+            }
+        }
+
+        corporate_action_message = {
+            "to": "FC9O3N75Ax7B1v4NAAAD",
+            "event": "Corporate Action",
+            "namespace": "/finstitutions/trading",
+            "data": {
+                "message": "Stock ABC has a corporate action tomorrow.",
+                "playSound": True,
+                "date": "29-Dec-2024",
+                "action": "Extraordinary Board Meeting"
+            }
+        }
+
+        # Keep sending the message in intervals:
+        while True:
+            producer_message = random.choice([
+                strategy_message,
+                corporate_action_message
+            ])
+            success = await my_producer.produce(producer_message)
+            print("PRODUCED:", success)
+            await asyncio.sleep(1.0)
+
+    async def main():
+        await asyncio.gather(*[keep_producing()])
+
+
+    asyncio.run(main())

playground/socketio/to_kafka.py

@@ -130,6 +130,8 @@ def to_kafka(tick: TradingTick) -> bool:
 
     success = False
     summary = tick.summary
+    print(json.to_string(summary))
+    print(json.to_string(tick.model_dump(), default=str))
     summary["messageType"] = "ticks"
     success = kafka_producer.produce(value = summary)
     if not success:
@@ -203,8 +205,15 @@ def main():
     # instruments += kite.instruments(exchange = "BCD")
 
     # # Pick the instruments of interest:
-    instruments = [TradingSymbol.from_zerodha_kite(i) for i in instruments[:1000]]
-    # instruments = [TradingSymbol.from_zerodha_kite(i) for i in instruments if i["tradingsymbol"] in symbols_of_interest]
+    # instruments = [TradingSymbol.from_zerodha_kite(i) for i in instruments[:1000]]
+    # instruments = [
+    #     TradingSymbol.from_zerodha_kite(i) for i in instruments
+    #     if i["tradingsymbol"] in symbols_of_interest or i["name"] in symbols_of_interest
+    # ]
+    instruments = [
+        TradingSymbol.from_zerodha_kite(i) for i in instruments
+        if i["instrument_token"] in [109760007]
+    ]
 
     # Create the lookup:
     for i in instruments: