analyser_hj3415 2.9.8__tar.gz → 2.9.10__tar.gz

analyser_hj3415-2.9.10/.gitignore

@@ -0,0 +1,6 @@
+/dist/
+/__pycache__/
+/tests/_trial_temp/
+.DS_Store
+*.pyc
+*.log

{analyser_hj3415-2.9.8 → analyser_hj3415-2.9.10}/PKG-INFO

@@ -1,18 +1,20 @@
 Metadata-Version: 2.3
 Name: analyser_hj3415
-Version: 2.9.8
+Version: 2.9.10
 Summary: Stock analyser and database processing programs
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 Requires-Dist: utils-hj3415>=2.9.2
-Requires-Dist: db-hj3415>=4.0.3
+Requires-Dist: db-hj3415>=4.1.4
 Requires-Dist: scikit-learn>=1.5.2
 Requires-Dist: plotly>=5.24.1
 Requires-Dist: yfinance>=0.2.44
 Requires-Dist: prophet>=1.1.6
 Requires-Dist: kaleido>=0.2.1
 Requires-Dist: matplotlib>=3.9.2
-Requires-Dist: tensorflow>=2.18.0
+Requires-Dist: tensorflow-macos>=2.16.2; platform_machine == 'arm64' and sys_platform == 'darwin'
+Requires-Dist: tensorflow-metal>=0.5.0; platform_machine == 'arm64' and sys_platform == 'darwin'
+Requires-Dist: tensorflow>=2.18.0; platform_machine != 'arm64' or sys_platform != 'darwin'
 
 ### analyser-hj3415
 

{analyser_hj3415-2.9.8 → analyser_hj3415-2.9.10}/analyser_hj3415/cli.py

@@ -4,8 +4,7 @@ import pprint
 
 from utils_hj3415 import noti, utils
 from utils_hj3415.helpers import SettingsManager
