kuhl-haus-mdp 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

kuhl_haus/mdp/analyzers/massive_data_analyzer.py

@@ -5,6 +5,7 @@ from massive.websocket.models import EventType
 
 from kuhl_haus.mdp.models.market_data_analyzer_result import MarketDataAnalyzerResult
 from kuhl_haus.mdp.models.market_data_cache_keys import MarketDataCacheKeys
+from kuhl_haus.mdp.models.market_data_cache_ttl import MarketDataCacheTTL
 
 
 class MassiveDataAnalyzer:
@@ -18,7 +19,7 @@ class MassiveDataAnalyzer:
             EventType.EquityQuote.value: self.handle_equity_quote_event,
         }
 
-    async def analyze_data(self, data: dict) -> Optional[List[MarketDataAnalyzerResult]]:
+    def analyze_data(self, data: dict) -> Optional[List[MarketDataAnalyzerResult]]:
         """
         Process raw market data message
 
@@ -30,73 +31,63 @@ class MassiveDataAnalyzer:
         """
         if "event_type" not in data:
             self.logger.info("Message missing 'event_type'")
-            return await self.handle_unknown_event(data)
+            return self.handle_unknown_event(data)
         event_type = data.get("event_type")
 
         if "symbol" not in data:
             self.logger.info("Message missing 'symbol'")
-            return await self.handle_unknown_event(data)
+            return self.handle_unknown_event(data)
         symbol = data.get("symbol")
 
         if event_type in self.event_handlers:
-            return await self.event_handlers[event_type](**{"data": data, "symbol": symbol})
+            return self.event_handlers[event_type](**{"data": data, "symbol": symbol})
         else:
             self.logger.warning(f"Unsupported message type: {event_type}")
-            return await self.handle_unknown_event(data)
+            return self.handle_unknown_event(data)
 
-    async def handle_luld_event(self, data: dict, symbol: str) -> Optional[List[MarketDataAnalyzerResult]]:
-        try:
-            return [MarketDataAnalyzerResult(
-                data=data,
-                cache_key=f"{MarketDataCacheKeys.HALTS.value}:{symbol}",
-                cache_ttl=28500,  # 7 hours, 55 minutes
-                publish_key=f"{MarketDataCacheKeys.HALTS.value}:{symbol}",
-            )]
-        except Exception as e:
-            self.logger.error(f"Error processing LULD message for {symbol}: {data}", e)
+    @staticmethod
+    def handle_luld_event(data: dict, symbol: str) -> Optional[List[MarketDataAnalyzerResult]]:
+        return [MarketDataAnalyzerResult(
+            data=data,
+            cache_key=f"{MarketDataCacheKeys.HALTS.value}:{symbol}",
+            cache_ttl=MarketDataCacheTTL.THREE_DAYS.value,
+            publish_key=f"{MarketDataCacheKeys.HALTS.value}:{symbol}",
+        )]
 
-    async def handle_equity_agg_event(self, data: dict, symbol: str) -> Optional[List[MarketDataAnalyzerResult]]:
-        try:
-            return [MarketDataAnalyzerResult(
-                data=data,
-                # cache_key=f"{MarketDataCacheKeys.AGGREGATE.value}:{symbol}",
-                # cache_ttl=259200,  # 3 days
-                publish_key=f"{MarketDataCacheKeys.AGGREGATE.value}:{symbol}",
-            )]
-        except Exception as e:
-            self.logger.error(f"Error processing EquityAgg message for {symbol}: {data}", e)
+    @staticmethod
+    def handle_equity_agg_event(data: dict, symbol: str) -> Optional[List[MarketDataAnalyzerResult]]:
+        return [MarketDataAnalyzerResult(
+            data=data,
+            cache_key=f"{MarketDataCacheKeys.AGGREGATE.value}:{symbol}",
+            cache_ttl=MarketDataCacheTTL.THREE_DAYS.value,
+            publish_key=f"{MarketDataCacheKeys.AGGREGATE.value}:{symbol}",
+        )]
 
-    async def handle_equity_trade_event(self, data: dict, symbol: str) -> Optional[List[MarketDataAnalyzerResult]]:
-        try:
-            return [MarketDataAnalyzerResult(
-                data=data,
-                # cache_key=f"{MarketDataCacheKeys.TRADES.value}:{symbol}",
-                # cache_ttl=28500,  # 7 hours, 55 minutes
-                publish_key=f"{MarketDataCacheKeys.TRADES.value}:{symbol}",
-            )]
-        except Exception as e:
-            self.logger.error(f"Error processing EquityTrade message for {symbol}: {data}", e)
+    @staticmethod
+    def handle_equity_trade_event(data: dict, symbol: str) -> Optional[List[MarketDataAnalyzerResult]]:
+        return [MarketDataAnalyzerResult(
+            data=data,
+            cache_key=f"{MarketDataCacheKeys.TRADES.value}:{symbol}",
+            cache_ttl=MarketDataCacheTTL.EIGHT_HOURS.value,
+            publish_key=f"{MarketDataCacheKeys.TRADES.value}:{symbol}",
+        )]
 
