analyser_hj3415 2.7.6__tar.gz → 2.7.8__tar.gz

{analyser_hj3415-2.7.6 → analyser_hj3415-2.7.8}/PKG-INFO

@@ -1,12 +1,18 @@
 Metadata-Version: 2.1
 Name: analyser_hj3415
-Version: 2.7.6
+Version: 2.7.8
 Summary: Stock analyser and database processing programs
 Author-email: Hyungjin Kim <hj3415@gmail.com>
 Description-Content-Type: text/markdown
 Classifier: License :: OSI Approved :: MIT License
 Requires-Dist: utils-hj3415>=2.9.2
 Requires-Dist: db-hj3415>=4.0.3
+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
 Project-URL: Home, https://www.hyungjin.kr
 
 ### analyser-hj3415

{analyser_hj3415-2.7.6 → analyser_hj3415-2.7.8}/analyser_hj3415/cli.py

@@ -117,7 +117,7 @@ def analyser_manager():
                 for i, code in enumerate(myredis.Corps.list_all_codes()):
                     red.code = code
                     print(f"*** {i} / {red} ***")
-                    pprint.pprint(red.get(args.refresh))
+                    pprint.pprint(red.get(args.refresh, verbose=False))
 
                 # 원래 저장되었던 기대수익률로 다시 복원
                 eval.Red.expect_earn = ee_orig
@@ -140,10 +140,7 @@ def analyser_manager():
                 noti.telegram_to('manager', f"오늘의 Red({args.code})를 레디스캐시에 저장했습니다.(유효 12시간)")
 
         elif args.command == 'ranking':
-            if args.refresh:
-                result = eval.Red.ranking(expect_earn=args.expect_earn, refresh=True)
-            else:
-                result = eval.Red.ranking(expect_earn=args.expect_earn, refresh=False)
+            result = eval.Red.ranking(expect_earn=args.expect_earn, refresh=args.refresh)
             print(result)
             if args.noti:
                 noti.telegram_to('manager', "오늘의 red ranking을 레디스캐시에 저장했습니다.(유효 12시간)")
@@ -156,7 +153,7 @@ def analyser_manager():
                 for i, code in enumerate(myredis.Corps.list_all_codes()):
                     mil.code = code
                     print(f"*** {i} / {mil} ***")
-                    pprint.pprint(mil.get(args.refresh))
+                    pprint.pprint(mil.get(args.refresh, verbose=False))
             else:
                 assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
                 mil = eval.Mil(args.code)
@@ -173,7 +170,7 @@ def analyser_manager():
                 for i, code in enumerate(myredis.Corps.list_all_codes()):
                     blue.code = code
                     print(f"*** {i} / {blue} ***")
-                    pprint.pprint(blue.get(args.refresh))
+                    pprint.pprint(blue.get(args.refresh, verbose=False))
             else:
                 assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
                 blue = eval.Blue(args.code)
@@ -190,7 +187,7 @@ def analyser_manager():
                 for i, code in enumerate(myredis.Corps.list_all_codes()):
                     growth.code = code
                     print(f"*** {i} / {growth} ***")
-                    pprint.pprint(growth.get(args.refresh))
+                    pprint.pprint(growth.get(args.refresh, verbose=False))
             else:
                 assert utils.is_6digit(args.code), "code 인자는 6자리 숫자이어야 합니다."
                 growth = eval.Growth(args.code)

{analyser_hj3415-2.7.6 → analyser_hj3415-2.7.8}/analyser_hj3415/eval.py

@@ -324,7 +324,7 @@ class Red:
             score = score,
         )
 
