analyser_hj3415 2.10.6__py3-none-any.whl → 3.0.1__py3-none-any.whl

analyser_hj3415/analyser/tsa/prophet.py

@@ -0,0 +1,207 @@
+from datetime import datetime, timedelta
+from typing import Optional
+import yfinance as yf
+import pandas as pd
+from prophet import Prophet
+from sklearn.preprocessing import StandardScaler
+import matplotlib.pyplot as plt  # Matplotlib 수동 임포트
+import plotly.graph_objs as go
+from plotly.offline import plot
+
+from utils_hj3415 import tools, setup_logger
+from db_hj3415 import myredis
+
+from analyser_hj3415.analyser import eval
+
+mylogger = setup_logger(__name__,'WARNING')
+
+
+class MyProphet:
+    def __init__(self, code: str):
+        assert tools.is_6digit(code), f'Invalid value : {code}'
+        self.scaler = StandardScaler()
+
+        self.model = Prophet()
+        self._code = code
+        self.name = myredis.Corps(code, 'c101').get_name()
+        self.raw_data = self._get_raw_data()
+        self.df_real = self._preprocessing_for_prophet()
+        self.df_forecast = self._make_forecast()
+
+    @property
+    def code(self) -> str:
+        return self._code
+
+    @code.setter
+    def code(self, code: str):
+        assert tools.is_6digit(code), f'Invalid value : {code}'
+        mylogger.info(f'change code : {self.code} -> {code}')
+        self.model = Prophet()
+        self._code = code
+        self.name = myredis.Corps(code, 'c101').get_name()
+        self.raw_data = self._get_raw_data()
+        self.df_real = self._preprocessing_for_prophet()
+        self.df_forecast = self._make_forecast()
+
+    @staticmethod
+    def is_valid_date(date_string):
+        try:
+            # %Y-%m-%d 형식으로 문자열을 datetime 객체로 변환 시도
+            datetime.strptime(date_string, '%Y-%m-%d')
+            return True
+        except ValueError:
+            # 변환이 실패하면 ValueError가 발생, 형식이 맞지 않음
+            return False
+
+    def _get_raw_data(self) -> pd.DataFrame:
+        """
+        야후에서 해당 종목의 4년간 주가 raw data를 받아온다.
+        :return:
+        """
+        # 오늘 날짜 가져오기
+        today = datetime.today()
+
+        # 4년 전 날짜 계산 (4년 = 365일 * 4)
+        four_years_ago = today - timedelta(days=365 * 4)
+
+        return yf.download(
+            self.code + '.KS',
+            start=four_years_ago.strftime('%Y-%m-%d'),
+            end=today.strftime('%Y-%m-%d')
+        )
+
+    def _preprocessing_for_prophet(self) -> pd.DataFrame:
+        """
+        Prophet이 사용할 수 있도록 데이터 준비
+        ds는 날짜, y는 주가
+        :return:
+        """
+        df = self.raw_data[['Close', 'Volume']].reset_index()
+        df.columns = ['ds', 'y', 'volume']  # Prophet의 형식에 맞게 열 이름 변경
+
+        # ds 열에서 타임존 제거
+        df['ds'] = df['ds'].dt.tz_localize(None)
+
+        # 추가 변수를 정규화
+        df['volume_scaled'] = self.scaler.fit_transform(df[['volume']])
+        mylogger.debug('_preprocessing_for_prophet')
+        mylogger.debug(df)
+        return df
+
+    def _make_forecast(self) -> pd.DataFrame:
+        # 정규화된 'volume_scaled' 변수를 외부 변수로 추가
+        self.model.add_regressor('volume_scaled')
+
+        self.model.fit(self.df_real)
+
+        # 향후 180일 동안의 주가 예측
+        future = self.model.make_future_dataframe(periods=180)
+        mylogger.debug('_make_forecast_future')
+        mylogger.debug(future)
+
+        # 미래 데이터에 거래량 추가 (평균 거래량을 사용해 정규화)
+        future_volume = pd.DataFrame({'volume': [self.raw_data['Volume'].mean()] * len(future)})
+        future['volume_scaled'] = self.scaler.transform(future_volume[['volume']])
+
+        forecast = self.model.predict(future)
+        mylogger.debug('_make_forecast')
+        mylogger.debug(forecast)
+        return forecast
+
+    def get_yhat(self) -> dict:
+        """
+        최근 날짜의 예측데이터를 반환한다.
+        :return: {'ds':..., 'yhat':.., 'yhat_lower':.., 'yhat_upper':..,}
+        """
+        df = self.df_forecast
+        last_real_date = self.df_real.iloc[-1]['ds']
+        mylogger.info(last_real_date)
+        yhat_dict = df[df['ds']==last_real_date].iloc[0][['ds', 'yhat_lower', 'yhat_upper', 'yhat']].to_dict()
+        mylogger.info(yhat_dict)
+        return yhat_dict
+
+    def visualization(self):
+        # 예측 결과 출력
+        print(self.df_forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail())
+        # 예측 결과 시각화 (Matplotlib 사용)
+        fig = self.model.plot(self.df_forecast)
+        # 추세 및 계절성 시각화
+        fig2 = self.model.plot_components(self.df_forecast)
+        plt.show()  # 시각화 창 띄우기
+
+    def export(self, to="str") -> Optional[str]:
+        """
+        prophet과 plotly로 그래프를 그려서 html을 문자열로 반환
+        :param to: str, png, htmlfile
+        :return:
+        """
+        # Plotly를 사용한 시각화
+        fig = go.Figure()
+
+        # 실제 데이터
+        fig.add_trace(go.Scatter(x=self.df_real['ds'], y=self.df_real['y'], mode='markers', name='실제주가'))
+        # 예측 데이터
+        fig.add_trace(go.Scatter(x=self.df_forecast['ds'], y=self.df_forecast['yhat'], mode='lines', name='예측치'))
+
+        # 상한/하한 구간
+        fig.add_trace(
+            go.Scatter(x=self.df_forecast['ds'], y=self.df_forecast['yhat_upper'], fill=None, mode='lines', name='상한'))
+        fig.add_trace(
+            go.Scatter(x=self.df_forecast['ds'], y=self.df_forecast['yhat_lower'], fill='tonexty', mode='lines', name='하한'))
+
+        fig.update_layout(
+            # title=f'{self.code} {self.name} 주가 예측 그래프(prophet)',
+            xaxis_title='일자',
+            yaxis_title='주가(원)',
+            xaxis = dict(
+                tickformat='%Y/%m',  # X축을 '연/월' 형식으로 표시
+            ),
+            yaxis = dict(
+                tickformat=".0f",  # 소수점 없이 원래 숫자 표시
+            ),
+            showlegend=False,
+        )
+
+        if to == 'str':
+            # 그래프 HTML로 변환 (string 형식으로 저장)
+            graph_html = plot(fig, output_type='div')
+            return graph_html
+        elif to == 'png':
+            # 그래프를 PNG 파일로 저장
+            fig.write_image(f"myprophet_{self.code}.png")
+            return None
+        elif to == 'htmlfile':
+            # 그래프를 HTML로 저장
+            plot(fig, filename=f'myprophet_{self.code}.html', auto_open=False)
+            return None
+        else:
+            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.
+
+        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')
+        yhat_dict = self.get_yhat()
+        mylogger.info(f'recent_price: {recent_price}, yhat_dict: {yhat_dict}')
+        yhat_lower = int(yhat_dict['yhat_lower'])
+        deviation = int(eval.Tools.cal_deviation(recent_price, yhat_lower))
+        if recent_price > yhat_lower:
+            score = -deviation
+        else:
+            score = deviation
+        mylogger.info(f"{self.code}/{self.name} date: {recent_date} 가격:{recent_price} 기대하한값:{yhat_lower} 편차:{deviation} score:{score}")
+        return score
+

