analyser_hj3415 2.6.8__py2.py3-none-any.whl → 2.7.0__py2.py3-none-any.whl

analyser_hj3415/cli.py

@@ -1,22 +1,23 @@
 import argparse
 import os
+import pprint
 
 from utils_hj3415 import noti, utils
 from utils_hj3415.helpers import SettingsManager
-from analyser_hj3415.myredis import red_ranking, mil_n_score, red_n_score
 from db_hj3415 import myredis
 
 
-# 파일 경로 상수
-SETTINGS_FILE = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'settings.json')
-
-
 class AnalyserSettingsManager(SettingsManager):
     DEFAULT_SETTINGS = {
         'EXPECT_EARN': 0.06,
+        'RED_RANKING_EXPECT_EARN': 0.06,
     }
     TITLES = DEFAULT_SETTINGS.keys()
 
+    def __init__(self):
+        settings_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'settings.json')
+        super().__init__(settings_file)
+
     def set_value(self, title: str, value: str):
         assert title in self.TITLES, f"title 인자는 {self.TITLES} 중에 있어야 합니다."
         self.settings_dict[title] = value
@@ -34,7 +35,7 @@ class AnalyserSettingsManager(SettingsManager):
 
 
 def analyser_manager():
-    settings_manager = AnalyserSettingsManager(SETTINGS_FILE)
+    settings_manager = AnalyserSettingsManager()
     expect_earn_from_setting = settings_manager.get_value('EXPECT_EARN')
 
     parser = argparse.ArgumentParser(description="Analyser Commands")
@@ -49,21 +50,41 @@ def analyser_manager():
     ranking_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
     ranking_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
     # red - score 파서
-    score_parser = red_subparser.add_parser('score', help='red score 책정 및 레디스 저장')
-    score_parser.add_argument('code', type=str, help='종목코드 or all')
-    score_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
-    score_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
-    score_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
+    red_score_parser = red_subparser.add_parser('score', help='red score 책정 및 레디스 저장')
+    red_score_parser.add_argument('code', type=str, help='종목코드 or all')
+    red_score_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
+    red_score_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
+    red_score_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
 
     # mil 명령어 서브파서
     mil_parser = type_subparsers.add_parser('mil', help='millennial 타입')
     mil_subparser = mil_parser.add_subparsers(dest='command', help='mil 관련된 명령')
     # mil - score 파서
