analyser_hj3415 2.10.1__tar.gz → 2.10.2__tar.gz

{analyser_hj3415-2.10.1 → analyser_hj3415-2.10.2}/PKG-INFO

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

{analyser_hj3415-2.10.1 → analyser_hj3415-2.10.2}/analyser_hj3415/cli.py

@@ -114,7 +114,7 @@ def analyser_manager():
 
     args = parser.parse_args()
 
-    from analyser_hj3415 import eval, tsa
+    from . import eval, tsa, score
 
     if args.type == 'red':
         if args.command == 'get':
@@ -206,8 +206,7 @@ def analyser_manager():
 
     elif args.type == 'prophet':
         if args.command == 'ranking':
-            myprophet = tsa.MyProphet
-            result = myprophet.ranking(refresh=args.refresh, expire_time_h=48)
+            result = score.Score.ranking(refresh=args.refresh, expire_time_h=48)
             print(result)
             mymongo.Logs.save('cli','INFO', 'run >> analyser prophet ranking')
 

{analyser_hj3415-2.10.1 → analyser_hj3415-2.10.2}/analyser_hj3415/eval.py

@@ -4,7 +4,7 @@ from utils_hj3415 import utils, helpers
 from typing import Tuple
 from db_hj3415 import myredis, mymongo
 import math
-from analyser_hj3415.cli import AnalyserSettingsManager
+from .cli import AnalyserSettingsManager
 from collections import OrderedDict
 import logging
 

analyser_hj3415-2.10.2/analyser_hj3415/score.py

@@ -0,0 +1,157 @@
+from db_hj3415 import myredis
+from utils_hj3415 import utils, helpers
+import datetime
+from collections import OrderedDict
+from . import eval, tsa
+
+import logging
+
+score_logger = helpers.setup_logger('score_logger', logging.WARNING)
+
+
+def is_within_last_three_days(date_to_check: datetime.datetime.date) -> bool:
+    today = datetime.datetime.now().date()  # 현재 날짜 (시간은 무시)
+    # print('today - ', today)
+    three_days_ago = today - datetime.timedelta(days=3)  # 3일 전 날짜
+    return three_days_ago <= date_to_check <= today
+
+
+class Score:
+    def __init__(self, code):
+        self._code = code
+        self.c101 = myredis.C101(code)
+        self.name = self.c101.get_name()
+        self.c108 = myredis.C108(code)
+        self.dart = myredis.Dart(code)
+        self.red = eval.Red(code)
+        self.mil = eval.Mil(code)
+        self.lstm = tsa.MyLSTM(code)
+        self.prophet = tsa.MyProphet(code)
+
+    @property
+    def code(self) -> str:
+        return self._code
+
+    @code.setter
+    def code(self, code: str):
+        assert utils.is_6digit(code), f'Invalid value : {code}'
+        score_logger.info(f'change code : {self.code} -> {code}')
+        self._code = code
+        self.c101.code = code
+        self.name = self.c101.get_name()
+        self.c108.code = code
+        self.dart.code = code
+        self.red.code = code
+        self.mil.code = code
+        self.lstm.code = code
+        self.prophet.code = code
+
+    def get(self, refresh=False, expire_time_h=24) -> dict:
+        """
+        한 종목의 각분야 평가를 모아서 딕셔너리 형태로 반환함.
+        redis_name = self.code + '_score'
+
+        Returns:
+            dict: A dictionary containing the following key-value pairs:
+
+                - 'code': str - 종목코드
+                - 'name': str - 종목명
+                - '시가총액': str - 시가총액
+                - 'is_update_c108': bool - 최근 3일 이내에 c108이 없데이트 되었는가
+                - 'is_update_dart': bool - 최근 3일 이내에 Dart가 없데이트 되었는가
+                - 'red_score': float - Red score
+                - '이익지표': float - Mil의 이익지표
+                - '주주수익률': float - Mil의 주주수익률
+                - 'is_lstm_up': Union[bool, None] - lstm 예측치가 상승인지 아닌지, returns None - 데이터가 없으면..
+                - 'prophet_score': int - prophet score
+        """
+        print(f"{self.code}/{self.name}의 scoring을 시작합니다.")
+        redis_name = self.code + '_score'
+        print(
+            f"redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time_h}h")
+
+        def fetch_score() -> dict:
+            score_logger.info("시가총액 데이터 추출중..")
+            시가총액 = utils.format_large_number(int(self.c101.get_recent()['시가총액']))
+
+            score_logger.info("C108 최근 데이터 추출중..")
+            # c108이 최근에 업데이트 되었는지...
+            c108_recent_date = self.c108.get_recent_date()
+            # print('code - ', code, ' | c108 recent date - ', c108_recent_date.date())
+            if c108_recent_date is None:
+                is_update_c108 = False
+            else:
+                is_update_c108 = is_within_last_three_days(c108_recent_date.date())
+
+            score_logger.info("Dart 최근 데이터 추출중..")
+            # dart가 최근에 업데이트 되었는지...
+            dart_recent_date = self.dart.get_recent_date()
+            # print('code - ', code, ' | dart recent date - ', dart_recent_date.date())
+            if dart_recent_date is None:
+                is_update_dart = False
+            else:
+                is_update_dart = is_within_last_three_days(dart_recent_date.date())
+
+            score_logger.info("Red score 계산중..")
+            red_score = self.red.get(verbose=False).score
+
+            score_logger.info("Mil data 계산중..")
+            mil_data = self.mil.get(verbose=False)
+
+            score_logger.info("Lstm 최근 데이터 조회중..")
+            if myredis.Base.exists(f'{self.code}_mylstm_predictions'):
+                is_lstm_up = self.lstm.is_up()
+            else:
+                is_lstm_up = None
+
+            score_logger.info("\tProphet 최근 데이터 조회중..")
+            prophet_score = self.prophet.scoring()
+
+            return {
+                'code': self.code,
+                'name': self.name,
+                '시가총액': 시가총액,
+                'is_update_c108': is_update_c108,
+                'is_update_dart': is_update_dart,
+                'red_score': red_score,
+                '이익지표': mil_data.이익지표,
+                '주주수익률': mil_data.주주수익률,
+                'is_lstm_up': is_lstm_up,
+                'prophet_score': prophet_score,
+            }
+        data_dict = myredis.Base.fetch_and_cache_data(redis_name, refresh, fetch_score, timer=expire_time_h * 3600)
+        return data_dict
+
+    @classmethod
+    def ranking(self, refresh=False, expire_time_h=24, top='all') -> OrderedDict:
+        print("**** Start score_ranking... ****")
+        redis_name = 'score_ranking'
+
+        print(
+            f"redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time_h}h")
+
+        def fetch_ranking() -> dict:
+            data = {}
+            s = Score('005930')
+            for code in myredis.Corps.list_all_codes():
+                try:
+                    s.code = code
+                except ValueError:
+                    score_logger.error(f'score ranking error : {code}')
+                    continue
+                score = s.get()
+                data[code] = score
+            return data
+
+        data_dict = myredis.Base.fetch_and_cache_data(redis_name, refresh, fetch_ranking, timer=expire_time_h * 3600)
+
+        # prophet_score를 기준으로 정렬
+        ranking = OrderedDict(sorted(data_dict.items(), key=lambda x: x[1]['prophet_score'], reverse=True))
+
+        if top == 'all':
+            return ranking
+        else:
+            if isinstance(top, int):
+                return OrderedDict(list(ranking.items())[:top])
+            else:
+                raise ValueError("top 인자는 'all' 이나 int형 이어야 합니다.")