-from db_hj3415 import myredis
-
+from db_hj3415 import myredis, mypeewee
 
 class AnalyserSettingsManager(SettingsManager):
     DEFAULT_SETTINGS = {
@@ -34,6 +33,8 @@ class AnalyserSettingsManager(SettingsManager):
         print(f"{title}이 기본값 ({self.DEFAULT_SETTINGS[title]}) 으로 초기화 되었습니다.")
 
 
+log_cli = mypeewee.LogManager('cli')
+
 def analyser_manager():
     settings_manager = AnalyserSettingsManager()
     expect_earn_from_setting = settings_manager.get_value('EXPECT_EARN')
@@ -47,7 +48,6 @@ def analyser_manager():
     # ranking 파서
     ranking_parser = prophet_subparser.add_parser('ranking', help='prophet 랭킹 책정 및 레디스 저장')
     ranking_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
-    ranking_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
 
     # lstm 명령어 서브파서
     lstm_parser = type_subparsers.add_parser('lstm', help='MyLSTM 타입')
@@ -56,12 +56,10 @@ def analyser_manager():
     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 타입')
@@ -70,13 +68,11 @@ def analyser_manager():
     ranking_parser = red_subparser.add_parser('ranking', help='red 랭킹 책정 및 레디스 저장')
     ranking_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
     ranking_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
-    ranking_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
     # red - get 파서
     red_get_parser = red_subparser.add_parser('get', help='red get 책정 및 레디스 저장')
     red_get_parser.add_argument('code', type=str, help='종목코드 or all')
     red_get_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
     red_get_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
-    red_get_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
 
     # mil 명령어 서브파서
     mil_parser = type_subparsers.add_parser('mil', help='millennial 타입')
@@ -86,7 +82,6 @@ def analyser_manager():
     mil_get_parser.add_argument('code', type=str, help='종목코드 or all')
     mil_get_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
     mil_get_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
-    mil_get_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
 
     # blue 명령어 서브파서
     blue_parser = type_subparsers.add_parser('blue', help='Blue 타입')
@@ -96,7 +91,6 @@ def analyser_manager():
     blue_get_parser.add_argument('code', type=str, help='종목코드 or all')
     blue_get_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
     blue_get_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
-    blue_get_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
 
     # growth 명령어 서브파서
     growth_parser = type_subparsers.add_parser('growth', help='Growth 타입')
@@ -106,7 +100,6 @@ def analyser_manager():
     growth_get_parser.add_argument('code', type=str, help='종목코드 or all')
     growth_get_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
     growth_get_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
-    growth_get_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
 
     # setting 명령어 서브파서
     setting_parser = type_subparsers.add_parser('setting', help='Set and Get settings')
@@ -158,14 +151,12 @@ def analyser_manager():
 
                 # 원래 저장되었던 기대수익률로 다시 복원
                 eval.Red.expect_earn = ee_orig
-            if args.noti:
-                noti.telegram_to('manager', f"오늘의 Red({args.code})를 레디스캐시에 저장했습니다.(유효 12시간)")
+            log_cli.save('INFO', f'run >> analyser red get {args.code}')
 
         elif args.command == 'ranking':
+            log_cli.save('INFO', 'run >> analyser red ranking')
             result = eval.Red.ranking(expect_earn=args.expect_earn, refresh=args.refresh)
             print(result)
-            if args.noti:
-                noti.telegram_to('manager', "오늘의 red ranking을 레디스캐시에 저장했습니다.(유효 12시간)")
 
     elif args.type == 'mil':
         if args.command == 'get':
@@ -181,8 +172,7 @@ def analyser_manager():
                 mil = eval.Mil(args.code)
                 print(f"*** Mil - {mil} ***")
                 pprint.pprint(mil.get(args.refresh))
-            if args.noti:
-                noti.telegram_to('manager', f"오늘의 Mil({args.code})를 레디스 캐시에 저장했습니다.(유효 12시간)")
+            log_cli.save('INFO', f'run >> analyser mil get {args.code}')
 
     elif args.type == 'blue':
         if args.command == 'get':
@@ -198,8 +188,7 @@ def analyser_manager():
                 blue = eval.Blue(args.code)
                 print(f"*** Blue - {blue} ***")
                 pprint.pprint(blue.get(args.refresh))
-            if args.noti:
-                noti.telegram_to('manager', f"오늘의 Blue({args.code})를 레디스 캐시에 저장했습니다.(유효 12시간)")
+            log_cli.save('INFO', f'run >> analyser blue get {args.code}')
 
     elif args.type == 'growth':
         if args.command == 'get':
@@ -215,31 +204,28 @@ def analyser_manager():
                 growth = eval.Growth(args.code)
                 print(f"*** growth - {growth} ***")
                 pprint.pprint(growth.get(args.refresh))
-            if args.noti:
-                noti.telegram_to('manager', f"오늘의 Growth({args.code})를 레디스 캐시에 저장했습니다.(유효 12시간)")
+            log_cli.save('INFO', f'run >> analyser growth get {args.code}')
+
     elif args.type == 'prophet':
         if args.command == 'ranking':
             myprophet = tsa.MyProphet
-            myprophet.expire_time_h = 72
-            result = myprophet.ranking(refresh=args.refresh)
+            result = myprophet.ranking(refresh=args.refresh, expire_time_h=24)
             print(result)
-            if args.noti:
-                noti.telegram_to('manager', f"오늘의 prophet ranking을 레디스캐시에 저장했습니다.(유효 {myprophet.expire_time_h}시간)")
+            log_cli.save('INFO', 'run >> analyser prophet ranking')
+
     elif args.type == 'lstm':
         mylstm = tsa.MyLSTM
-        mylstm.expire_time_h = 72
         if args.command == 'caching':
             if args.top:
-                mylstm.caching_based_on_prophet_ranking(refresh=args.refresh, top=args.top)
+                mylstm.caching_based_on_prophet_ranking(refresh=args.refresh, expire_time_h=24, top=args.top)
             else:
-                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})을 레디스캐시에 저장했습니다.(유효 {mylstm.expire_time_h}시간)")
+                mylstm.caching_based_on_prophet_ranking(refresh=args.refresh, expire_time_h=24)
+            log_cli.save('INFO', f'run >> analyser lstm caching / top={args.top if args.top else 20})')
         elif args.command == 'get':
             assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