-    async def handle_equity_quote_event(self, data: dict, symbol: str) -> Optional[List[MarketDataAnalyzerResult]]:
-        try:
-            return [MarketDataAnalyzerResult(
-                data=data,
-                # cache_key=f"{MarketDataCacheKeys.QUOTES.value}:{symbol}",
-                # cache_ttl=259200,  # 3 days
-                publish_key=f"{MarketDataCacheKeys.QUOTES.value}:{symbol}",
-            )]
-        except Exception as e:
-            self.logger.error(f"Error processing EquityQuote message for {symbol}: {data}", e)
+    @staticmethod
+    def handle_equity_quote_event(data: dict, symbol: str) -> Optional[List[MarketDataAnalyzerResult]]:
+        return [MarketDataAnalyzerResult(
+            data=data,
+            cache_key=f"{MarketDataCacheKeys.QUOTES.value}:{symbol}",
+            cache_ttl=MarketDataCacheTTL.THREE_DAYS.value,
+            publish_key=f"{MarketDataCacheKeys.QUOTES.value}:{symbol}",
+        )]
 
-    async def handle_unknown_event(self, data: dict) -> Optional[List[MarketDataAnalyzerResult]]:
-        try:
-            timestamp = f"{time()}".replace('.','')
-            cache_key = f"{MarketDataCacheKeys.UNKNOWN.value}:{timestamp}"
-            return [MarketDataAnalyzerResult(
-                data=data,
-                cache_key=cache_key,
-                cache_ttl=86400,  # 1 days
-                publish_key=f"{MarketDataCacheKeys.UNKNOWN.value}",
-            )]
-        except Exception as e:
-            self.logger.error(f"Error processing unknown message type: {data}", e)
+    @staticmethod
+    def handle_unknown_event(data: dict) -> Optional[List[MarketDataAnalyzerResult]]:
+        timestamp = f"{time()}".replace('.','')
+        cache_key = f"{MarketDataCacheKeys.UNKNOWN.value}:{timestamp}"
+        return [MarketDataAnalyzerResult(
+            data=data,
+            cache_key=cache_key,
+            cache_ttl=MarketDataCacheTTL.ONE_DAY.value,
+            publish_key=f"{MarketDataCacheKeys.UNKNOWN.value}",
+        )]

kuhl_haus/mdp/integ/massive_data_processor.py

@@ -114,8 +114,8 @@ class MassiveDataProcessor:
                     web_socket_message = json.loads(message.body.decode())
                     data = WebSocketMessageSerde.to_dict(web_socket_message)
 
-                    # Delegate to analyzer (async)
-                    analyzer_results = await self.analyzer.analyze_data(data)
+                    # Delegate to analyzer
+                    analyzer_results = self.analyzer.analyze_data(data)
                     if analyzer_results:
                         self.processed += 1
                         for analyzer_result in analyzer_results:

kuhl_haus/mdp/models/market_data_cache_ttl.py

@@ -0,0 +1,19 @@
+from enum import Enum
+
+
+class MarketDataCacheTTL(Enum):
+    # Hours
+    ONE_HOUR = 3600
+    TWO_HOURS = 7200
+    FOUR_HOURS = 14400
+    SIX_HOURS = 21600
+    EIGHT_HOURS = 28800
+
+    # Days
+    ONE_DAY = 86400
+    TWO_DAYS = 172800
+    THREE_DAYS = 259200
+    FOUR_DAYS = 345600
+    FIVE_DAYS = 432000
+    SIX_DAYS = 518400
+    SEVEN_DAYS = 604800