-    score_parser = mil_subparser.add_parser('score', help='mil score 책정 및 레디스 저장')
-    score_parser.add_argument('code', type=str, help='종목코드 or all')
-    score_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
-    score_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
-    score_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
+    mil_score_parser = mil_subparser.add_parser('score', help='mil score 책정 및 레디스 저장')
+    mil_score_parser.add_argument('code', type=str, help='종목코드 or all')
+    mil_score_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
+    mil_score_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
+    mil_score_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
+
+    # blue 명령어 서브파서
+    blue_parser = type_subparsers.add_parser('blue', help='Blue 타입')
+    blue_subparser = blue_parser.add_subparsers(dest='command', help='blue 관련된 명령')
+    # blue - score 파서
+    blue_score_parser = blue_subparser.add_parser('score', help='blue score 책정 및 레디스 저장')
+    blue_score_parser.add_argument('code', type=str, help='종목코드 or all')
+    blue_score_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
+    blue_score_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
+    blue_score_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
+
+    # growth 명령어 서브파서
+    growth_parser = type_subparsers.add_parser('growth', help='Growth 타입')
+    growth_subparser = growth_parser.add_subparsers(dest='command', help='growth 관련된 명령')
+    # growth - score 파서
+    growth_score_parser = growth_subparser.add_parser('score', help='growth score 책정 및 레디스 저장')
+    growth_score_parser.add_argument('code', type=str, help='종목코드 or all')
+    growth_score_parser.add_argument('-e', '--expect_earn', type=float, help='기대수익률 (실수 값 입력)')
+    growth_score_parser.add_argument('-r', '--refresh', action='store_true', help='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
+    growth_score_parser.add_argument('-n', '--noti', action='store_true', help='작업 완료 후 메시지 전송 여부')
 
     # setting 명령어 서브파서
     setting_parser = type_subparsers.add_parser('setting', help='Set and Get settings')
@@ -80,52 +101,101 @@ def analyser_manager():
 
     args = parser.parse_args()
 
+    from analyser_hj3415 import eval
+
     if args.type == 'red':
         if args.command == 'score':
-            # 기대 수익률과 refresh 값 처리
-            expect_earn = float(args.expect_earn if args.expect_earn is not None else expect_earn_from_setting)
-            refresh = args.refresh if args.refresh else False
-
-            # print(expect_earn, refresh)
             if args.code == 'all':
-                print("**** red_n_score all code ****")
+                # 저장된 기대수익률을 불러서 임시저장
+                ee_orig = eval.Red.expect_earn
+
+                red = eval.Red('005930')
+                if args.expect_earn:
+                    eval.Red.expect_earn = float(args.expect_earn)
+
+                print("**** Red - all codes ****")
                 for i, code in enumerate(myredis.Corps.list_all_codes()):
-                    print(f'{i} / {code}')
-                    red_n_score(code, expect_earn, refresh)
+                    red.code = code
+                    print(f"*** {i} / {red} ***")
+                    pprint.pprint(red.get(args.refresh))
+
+                # 원래 저장되었던 기대수익률로 다시 복원
+                eval.Red.expect_earn = ee_orig
             else:
                 assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
-                print(args.code, red_n_score(args.code, expect_earn, refresh))
+                # 저장된 기대수익률을 불러서 임시저장
+                ee_orig = eval.Red.expect_earn
+
+                red = eval.Red(args.code)
+                if args.expect_earn:
+                    args.refresh = True
+                    eval.Red.expect_earn = float(args.expect_earn)
+
+                print(f"*** Red - {red} ***")
+                pprint.pprint(red.get(args.refresh))
+
+                # 원래 저장되었던 기대수익률로 다시 복원
+                eval.Red.expect_earn = ee_orig
             if args.noti:
-                noti.telegram_to('manager', f"오늘의 red and score({args.code})를 레디스캐시에 저장했습니다.(유효 12시간)")
+                noti.telegram_to('manager', f"오늘의 Red({args.code})를 레디스캐시에 저장했습니다.(유효 12시간)")
 
         elif args.command == 'ranking':
-            # 기대 수익률과 refresh 값 처리
-            expect_earn = float(args.expect_earn if args.expect_earn is not None else expect_earn_from_setting)
-            refresh = args.refresh if args.refresh else False
-            print(expect_earn, refresh)
-
-            # red_ranking 함수 호출
-            result = red_ranking(expect_earn, refresh)
+            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 == 'score':
-            # 기대 수익률과 refresh 값 처리
-            expect_earn = float(args.expect_earn if args.expect_earn is not None else expect_earn_from_setting)
-            refresh = args.refresh if args.refresh else False
+            if args.code == 'all':
+                mil = eval.Mil('005930')
+                print("**** Mil - all codes ****")
+                for i, code in enumerate(myredis.Corps.list_all_codes()):
+                    mil.code = code
+                    print(f"*** {i} / {mil} ***")
+                    pprint.pprint(mil.get(args.refresh))
+            else:
+                assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
+                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시간)")
+
+    elif args.type == 'blue':
+        if args.command == 'score':
+            if args.code == 'all':
+                blue = eval.Blue('005930')
+                print("**** Blue - all codes ****")
+                for i, code in enumerate(myredis.Corps.list_all_codes()):
+                    blue.code = code
+                    print(f"*** {i} / {blue} ***")
+                    pprint.pprint(blue.get(args.refresh))
+            else:
+                assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
+                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시간)")
 
+    elif args.type == 'growth':
+        if args.command == 'score':
             if args.code == 'all':
-                print("**** mil_n_score all code ****")
+                growth = eval.Growth('005930')
+                print("**** Growth - all codes ****")
                 for i, code in enumerate(myredis.Corps.list_all_codes()):
-                    print(f'{i} / {code}')
-                    mil_n_score(code, expect_earn, refresh)
+                    growth.code = code
+                    print(f"*** {i} / {growth} ***")
+                    pprint.pprint(growth.get(args.refresh))
             else:
                 assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
-                print(args.code, mil_n_score(args.code, expect_earn, refresh))
+                growth = eval.Growth(args.code)
+                print(f"*** growth - {growth} ***")
+                pprint.pprint(growth.get(args.refresh))
             if args.noti:
-                noti.telegram_to('manager', f"오늘의 mil and score({args.code})를 레디스캐시에 저장했습니다.(유효 12시간)")
+                noti.telegram_to('manager', f"오늘의 Growth({args.code})를 레디스 캐시에 저장했습니다.(유효 12시간)")
+
     elif args.type == 'setting':
         if args.command == 'set':
             settings_manager.set_value(args.title, args.value)