-            result = mylstm(args.code).get_final_predictions(refresh=args.refresh)
-            if args.noti:
-                noti.telegram_to('manager', f"LSTM 분석을({args.code})를 레디스 캐시에 저장했습니다.(유효 {mylstm.expire_time_h}시간)")
+            result = mylstm(args.code).get_final_predictions(refresh=args.refresh, expire_time_h=24)
+            log_cli.save('INFO', f'run >> analyser lstm get {args.code}')
+
     elif args.type == 'setting':
         if args.command == 'set':
             settings_manager.set_value(args.title, args.value)
@@ -248,5 +234,8 @@ def analyser_manager():
             print(f"{args.title} 값: {value}")
         elif args.command == 'print':
             print(settings_manager.load_settings())
+
     else:
         parser.print_help()
+
+log_cli.close()

{analyser_hj3415-2.9.8 → analyser_hj3415-2.9.10}/analyser_hj3415/tsa.py

@@ -30,10 +30,8 @@ import logging
 
 tsa_logger = helpers.setup_logger('tsa_logger', logging.WARNING)
 
-expire_time = 3600 * 24
 
 class MyProphet:
-    expire_time_h = 24
     def __init__(self, code: str):
         assert utils.is_6digit(code), f'Invalid value : {code}'
         self.scaler = StandardScaler()
@@ -195,7 +193,7 @@ class MyProphet:
             Exception("to 인자가 맞지 않습니다.")
 
     @classmethod
-    def ranking(cls, refresh = False) -> OrderedDict:
+    def ranking(cls, refresh = False, expire_time_h = 24) -> OrderedDict:
         """
         가장 최근 날짜의 랭킹 분석
         :param refresh:
@@ -205,7 +203,7 @@ class MyProphet:
         redis_name = 'myprophet_ranking'
 
         print(
-            f"redisname: '{redis_name}' / refresh : {refresh} / expire_time : {MyProphet.expire_time_h}h")
+            f"redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time_h}h")
 
         def fetch_ranking() -> dict:
             data = {}
@@ -224,7 +222,7 @@ class MyProphet:
                     print(f"{i}.{p.code}/{p.name} date: {recent_date} 가격:{recent_price} 기대하한값:{yhat_lower} 편차:{deviation}")
             return data
 
-        data_dict = myredis.Base.fetch_and_cache_data(redis_name, refresh, fetch_ranking, timer=MyProphet.expire_time_h * 3600)
+        data_dict = myredis.Base.fetch_and_cache_data(redis_name, refresh, fetch_ranking, timer=expire_time_h * 3600)
 
         return OrderedDict(sorted(data_dict.items(), key=lambda item: item[1], reverse=True))
 
@@ -265,7 +263,6 @@ class MyLSTM:
     """
     # 미래 몇일을 예측할 것인가?
     future_days = 30
-    expire_time_h = 24
 
     def __init__(self, code: str):
         assert utils.is_6digit(code), f'Invalid value : {code}'
@@ -489,7 +486,7 @@ class MyLSTM:
             test_r2=test_r2,
         )
 
-    def get_final_predictions(self, refresh, num=5) -> tuple:
+    def get_final_predictions(self, refresh: bool, expire_time_h: int, num=5) -> tuple:
         """
         미래 예측치를 레디스 캐시를 이용하여 반환함
         :param refresh:
@@ -500,7 +497,7 @@ class MyLSTM:
         redis_name = f'{self.code}_mylstm_predictions'
 
         print(
-            f"redisname: '{redis_name}' / refresh : {refresh} / expire_time : {MyLSTM.expire_time_h}h")
+            f"redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time_h}h")
 
         def fetch_final_predictions(num_in) -> tuple:
             """