analyser_hj3415/cli.py

@@ -1,42 +1,12 @@
 import argparse
-import os
 import pprint
 
-from utils_hj3415 import utils
-from utils_hj3415.helpers import SettingsManager
+from utils_hj3415 import tools
+from analyser_hj3415.analyser import eval, tsa, score
 from db_hj3415 import myredis, mymongo
 
-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
-        self.save_settings()
-        print(f"{title}: {value}가 저장되었습니다")
-
-    def get_value(self, title: str):
-        assert title in self.TITLES, f"title 인자는 {self.TITLES} 중에 있어야 합니다."
-        return self.settings_dict.get(title, self.DEFAULT_SETTINGS[title])
-
-    def reset_value(self, title: str):
-        assert title in self.TITLES, f"title 인자는 {self.TITLES} 중에 있어야 합니다."
-        self.set_value(title, self.DEFAULT_SETTINGS[title])
-        print(f"{title}이 기본값 ({self.DEFAULT_SETTINGS[title]}) 으로 초기화 되었습니다.")
-
 
 def analyser_manager():
-    settings_manager = AnalyserSettingsManager()
-    expect_earn_from_setting = settings_manager.get_value('EXPECT_EARN')
-
     parser = argparse.ArgumentParser(description="Analyser Commands")
     type_subparsers = parser.add_subparsers(dest='type', help='분석 타입')
 