-    def get(self, refresh = False) -> RedData:
+    def get(self, refresh = False, verbose = True) -> RedData:
         """
         RedData 형식의 데이터를 계산하여 리턴하고 레디스 캐시에 저장한다.
         :param refresh:
@@ -333,7 +333,8 @@ class Red:
         redis_name = f"{self.code}_red"
         analyser_logger.info(f"{self} RedData를 레디스캐시에서 가져오거나 새로 생성합니다.. refresh : {refresh}")
         expire_time = 3600 * 12
-        print(f"{self} redisname: '{redis_name}' / expect_earn: {Red.expect_earn} / refresh : {refresh} / expire_time : {expire_time/3600}h")
+        if verbose:
+            print(f"{self} redisname: '{redis_name}' / expect_earn: {Red.expect_earn} / refresh : {refresh} / expire_time : {expire_time/3600}h")
 
         def fetch_generate_data(refresh_in: bool) -> dict:
             return asdict(self._generate_data(refresh_in))
@@ -375,7 +376,7 @@ class Red:
             red = Red('005930')
             for i, code in enumerate(myredis.Corps.list_all_codes()):
                 red.code = code
-                red_score = red.get(refresh=refresh_in).score
+                red_score = red.get(refresh=refresh_in, verbose=False).score
                 if red_score > 0:
                     data[code] = red_score
                     print(f"{i}: {red} - {red_score}")
@@ -632,7 +633,7 @@ class Mil:
             date = date_list,
         )
 
-    def get(self, refresh = False) -> MilData:
+    def get(self, refresh = False, verbose = True) -> MilData:
         """
         MilData 형식의 데이터를 계산하여 리턴하고 레디스 캐시에 저장한다.
         :param refresh:
@@ -640,7 +641,8 @@ class Mil:
         """
         redis_name = f"{self.code}_mil"
         analyser_logger.info(f"{self} MilData를 레디스캐시에서 가져오거나 새로 생성합니다.. refresh : {refresh}")
-        print(f"{self} redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time/3600}h")
+        if verbose:
+            print(f"{self} redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time/3600}h")
 
         def fetch_generate_data(refresh_in: bool) -> dict:
             return asdict(self._generate_data(refresh_in))
@@ -804,7 +806,7 @@ class Blue:
             date= date_list,
         )
 
-    def get(self, refresh = False) -> BlueData:
+    def get(self, refresh = False, verbose = True) -> BlueData:
         """
         BlueData 형식의 데이터를 계산하여 리턴하고 레디스 캐시에 저장한다.
         :param refresh:
@@ -812,7 +814,8 @@ class Blue:
         """
         redis_name = f"{self.code}_blue"
         analyser_logger.info(f"{self} BlueData를 레디스캐시에서 가져오거나 새로 생성합니다.. refresh : {refresh}")
-        print(f"{self} redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time/3600}h")
+        if verbose:
+            print(f"{self} redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time/3600}h")
 
         def fetch_generate_data(refresh_in: bool) -> dict:
             return asdict(self._generate_data(refresh_in))
@@ -903,7 +906,7 @@ class Growth:
             date= date_list,
         )
 
-    def get(self, refresh = False) -> GrowthData:
+    def get(self, refresh = False, verbose = True) -> GrowthData:
         """
         GrowthData 형식의 데이터를 계산하여 리턴하고 레디스 캐시에 저장한다.
         :param refresh:
