analyser_hj3415 4.4.0__py3-none-any.whl → 4.5.0__py3-none-any.whl

analyser_hj3415/analyser/tsa/lstm.py

@@ -15,6 +15,7 @@ from marshmallow import fields
 
 from utils_hj3415 import tools, setup_logger
 from db_hj3415 import myredis
+from db_hj3415.mymongo.flask import Favorites
 from analyser_hj3415.analyser import MIs, tsa
 from analyser_hj3415.analyser.tsa.common import PandasTimestampField, ChartPoint
 
@@ -614,6 +615,13 @@ class CorpLSTM(MyLSTM):
         mylogger.debug(ranking_topn)
         MyLSTM.caching_chart_data([CorpLSTM.code_to_ticker(code) for code in ranking_topn.keys() ], num=num)
 
+    @staticmethod
+    def caching_chart_data_favorites(num=5):
+        mylogger.info(f"*** caching_chart_data_favorites ***")
+        fav_codes = Favorites().get_all_codes()
+        mylogger.debug(fav_codes)
+        MyLSTM.caching_chart_data([CorpLSTM.code_to_ticker(code) for code in fav_codes], num=num)
+
     @staticmethod
     def caching_chart_data_bulk(codes: List[str], num=5):
         mylogger.info(f"*** caching_chart_data_bulk : {len(codes)} items  ***")
@@ -662,11 +670,28 @@ class MILSTM(MyLSTM):
         self._mi_type = mi_type
         self.ticker = getattr(MIs, mi_type)
 
+    @staticmethod
+    def ticker_to_mitype(ticker: str):
+        dict_fields = MIs._asdict()
+        reverse_map = {value: key for key, value in dict_fields.items()}
+        return reverse_map.get(ticker)
+
+    @staticmethod
+    def mitype_to_ticker(mi_type: str):
+        return getattr(MIs, mi_type)
+
     @staticmethod
     def caching_chart_data_mi_all(num=5):
         mylogger.info(f"*** caching_chart_data_mi_all ***")
         mylogger.debug(f"mi_type : {MIs._fields}")
         MyLSTM.caching_chart_data([getattr(MIs, mi_type) for mi_type in MIs._fields], num=num)
 
+    @staticmethod
+    def bulk_get_chart_data(mi_types: List[str]) -> Dict[str, Optional[LSTMChartData]]:
+        ticker_data = MyLSTM.bulk_get_chart_data([MILSTM.mitype_to_ticker(mi_type) for mi_type in mi_types])
+        mi_data = {}
+        for ticker, data in ticker_data.items():
+            mi_data[MILSTM.ticker_to_mitype(ticker)] = data
+        return mi_data
 
 

analyser_hj3415/cli.py

@@ -31,6 +31,9 @@ def analyser_manager():
     corp_parser = target_subparser.add_parser('corp', help='corp lstm top n 데이터 산출 및 레디스 저장')
     corp_parser.add_argument('-n', '--num', type=int, help='앙상블트레이닝을 몇회반복할것인지')
     corp_parser.add_argument('-t', '--top', type=int, help='prophet ranking 몇위까지 작업을 할지')
+    # lstm - caching - favorites 파서
+    corp_parser = target_subparser.add_parser('favorites', help='corp lstm top n 데이터 산출 및 레디스 저장')
+    corp_parser.add_argument('-n', '--num', type=int, help='앙상블트레이닝을 몇회반복할것인지')
     # lstm - caching - mi 파서
     mi_parser = target_subparser.add_parser('mi', help='mi lstm 데이터 산출 및 레디스 저장')
     mi_parser.add_argument('-n', '--num', type=int, help='앙상블트레이닝을 몇회반복할것인지')
@@ -193,6 +196,10 @@ def analyser_manager():
                     tsa.CorpLSTM.caching_chart_data_topn(top=args.top, num=args.num)
                     mymongo.Logs.save('cli', 'INFO',
                                       f'run >> analyser lstm caching corp -t {args.top} -n {args.num}')
+                elif args.target == 'favorites':
+                    tsa.CorpLSTM.caching_chart_data_favorites(num=args.num)
+                    mymongo.Logs.save('cli', 'INFO',
+                                      f'run >> analyser lstm caching favorites -n {args.num}')
                 elif args.target == 'mi':
                     tsa.MILSTM.caching_chart_data_mi_all(num=args.num)
                     mymongo.Logs.save('cli', 'INFO',

{analyser_hj3415-4.4.0.dist-info → analyser_hj3415-4.5.0.dist-info}/METADATA

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

{analyser_hj3415-4.4.0.dist-info → analyser_hj3415-4.5.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 analyser_hj3415/.DS_Store,sha256=fppknfhS_OdwBFgHHQaBDBRZG0WleCeML46EOj8nxIo,6148
 analyser_hj3415/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-analyser_hj3415/cli.py,sha256=0oFfvaTXSh7UbFa8WcAlLpbkT6vxnSZIk7hxVL5PTr8,11410
+analyser_hj3415/cli.py,sha256=drfh0cqZ0LUHUJMJeyXcv1QTD7VYJMr58gcY4WHsVk4,11956
 analyser_hj3415/analyser/__init__.py,sha256=N0XyBfWJNpDS_6JYziKETWePO_jtFB1m7E8Qbwt1w0Q,1096
 analyser_hj3415/analyser/eval/__init__.py,sha256=IP1d0Q3nOCAD3zK1qxrC685MkJQfUh-qaXc7xptTxk8,80
 analyser_hj3415/analyser/eval/blue.py,sha256=KBoKevikS74ys9zcVls6GaEdBkW-avcxHWkSPywOcPU,10644
@@ -10,9 +10,9 @@ analyser_hj3415/analyser/eval/mil.py,sha256=6_1SuWqG1fsuZUKuaABhSqNsYCgj5np35auW
 analyser_hj3415/analyser/eval/red.py,sha256=Abf5HPsNWKnDF4cbZ8xhlNsMF6ljN_sShCAYyPAhCYs,12028
 analyser_hj3415/analyser/tsa/__init__.py,sha256=7j-WshikzsDGGo_wuFoMPNmYfY-bLSEMd6o1MKSQKLI,150
 analyser_hj3415/analyser/tsa/common.py,sha256=iRwk88zBtEIqzSmTQWmQIWiiqZ9gN7GLtDUa5rx8IGM,1918
-analyser_hj3415/analyser/tsa/lstm.py,sha256=howngiVaEFKccYif6vrREm6skZiDzxazq44SdY_82yQ,28906
+analyser_hj3415/analyser/tsa/lstm.py,sha256=8wg-9IjJGRzsZd1_KsrHhx3MExhf6DlrY5EJgqL6qR4,29929
 analyser_hj3415/analyser/tsa/myprophet.py,sha256=QHeIHinCZzMhuLpAqrMdy4E56MFbCGd89i9dR8cGXrw,18976
-analyser_hj3415-4.4.0.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
-analyser_hj3415-4.4.0.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
-analyser_hj3415-4.4.0.dist-info/METADATA,sha256=4m0sEx7lcK10xFNx7-C4gDVvWwWXlcJesac3n7rwPgA,6811
-analyser_hj3415-4.4.0.dist-info/RECORD,,
+analyser_hj3415-4.5.0.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
+analyser_hj3415-4.5.0.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
+analyser_hj3415-4.5.0.dist-info/METADATA,sha256=FFrnXcurjCX0QDspgxB74WbJhP7Du1AjJhAmfH02SGc,6811
+analyser_hj3415-4.5.0.dist-info/RECORD,,