@@ -78,7 +48,6 @@ def analyser_manager():
     # mil - get 파서
     mil_get_parser = mil_subparser.add_parser('get', help='mil get 책정 및 레디스 저장')
     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='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
 
     # blue 명령어 서브파서
@@ -87,7 +56,6 @@ def analyser_manager():
     # blue - get 파서
     blue_get_parser = blue_subparser.add_parser('get', help='blue get 책정 및 레디스 저장')
     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='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
 
     # growth 명령어 서브파서
@@ -96,64 +64,37 @@ def analyser_manager():
     # growth - get 파서
     growth_get_parser = growth_subparser.add_parser('get', help='growth get 책정 및 레디스 저장')
     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='래디스 캐시를 사용하지 않고 강제로 재계산 할지')
 
-    # setting 명령어 서브파서
-    setting_parser = type_subparsers.add_parser('setting', help='Set and Get settings')
-    setting_subparser = setting_parser.add_subparsers(dest='command', help='setting 관련된 명령')
-    # setting - set 파서
-    set_parser = setting_subparser.add_parser('set', help='세팅값 저장')
-    set_parser.add_argument('title', choices=AnalyserSettingsManager.TITLES, help='타이틀')
-    set_parser.add_argument('value', help='세팅값')
-    # setting - get 파서
-    get_parser = setting_subparser.add_parser('get', help='타이틀 세팅값 불러오기')
-    get_parser.add_argument('title', choices=AnalyserSettingsManager.TITLES, help='타이틀')
-    # setting - print 파서
-    setting_subparser.add_parser('print', help='전체 세팅값 출력')
-
     args = parser.parse_args()
 
-    from . import eval, tsa, score
-
     if args.type == 'red':
         if args.command == 'get':
-            if args.code == 'all':
-                # 저장된 기대수익률을 불러서 임시저장
-                ee_orig = eval.Red.expect_earn
-
+            if args.expect_earn is None:
                 red = eval.Red('005930')
-                if args.expect_earn:
-                    eval.Red.expect_earn = float(args.expect_earn)
+            else:
+                red = eval.Red('005930', expect_earn=args.expect_earn)
+            if args.code == 'all':
 
                 print("**** Red - all codes ****")
                 for i, code in enumerate(myredis.Corps.list_all_codes()):
                     red.code = code
                     print(f"*** {i} / {red} ***")
                     pprint.pprint(red.get(args.refresh, verbose=False))