@@ -911,7 +914,8 @@ class Growth:
         """
         redis_name = f"{self.code}_growth"
         analyser_logger.info(f"{self} GrowthData를 레디스캐시에서 가져오거나 새로 생성합니다.. refresh : {refresh}")
-        print(f"{self} redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time/3600}h")
+        if verbose:
+            print(f"{self} redisname: '{redis_name}' / refresh : {refresh} / expire_time : {expire_time/3600}h")
 
         def fetch_generate_data(refresh_in: bool) -> dict:
             return asdict(self._generate_data(refresh_in))

analyser_hj3415-2.7.8/analyser_hj3415/myprophet.py

@@ -0,0 +1,144 @@
+import yfinance as yf
+from datetime import datetime, timedelta
+import pandas as pd
+from prophet import Prophet
+from sklearn.preprocessing import StandardScaler
+from utils_hj3415 import utils
+from typing import Optional
+import plotly.graph_objs as go
+from plotly.offline import plot
+import matplotlib.pyplot as plt  # Matplotlib 수동 임포트
+from db_hj3415 import myredis
+
+class MyProphet:
+    def __init__(self, code: str):
+        assert utils.is_6digit(code), f'Invalid value : {code}'
+        self._code = code
+        self.name = myredis.Corps(code, 'c101').get_name()
+        self.raw_data = self.get_raw_data()
+
+        self.scaler = StandardScaler()
+        self.model = Prophet()
+
+    @property
+    def code(self) -> str:
+        return self._code
+
+    @code.setter
+    def code(self, code: str):
+        assert utils.is_6digit(code), f'Invalid value : {code}'
+        self._code = code
+        self.name = myredis.Corps(code, 'c101').get_name()
+        self.raw_data = self.get_raw_data()
+
+    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의 형식에 맞게 열 이름 변경
+
+        # 추가 변수를 정규화
+        df['volume_scaled'] = self.scaler.fit_transform(df[['volume']])
+        return df
+
+    def make_forecast(self) -> pd.DataFrame:
+        # Prophet을 위한 dataframe 만들기
+        df_for_prophet = self.preprocessing_for_prophet()
+
+        # 정규화된 'volume_scaled' 변수를 외부 변수로 추가
+        self.model.add_regressor('volume_scaled')
+
+        self.model.fit(df_for_prophet)
+
+        # 향후 180일 동안의 주가 예측
+        future = self.model.make_future_dataframe(periods=180)
+
+        # 미래 데이터에 거래량 추가 (평균 거래량을 사용해 정규화)
+        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)
+        return forecast
+
+    def export_to(self, to="str") -> Optional[str]:
+        """
+        prophet과 plotly로 그래프를 그려서 html을 문자열로 반환
+        :param to: str, png, htmlfile, show
+        :return:
+        """
+        # 실제 데이터
+        df = self.preprocessing_for_prophet()
+        # 예측 데이터
+        forecast = self.make_forecast()
+
+        # Plotly를 사용한 시각화
+        fig = go.Figure()
+
+        # 실제 데이터
+        fig.add_trace(go.Scatter(x=df['ds'], y=df['y'], mode='markers', name='실제주가'))
+        # 예측 데이터
+        fig.add_trace(go.Scatter(x=forecast['ds'], y=forecast['yhat'], mode='lines', name='예측치'))
+
+        # 상한/하한 구간
+        fig.add_trace(
+            go.Scatter(x=forecast['ds'], y=forecast['yhat_upper'], fill=None, mode='lines', name='상한'))
+        fig.add_trace(
+            go.Scatter(x=forecast['ds'], y=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",  # 소수점 없이 원래 숫자 표시
+            )
+        )
+
+        if to == 'str':
+            # 그래프 HTML로 변환 (string 형식으로 저장)
+            graph_html = plot(fig, output_type='div')
+            return graph_html
+        elif to == 'png':
+            # 그래프를 PNG 파일로 저장
+            fig.write_image("plotly_graph.png")
+        elif to == 'htmlfile':
+            # 그래프를 HTML로 저장
+            plot(fig, filename='graph_plotly.html', auto_open=False)
+            return None
+        elif to == 'show':
+            # 예측 결과 출력
+            print(forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail())
+            # 예측 결과 시각화 (Matplotlib 사용)
+            fig = self.model.plot(forecast)
+            # 추세 및 계절성 시각화
+            fig2 = self.model.plot_components(forecast)
+            plt.show()  # 시각화 창 띄우기
+        else:
+            Exception("to 인자가 맞지 않습니다.")
+
+
+
+

{analyser_hj3415-2.7.6 → analyser_hj3415-2.7.8}/analyser_hj3415/workroom/lstm.py

@@ -9,11 +9,11 @@ import matplotlib.pyplot as plt
 # 1. 데이터 다운로드 (애플 주식 데이터를 사용)
 #stock_data = yf.download('AAPL', start='2020-01-01', end='2023-01-01')
 # 삼성전자 주식 데이터 가져오기 (KOSPI 상장)
-#stock_data = yf.download('005930.KS', start='2019-01-01', end='2024-10-10')
+#stock_data = yf.download('005930.KS', start='2019-01-01', end='2024-10-11')
 # 크래프톤 주식 데이터 가져오기 (KOSPI 상장)
-#stock_data = yf.download('259960.KS', start='2020-01-01', end='2024-10-08')
+stock_data = yf.download('259960.KS', start='2020-01-01', end='2024-10-11')
 # 하이닉스 주식 데이터 가져오기 (KOSPI 상장)
-stock_data = yf.download('000660.KS', start='2019-01-01', end='2024-10-10')
+#stock_data = yf.download('000660.KS', start='2019-01-01', end='2024-10-11')
 
 
 # 2. 필요한 열만 선택 (종가만 사용)

{analyser_hj3415-2.7.6 → analyser_hj3415-2.7.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "analyser_hj3415"
-version = "2.7.6"
+version = "2.7.8"
 authors = [{name = "Hyungjin Kim", email = "hj3415@gmail.com"}]
 description = "Stock analyser and database processing programs"
 readme = "README.md"
@@ -13,6 +13,12 @@ classifiers = ["License :: OSI Approved :: MIT License"]
 dependencies = [
     "utils-hj3415>=2.9.2",
     "db-hj3415>=4.0.3",
+    "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",
 ]
 
 [project.scripts]

analyser_hj3415-2.7.6/analyser_hj3415/workroom/myprophet.py

@@ -1,58 +0,0 @@
-# prophet 사용 공부
-
-import yfinance as yf
-import pandas as pd
-from prophet import Prophet
-import matplotlib.pyplot as plt  # Matplotlib 수동 임포트
-
-
-# 애플 주식 데이터를 다운로드
-#stock_data = yf.download('AAPL', start='2020-01-01', end='2024-10-09')
-# 삼성전자 주식 데이터 가져오기 (KOSPI 상장)
-#stock_data = yf.download('005930.KS', start='2020-01-01', end='2024-08-01')
-# 크래프톤 주식 데이터 가져오기 (KOSPI 상장)
-#stock_data = yf.download('259960.KS', start='2020-01-01', end='2024-10-08')
-# 하이닉스 주식 데이터 가져오기 (KOSPI 상장)
-stock_data = yf.download('000660.KS', start='2020-01-01', end='2024-10-08')
-
-
-# Prophet이 사용할 수 있도록 데이터 준비
-df = stock_data[['Close', 'Volume']].reset_index()
-df.columns = ['ds', 'y', 'volume']  # Prophet의 형식에 맞게 열 이름 변경
-
-from sklearn.preprocessing import StandardScaler
-
-# 추가 변수를 정규화
-scaler = StandardScaler()
-df['volume_scaled'] = scaler.fit_transform(df[['volume']])
-
-# Prophet 모델 생성 및 학습
-model = Prophet()
-
-# 정규화된 'volume_scaled' 변수를 외부 변수로 추가
-model.add_regressor('volume_scaled')
-
-model.fit(df)
-
-# 향후 180일 동안의 주가 예측
-future = model.make_future_dataframe(periods=180)
-
-# 미래 데이터에 거래량 추가 (평균 거래량을 사용해 정규화)
-future_volume = pd.DataFrame({'volume': [stock_data['Volume'].mean()] * len(future)})
-future['volume_scaled'] = scaler.transform(future_volume[['volume']])
-
-
-forecast = model.predict(future)
-
-
-
-# 예측 결과 출력
-print(forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail())
-
-# 예측 결과 시각화 (Matplotlib 사용)
-fig = model.plot(forecast)
-
-# 추세 및 계절성 시각화
-fig2 = model.plot_components(forecast)
-
-plt.show()  # 시각화 창 띄우기