analyser_hj3415 4.0.10__py3-none-any.whl → 4.1.0__py3-none-any.whl

analyser_hj3415/analyser/tsa/lstm.py

@@ -17,6 +17,7 @@ from utils_hj3415 import tools, setup_logger
 from db_hj3415 import myredis
 from analyser_hj3415.analyser import MIs, tsa
 from analyser_hj3415.analyser.tsa.common import PandasTimestampField, ChartPoint
+import pickle
 
 mylogger = setup_logger(__name__,'INFO')
 expire_time = tools.to_int(os.getenv('DEFAULT_EXPIRE_TIME_H', 48)) * 3600
@@ -99,6 +100,7 @@ class MyLSTM:
     """
     # 미래 몇일을 예측할 것인가?
     future_days = 30
+    REDIS_LSTM_DATA_SUFFIX = "lstm_chart_data"
 
     def __init__(self, ticker: str):
         mylogger.debug(f'MyLSTM set up ticker : {ticker}')
@@ -479,7 +481,7 @@ class MyLSTM:
 
     def generate_chart_data(self, refresh:bool, num=5) -> LSTMChartData:
         mylogger.info("**** Start generate_lstm_chart_data... ****")
-        redis_name = f'{self.ticker}_lstm_chart_data'
+        redis_name = f'{self.ticker}_{MyLSTM.REDIS_LSTM_DATA_SUFFIX}'
 
         mylogger.info(
             f"redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time / 3600}h")
@@ -546,13 +548,34 @@ class MyLSTM:
 
     def get_chart_data(self) -> Optional[LSTMChartData]:
         mylogger.debug("**** Start get_lstm_chart_data... ****")
-        redis_name = f'{self.ticker}_lstm_chart_data'
+        redis_name = f'{self.ticker}_{MyLSTM.REDIS_LSTM_DATA_SUFFIX}'
         if myredis.Base.exists(redis_name):
             mylogger.debug(f"ttl: {myredis.Base.get_ttl(redis_name)}")
             return myredis.Base.get_value(redis_name)
         else:
             return None
 
+    @staticmethod
+    def bulk_get_chart_data(tickers: List[str]) -> Dict[str, Optional[LSTMChartData]]:
+        # --- (1) 파이프라인 GET ---
+        pipe = myredis.Base.redis_client.pipeline()
+        redis_keys = [f"{ticker}_{MyLSTM.REDIS_LSTM_DATA_SUFFIX}" for ticker in tickers]
+        for redis_key in redis_keys:
+            pipe.get(redis_key)
+        results_from_redis = pipe.execute()  # [val1, val2, ...]
+
+        final_results = {}
+
+        for ticker, val in zip(tickers, results_from_redis):
+            if val is None:
+                final_results[ticker] = None
+            else:
+                # Redis에 pickled 데이터가 있다면 언피클해서 담기
+                lstm_data = pickle.loads(val)
+                final_results[ticker] = lstm_data
+
+        return final_results
+
     @staticmethod
     def caching_chart_data(tickers:list, num:int):
         mylogger.info(f"*** caching_chart_data ***")
@@ -605,10 +628,18 @@ class CorpLSTM(MyLSTM):
         MyLSTM.caching_chart_data([CorpLSTM.code_to_ticker(code) for code in ranking_topn.keys() ], num=num)
 
     @staticmethod
-    def caching_chart_data_bulk(codes: list, num=5):
+    def caching_chart_data_bulk(codes: List[str], num=5):
         mylogger.info(f"*** caching_chart_data_bulk : {len(codes)} items  ***")
         MyLSTM.caching_chart_data([CorpLSTM.code_to_ticker(code) for code in codes], num=num)
 
+    @staticmethod
+    def bulk_get_chart_data(codes: List[str]) -> Dict[str, Optional[LSTMChartData]]:
+        ticker_data = MyLSTM.bulk_get_chart_data([CorpLSTM.code_to_ticker(code) for code in codes])
+        code_data = {}
+        for ticker, data in ticker_data.items():
+            code_data[CorpLSTM.ticker_to_code(ticker)] = data
+        return code_data
+
 
 class MILSTM(MyLSTM):
     """

{analyser_hj3415-4.0.10.dist-info → analyser_hj3415-4.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: analyser_hj3415
-Version: 4.0.10
+Version: 4.1.0
 Summary: Stock analyser and database processing programs
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown

{analyser_hj3415-4.0.10.dist-info → analyser_hj3415-4.1.0.dist-info}/RECORD

@@ -9,9 +9,9 @@ analyser_hj3415/analyser/eval/mil.py,sha256=uaS7MHNlytrjEgdLc9S1zhXb5bzVAwlSMlnQ
 analyser_hj3415/analyser/eval/red.py,sha256=DZBV0qMzWD8KGvAbdxrohCVT0a5luduuhKQoN3OtNwc,13909
 analyser_hj3415/analyser/tsa/__init__.py,sha256=pg20ZQRABedTdaIoOr5t043RNKtJ7ji_WmnZrD1IhPg,147
 analyser_hj3415/analyser/tsa/common.py,sha256=ZLUkifupOlLKsrPiqR3y6FaEN4M_loZhxCZXYxkX0us,1874
-analyser_hj3415/analyser/tsa/lstm.py,sha256=tezdQcyLJn5rSNXbU_zApZV-kCTHPX9i7x5YujioWZI,28129
+analyser_hj3415/analyser/tsa/lstm.py,sha256=X59xHpMxEaR95Tf9OIdBT4KXFTMAwZF-_s0CMHRCTk0,29401
 analyser_hj3415/analyser/tsa/prophet.py,sha256=a8XPiikRFbqHjbDT7C3hK5RsqOp5JyAbY7lDFNQfBzM,17884
-analyser_hj3415-4.0.10.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
-analyser_hj3415-4.0.10.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
-analyser_hj3415-4.0.10.dist-info/METADATA,sha256=T5Qh8sutQmZYCb_RUFYKCB5kcVUJnwqYHhzO0mJbpx4,6778
-analyser_hj3415-4.0.10.dist-info/RECORD,,
+analyser_hj3415-4.1.0.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
+analyser_hj3415-4.1.0.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
+analyser_hj3415-4.1.0.dist-info/METADATA,sha256=cVVNfXeU__3S1w621s_3t71x6DIQpg967nhc1Wj-4oM,6777
+analyser_hj3415-4.1.0.dist-info/RECORD,,