{analyser_hj3415-2.10.1 → analyser_hj3415-2.10.2}/analyser_hj3415/tsa.py

@@ -193,6 +193,22 @@ class MyProphet:
             Exception("to 인자가 맞지 않습니다.")
 
     def scoring(self) -> int:
+        """
+        prophet의 yhat_lower 예측치와 주가를 비교하여 주가가 낮으면 양의 점수를 높으면 음의 점수를 준다.
+
+        Returns:
+            int: The calculated score based on the deviation between the recent price
+            and the expected lower limit.
+
+        Parameters:
+            None
+
+        Raises:
+            AttributeError: Raised if the necessary attributes like `df_real` or methods like `get_yhat`
+            are not correctly set or implemented.
+            KeyError: Raised if the expected keys (`'yhat_lower'` or `'y'`) are not found in the data involved.
+            ValueError: Raised if the format of data does not conform to expected structure for calculations.
+        """
         last_real_data = self.df_real.iloc[-1]
         recent_price = last_real_data['y']
         recent_date = datetime.strftime(last_real_data['ds'], '%Y-%m-%d')
@@ -204,45 +220,9 @@ class MyProphet:
             score = -deviation
         else:
             score = deviation
-        print(f"{self.code}/{self.name} date: {recent_date} 가격:{recent_price} 기대하한값:{yhat_lower} 편차:{deviation} score:{score}")
+        tsa_logger.info(f"{self.code}/{self.name} date: {recent_date} 가격:{recent_price} 기대하한값:{yhat_lower} 편차:{deviation} score:{score}")
         return score
 
-    @classmethod
-    def ranking(cls, refresh = False, expire_time_h = 24, top='all') -> OrderedDict:
-        """
-        가장 최근 날짜의 랭킹 분석
-        :param refresh:
-        :return:
-        """
-        print("**** Start myprophet_ranking... ****")
-        redis_name = 'myprophet_ranking'
-
-        print(
-            f"redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time_h}h")
-
-        def fetch_ranking() -> dict:
-            data = {}
-            p = MyProphet('005930')
-            for code in myredis.Corps.list_all_codes():
-                try:
-                    p.code = code
-                except ValueError:
-                    tsa_logger.error(f'myprophet ranking error : {code}/{myredis.Corps(code, "c101").get_name()}')
-                    continue
-                score = p.scoring()
-                data[code] = score
-            return data
-
-        data_dict = myredis.Base.fetch_and_cache_data(redis_name, refresh, fetch_ranking, timer=expire_time_h * 3600)
-
-        ranking = OrderedDict(sorted(data_dict.items(), key=lambda item: item[1], reverse=True))
-        if top == 'all':
-            return ranking
-        else:
-            if isinstance(top, int):
-                return OrderedDict(list(ranking.items())[:top])
-            else:
-                raise ValueError("top 인자는 'all' 이나 int형 이어야 합니다.")
 
 @dataclass
 class LSTMData:
@@ -698,6 +678,12 @@ class MyLSTM:
         plt.show()"""
 
     def is_up(self)-> bool:
+        """
+        lstm 데이터가 증가하는 추세인지 확인후 참/거짓 반환
+
+        Returns:
+            bool: True if the data is strictly increasing, False otherwise.
+        """
         # 튜플의 [0]은 날짜 [1]은 값 배열
         data = self.get_final_predictions(refresh=False, expire_time_h=24)[1]
         # 데이터를 1D 배열로 변환

{analyser_hj3415-2.10.1 → analyser_hj3415-2.10.2}/pyproject.toml

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