-
-                # 원래 저장되었던 기대수익률로 다시 복원
-                eval.Red.expect_earn = ee_orig
             else:
-                assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
+                assert tools.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
                 # 저장된 기대수익률을 불러서 임시저장
-                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)
-
+                red.code = args.code
                 print(f"*** Red - {red} ***")
                 pprint.pprint(red.get(args.refresh))
-
-                # 원래 저장되었던 기대수익률로 다시 복원
-                eval.Red.expect_earn = ee_orig
             mymongo.Logs.save('cli','INFO', f'run >> analyser red get {args.code}')
 
         elif args.command == 'ranking':
             mymongo.Logs.save('cli','INFO', 'run >> analyser red ranking')
-            result = eval.Red.ranking(expect_earn=args.expect_earn, refresh=args.refresh)
+            if args.expect_earn is None:
+                result = eval.Red.ranking(refresh=args.refresh)
+            else:
+                result = eval.Red.ranking(expect_earn=args.expect_earn, refresh=args.refresh)
             print(result)
 
     elif args.type == 'mil':
@@ -166,7 +107,7 @@ def analyser_manager():
                     print(f"*** {i} / {mil} ***")
                     pprint.pprint(mil.get(args.refresh, verbose=False))
             else:
-                assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
+                assert tools.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
                 mil = eval.Mil(args.code)
                 print(f"*** Mil - {mil} ***")
                 pprint.pprint(mil.get(args.refresh))
@@ -182,7 +123,7 @@ def analyser_manager():
                     print(f"*** {i} / {blue} ***")
                     pprint.pprint(blue.get(args.refresh, verbose=False))
             else:
-                assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
+                assert tools.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
                 blue = eval.Blue(args.code)
                 print(f"*** Blue - {blue} ***")
                 pprint.pprint(blue.get(args.refresh))
@@ -198,7 +139,7 @@ def analyser_manager():
                     print(f"*** {i} / {growth} ***")
                     pprint.pprint(growth.get(args.refresh, verbose=False))
             else:
-                assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
+                assert tools.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
                 growth = eval.Growth(args.code)
                 print(f"*** growth - {growth} ***")
                 pprint.pprint(growth.get(args.refresh))
@@ -206,7 +147,7 @@ def analyser_manager():
 
     elif args.type == 'prophet':
         if args.command == 'ranking':
-            result = score.Score.ranking(refresh=args.refresh, expire_time_h=48)
+            result = score.Score.ranking(refresh=args.refresh)
             print(result)
             mymongo.Logs.save('cli','INFO', 'run >> analyser prophet ranking')
 
@@ -214,23 +155,13 @@ def analyser_manager():
         mylstm = tsa.MyLSTM
         if args.command == 'caching':
             if args.top:
-                mylstm.caching_based_on_prophet_ranking(refresh=args.refresh, expire_time_h=96, top=args.top)
+                mylstm.caching_based_on_prophet_ranking(refresh=args.refresh, top=args.top)
             else:
