analyser_hj3415 2.9.5__tar.gz → 2.9.7__tar.gz

{analyser_hj3415-2.9.5 → analyser_hj3415-2.9.7}/PKG-INFO

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

{analyser_hj3415-2.9.5 → analyser_hj3415-2.9.7}/analyser_hj3415/cli.py

@@ -55,7 +55,13 @@ def analyser_manager():
     # caching 파서
     caching_parser = lstm_subparser.add_parser('caching', help='lstm 랭킹 책정 및 레디스 저장')
     caching_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
+    caching_parser.add_argument('-t', '--top', type=int, help='prophet ranking 몇위까지 작업을 할지')
     caching_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
+    # red - get 파서
+    lstm_get_parser = lstm_subparser.add_parser('get', help='lstm get 책정 및 레디스 저장')
+    lstm_get_parser.add_argument('code', type=str, help='종목코드')
+    lstm_get_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
+    lstm_get_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
 
     # red 명령어 서브파서
     red_parser = type_subparsers.add_parser('red', help='red 타입')
@@ -219,9 +225,17 @@ def analyser_manager():
                 noti.telegram_to('manager', "오늘의 prophet ranking을 레디스캐시에 저장했습니다.(유효 24시간)")
     elif args.type == 'lstm':
         if args.command == 'caching':
-            result = tsa.MyLSTM('005930').caching_based_on_prophet_ranking(refresh=args.refresh)
+            if args.top:
+                tsa.MyLSTM.caching_based_on_prophet_ranking(refresh=args.refresh, top=args.top)
+            else:
+                tsa.MyLSTM.caching_based_on_prophet_ranking(refresh=args.refresh)
+            if args.noti:
+                noti.telegram_to('manager', f"오늘의 lstm caching(top={args.top if args.top else 20})을 레디스캐시에 저장했습니다.(유효 24시간)")
+        elif args.command == 'get':
+            assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
+            result = tsa.MyLSTM(args.code).get_final_predictions(refresh=args.refresh)
             if args.noti:
-                noti.telegram_to('manager', "오늘의 lstm caching을 레디스캐시에 저장했습니다.(유효 24시간)")
+                noti.telegram_to('manager', f"LSTM 분석을({args.code})를 레디스 캐시에 저장했습니다.(유효 12시간)")
     elif args.type == 'setting':
         if args.command == 'set':
             settings_manager.set_value(args.title, args.value)

{analyser_hj3415-2.9.5 → analyser_hj3415-2.9.7}/analyser_hj3415/tsa.py

@@ -661,14 +661,16 @@ class MyLSTM:
         plt.title('Stock Price Prediction with LSTM Ensemble')
         plt.show()"""
 
-    def caching_based_on_prophet_ranking(self, refresh: bool, top=20):
+    @staticmethod
+    def caching_based_on_prophet_ranking(refresh: bool, top=20):
         ranking_topn = OrderedDict(itertools.islice(MyProphet.ranking().items(), top))
         tsa_logger.info(ranking_topn)
+        mylstm = MyLSTM('005930')
         print(f"*** LSTM prediction redis cashing top{top} items ***")
         for i, (code, _) in enumerate(ranking_topn.items()):
-            print(f"{i+1}. {self.code}/{self.name}")
-            self.code = code
-            self.get_final_predictions(refresh=refresh, num=5)
+            mylstm.code = code
+            print(f"{i+1}. {mylstm.code}/{mylstm.name}")
+            mylstm.get_final_predictions(refresh=refresh, num=5)
 
 
 

{analyser_hj3415-2.9.5 → analyser_hj3415-2.9.7}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "analyser_hj3415"
-version = "2.9.5"
+version = "2.9.7"
 description = "Stock analyser and database processing programs"
 readme = "README.md"
 requires-python = ">=3.6"