{kuhl_haus_mdp-0.1.1.dist-info → kuhl_haus_mdp-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kuhl-haus-mdp
-Version: 0.1.1
+Version: 0.1.2
 Summary: Market data processing pipeline for stock market scanner
 Author-Email: Tom Pounders <git@oldschool.engineer>
 License: The MIT License (MIT)

{kuhl_haus_mdp-0.1.1.dist-info → kuhl_haus_mdp-0.1.2.dist-info}/RECORD

@@ -1,7 +1,7 @@
 kuhl_haus/mdp/__init__.py,sha256=5dEpAdB3kypH8tCRECoXwbly1WV9kFU5kh8ldGSa0VI,349
 kuhl_haus/mdp/analyzers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kuhl_haus/mdp/analyzers/analyzer.py,sha256=eluYM2Iib5kgbpNZUSk2qEUL-j83ZTb3zmEmRazrmiM,404
-kuhl_haus/mdp/analyzers/massive_data_analyzer.py,sha256=4EEKDJjJmxCdL5nFa87AogabBHgGVld02UuWTgqfFjI,4536
+kuhl_haus/mdp/analyzers/massive_data_analyzer.py,sha256=WSb7T8X4u2ue7Du7sf_fqxjgjEbR6ThllSNT1CncIM0,3866
 kuhl_haus/mdp/analyzers/top_stocks.py,sha256=AbRnPHSVrJgUq3CDV8SaNstldqoimlI23gpG69lzYBM,18759
 kuhl_haus/mdp/components/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kuhl_haus/mdp/components/market_data_cache.py,sha256=r5sJHuSuLiw9BVckW--aWZHHIMqOTCf-pFURA7kef3Q,1070
@@ -12,18 +12,19 @@ kuhl_haus/mdp/helpers/process_manager.py,sha256=Is3Jx8nlBWvywQ1acdsdaSJTAG0olKsk
 kuhl_haus/mdp/helpers/queue_name_resolver.py,sha256=l_zfRLxrjR9uwRCV2VDO4vPWLK_lj5KVG2p4Lh8xWiw,770
 kuhl_haus/mdp/integ/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kuhl_haus/mdp/integ/massive_data_listener.py,sha256=fPEYc6zZzHzFFjbP3zFInajKtEGInj8UQKKo3nKQEwQ,5098
-kuhl_haus/mdp/integ/massive_data_processor.py,sha256=9I0tH9sZNs9Y0TyKIBKix_qlEksEZt5NRfP-Zf3FovE,8708
+kuhl_haus/mdp/integ/massive_data_processor.py,sha256=qktzLfuqrOgE4C9iZs4mXFvHt2BckgevRP8pEakzggA,8694
 kuhl_haus/mdp/integ/massive_data_queues.py,sha256=zC_uV2vwZCMyVerDQ18RAQwIMMF75iK4qUSqwuWqgwc,5050
 kuhl_haus/mdp/integ/utils.py,sha256=9JEpl2yr2LghOLrJUDxi-4dtDK3DZ1wBTZ1uxBJsFbQ,1309
 kuhl_haus/mdp/integ/web_socket_message_serde.py,sha256=XdaoaByc7IhtzbPDXBtXKOTjyDzfPSDuZVCoHSIaTl4,5468
 kuhl_haus/mdp/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kuhl_haus/mdp/models/market_data_analyzer_result.py,sha256=iICb5GVCtuqARNbR1JNCAfbxMijM3uppDNdL8_FB3eI,422
 kuhl_haus/mdp/models/market_data_cache_keys.py,sha256=5iScBMhVQaG3p9P45veE-uRT7c6JY7k6j4DcvSEXENA,942
+kuhl_haus/mdp/models/market_data_cache_ttl.py,sha256=4KvsPeg84-sp4viUX6reN8CZYiM2aF9FgfXQmPbj3hw,348
 kuhl_haus/mdp/models/market_data_pubsub_keys.py,sha256=PEIPXK9jBehJB7G4pqoSuQZcfMZgOQq8Yho1itqv-1A,1306
 kuhl_haus/mdp/models/market_data_scanner_names.py,sha256=BYn1C0rYgGF1Sq583BkHADKUu-28ytNZQ-XgptuCH-Y,260
 kuhl_haus/mdp/models/massive_data_queue.py,sha256=MfYBcjVc4Fi61DWIvvhhWLUOiLmRpE9egtW-2KH6FTE,188
-kuhl_haus_mdp-0.1.1.dist-info/METADATA,sha256=Pwfg52rEA4A6MQdJtiUdF4BgL-U2NjSFPDCAYJYdaYE,8688
-kuhl_haus_mdp-0.1.1.dist-info/WHEEL,sha256=tsUv_t7BDeJeRHaSrczbGeuK-TtDpGsWi_JfpzD255I,90
-kuhl_haus_mdp-0.1.1.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-kuhl_haus_mdp-0.1.1.dist-info/licenses/LICENSE.txt,sha256=DRkJftAJcMqoTkQ_Y6-HtKj3nm4pZah_p8XBZiYnw-c,1079
-kuhl_haus_mdp-0.1.1.dist-info/RECORD,,
+kuhl_haus_mdp-0.1.2.dist-info/METADATA,sha256=iOqk7u1RoyY-mBiIDGVI06TSDGulUuHo0GF9G6JYcdQ,8688
+kuhl_haus_mdp-0.1.2.dist-info/WHEEL,sha256=tsUv_t7BDeJeRHaSrczbGeuK-TtDpGsWi_JfpzD255I,90
+kuhl_haus_mdp-0.1.2.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+kuhl_haus_mdp-0.1.2.dist-info/licenses/LICENSE.txt,sha256=DRkJftAJcMqoTkQ_Y6-HtKj3nm4pZah_p8XBZiYnw-c,1079
+kuhl_haus_mdp-0.1.2.dist-info/RECORD,,