@@ -531,7 +528,7 @@ class MyLSTM:
 
             return future_dates_str, final_future_predictions.tolist()
 
-        future_dates_str, final_future_predictions = myredis.Base.fetch_and_cache_data(redis_name, refresh, fetch_final_predictions, num, timer=MyLSTM.expire_time_h * 3600)
+        future_dates_str, final_future_predictions = myredis.Base.fetch_and_cache_data(redis_name, refresh, fetch_final_predictions, num, timer=expire_time_h * 3600)
 
         # 문자열을 날짜 형식으로 변환
         future_dates = [datetime.strptime(date, '%Y-%m-%d') for date in future_dates_str]
@@ -541,14 +538,14 @@ class MyLSTM:
 
         return future_dates, final_future_predictions
 
-    def export(self, refresh=False, to="str") -> Optional[str]:
+    def export(self, refresh=False, expire_time_h=24, to="str") -> Optional[str]:
         """
         prophet과 plotly로 그래프를 그려서 html을 문자열로 반환
         :param refresh:
         :param to: str, htmlfile, png
         :return:
         """
-        future_dates, final_future_predictions = self.get_final_predictions(refresh=refresh)
+        future_dates, final_future_predictions = self.get_final_predictions(refresh=refresh, expire_time_h=expire_time_h)
         final_future_predictions = final_future_predictions.reshape(-1) # 차원을 하나 줄인다.
 
         # 데이터 준비
@@ -636,7 +633,7 @@ class MyLSTM:
             Exception("to 인자가 맞지 않습니다.")
 
     def visualization(self, refresh=True):
-        future_dates, final_future_predictions = self.get_final_predictions(refresh=refresh)
+        future_dates, final_future_predictions = self.get_final_predictions(refresh=refresh, expire_time_h=1)
 
         # 시각화1
         plt.figure(figsize=(10, 6))
@@ -664,15 +661,15 @@ class MyLSTM:
         plt.show()"""
 
     @staticmethod
-    def caching_based_on_prophet_ranking(refresh: bool, top=20):
-        ranking_topn = OrderedDict(itertools.islice(MyProphet.ranking().items(), top))
+    def caching_based_on_prophet_ranking(refresh: bool, expire_time_h: int, top=20):
+        ranking_topn = OrderedDict(itertools.islice(MyProphet.ranking(refresh=False).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()):
             mylstm.code = code
             print(f"{i+1}. {mylstm.code}/{mylstm.name}")
-            mylstm.get_final_predictions(refresh=refresh, num=5)
+            mylstm.get_final_predictions(refresh=refresh, expire_time_h=expire_time_h, num=5)
 
 
 

{analyser_hj3415-2.9.8 → analyser_hj3415-2.9.10}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "analyser_hj3415"
-version = "2.9.8"
+version = "2.9.10"
 description = "Stock analyser and database processing programs"
 readme = "README.md"
 requires-python = ">=3.6"
@@ -21,14 +21,19 @@ requires-python = ">=3.6"
 #]
 dependencies = [
     "utils-hj3415>=2.9.2",
-    "db-hj3415>=4.0.3",
+    "db-hj3415>=4.1.4",
     "scikit-learn>=1.5.2",
     "plotly>=5.24.1",
     "yfinance>=0.2.44",
     "prophet>=1.1.6",
     "kaleido>=0.2.1",   #plotly로 이미지출력위해
     "matplotlib>=3.9.2",
-    "tensorflow>=2.18.0",
+    # Apple Silicon (ARM, macOS)에서 tensorflow-macos 사용
+    "tensorflow-macos>=2.16.2; platform_machine == 'arm64' and sys_platform == 'darwin'",
+    "tensorflow-metal>=0.5.0; platform_machine == 'arm64' and sys_platform == 'darwin'",
+    # 그 외 환경에서는 일반 tensorflow 사용
+    "tensorflow>=2.18.0; platform_machine != 'arm64' or sys_platform != 'darwin'"
+
 ]
 
 [project.scripts]