-                mylstm.caching_based_on_prophet_ranking(refresh=args.refresh, expire_time_h=96)
+                mylstm.caching_based_on_prophet_ranking(refresh=args.refresh)
             mymongo.Logs.save('cli','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, expire_time_h=96)
+            assert tools.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
+            result = mylstm(args.code).get_final_predictions(refresh=args.refresh)
             mymongo.Logs.save('cli','INFO', f'run >> analyser lstm get {args.code}')
-
-    elif args.type == 'setting':
-        if args.command == 'set':
-            settings_manager.set_value(args.title, args.value)
-        elif args.command == 'get':
-            value = settings_manager.get_value(args.title)
-            print(f"{args.title} 값: {value}")
-        elif args.command == 'print':
-            print(settings_manager.load_settings())
-
     else:
         parser.print_help()

{analyser_hj3415-2.10.6.dist-info → analyser_hj3415-3.0.1.dist-info}/METADATA

@@ -1,11 +1,11 @@
 Metadata-Version: 2.3
 Name: analyser_hj3415
-Version: 2.10.6
+Version: 3.0.1
 Summary: Stock analyser and database processing programs
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
-Requires-Dist: utils-hj3415>=2.9.4
-Requires-Dist: db-hj3415>=4.2.3
+Requires-Dist: utils-hj3415>=3.0.8
+Requires-Dist: db-hj3415>=4.3.0
 Requires-Dist: scikit-learn>=1.5.2
 Requires-Dist: plotly>=5.24.1
 Requires-Dist: yfinance>=0.2.44

analyser_hj3415-3.0.1.dist-info/RECORD

@@ -0,0 +1,22 @@
+analyser_hj3415/__init__.py,sha256=huF0tm8fFC96sODN9TmVRmPhDizQ0yBTKNOw8miOBh0,448
+analyser_hj3415/cli.py,sha256=b1oK0_mt248eCQuavKh6R4irgtA9XxkSUFfAgLf8hAo,9204
+analyser_hj3415/analyser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+analyser_hj3415/analyser/score.py,sha256=cqTHXLBh38N9zlmNzChWZJ42O7gdXhjw_jVevOxDUo8,6420
+analyser_hj3415/analyser/eval/__init__.py,sha256=IP1d0Q3nOCAD3zK1qxrC685MkJQfUh-qaXc7xptTxk8,80
+analyser_hj3415/analyser/eval/blue.py,sha256=fq_eln7-EdwICNQ2sMXAfZKXGhQ29EaJwsGvn7xA29U,7632
+analyser_hj3415/analyser/eval/common.py,sha256=JBTVvzO6YoeMjKl6X2MF7wgB6fvPTEToHLJSv-V7V4k,12425
+analyser_hj3415/analyser/eval/growth.py,sha256=OXuBuPzVw3q6kpelNigiX7xvJNUP1gv26DNetc4aews,3315
+analyser_hj3415/analyser/eval/mil.py,sha256=wN1jPu5zLJjoDk3kL7piL-4jWhpsAMeQpsTZll0a2dw,10568
+analyser_hj3415/analyser/eval/red.py,sha256=SegWvklGFCCFpzg9qwKDEU0-iSq-d_lM6XRNhth8VWM,12529
+analyser_hj3415/analyser/tsa/__init__.py,sha256=rPnn4b8aEym2O9l_sAaIRUpOeXJw6sDj2XfUzy-mpvg,42
+analyser_hj3415/analyser/tsa/lstm.py,sha256=jrJlugPTugSH9liNGpiRMaigr3mq3FeGxDNVvcll3N8,28302
+analyser_hj3415/analyser/tsa/prophet.py,sha256=zL6RNSiwFKKliQVsR3yOHz13S2SrCfuXvBj62lqP80Y,7984
+analyser_hj3415/workroom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+analyser_hj3415/workroom/mysklearn.py,sha256=wJXKz5MqqTzADdG2mqRMMzc_G9RzwYjj5_j4gyOopxQ,2030
+analyser_hj3415/workroom/mysklearn2.py,sha256=1lIy6EWEQHkOzDS-av8U0zQH6DuCLKWMI73dnJx5KRs,1495
+analyser_hj3415/workroom/score.py,sha256=P6nHBJYmyhigGtT4qna4BmNtvt4B93b7SKyzdstJK24,17376
+analyser_hj3415/workroom/trash.py,sha256=zF-W0piqkGr66UP6-iybo9EXh2gO0RP6R1FnIpsGkl8,12262
+analyser_hj3415-3.0.1.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
+analyser_hj3415-3.0.1.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
+analyser_hj3415-3.0.1.dist-info/METADATA,sha256=a1hvHhQURatp_A7dL52s5HKQ2KU8Ok6Q_-2AEHsgldQ,6776
+analyser_hj3415-3.0.1.dist-info/RECORD,,