analyser_hj3415 4.3.2__py3-none-any.whl → 4.4.0__py3-none-any.whl

analyser_hj3415/analyser/tsa/myprophet.py

@@ -1,6 +1,7 @@
 from collections import OrderedDict
-from datetime import datetime, timedelta, date
+import datetime
 from typing import Tuple, List, Dict, Union
+import time
 
 import yfinance as yf
 import pandas as pd
@@ -22,7 +23,7 @@ mylogger = setup_logger(__name__,'WARNING')
 class ProphetLatestData:
     ticker: str
 
-    date: date = field(metadata={"marshmallow_field": fields.Date()})
+    date: datetime.date = field(metadata={"marshmallow_field": fields.Date()})
     price: float
     yhat: float
     yhat_upper: float
@@ -96,7 +97,7 @@ class MyProphet:
         - 데이터를 Prophet 형식에 맞게 전처리합니다.
         - Prophet 모델을 사용하여 예측 데이터를 생성합니다.
         """
-        def get_raw_data() -> pd.DataFrame:
+        def get_raw_data(max_retries: int = 3, delay_sec: int = 2) -> pd.DataFrame:
             """
             Yahoo Finance에서 4년간의 주가 데이터를 가져옵니다.
 
@@ -104,16 +105,34 @@ class MyProphet:
                 pd.DataFrame: 가져온 주가 데이터프레임.
             """
             # 오늘 날짜 가져오기
-            today = datetime.today()
+            today = datetime.datetime.today()
 
             # 4년 전 날짜 계산 (4년 = 365일 * 4)
-            four_years_ago = today - timedelta(days=365 * 4)
-
-            return yf.download(
-                tickers=self.ticker,
-                start=four_years_ago.strftime('%Y-%m-%d'),
-                end=today.strftime('%Y-%m-%d')
-            )
+            four_years_ago = today - datetime.timedelta(days=365 * 4)
+
+            # 시도 횟수만큼 반복
+            for attempt in range(1, max_retries + 1):
+                try:
+                    data = yf.download(
+                        tickers=self.ticker,
+                        start=four_years_ago.strftime('%Y-%m-%d'),
+                        end=today.strftime('%Y-%m-%d')
+                    )
+                    # 데이터가 비어있지 않다면 성공으로 판단
+                    if not data.empty:
+                        return data
+                    else:
+                        # 다운로드 자체는 성공했지만, 빈 데이터가 반환될 경우
+                        print(f"[{attempt}/{max_retries}] 다운로드 결과가 비어 있습니다. 재시도합니다...")
+                except Exception as e:
+                    # 예외 발생 시, 재시도
+                    print(f"[{attempt}/{max_retries}] 다운로드 실패: {e}. 재시도합니다...")
+
+                # 재시도 전 대기
+                time.sleep(delay_sec)
+
+            mylogger.error(f"주가 데이터를 다운로드하지 못했습니다 (최대 {max_retries}회 시도 실패).")
+            return pd.DataFrame()
 
         def preprocessing_for_prophet() -> pd.DataFrame:
             """
@@ -246,10 +265,10 @@ class MyProphet:
             )
 
             data.trading_action, data.score = scoring(data.price, data.yhat_lower, data.yhat_upper)
-        except Exception:
+        except Exception as e:
             data = ProphetLatestData(
                 ticker=self.ticker,
-                date=datetime.now().date(),
+                date=datetime.datetime.now().date(),
                 price=float('nan'),
                 yhat=float('nan'),
                 yhat_lower=float('nan'),
@@ -341,7 +360,7 @@ class MyProphet:
         """
         try:
             # %Y-%m-%d 형식으로 문자열을 datetime 객체로 변환 시도
-            datetime.strptime(date_string, '%Y-%m-%d')
+            datetime.datetime.strptime(date_string, '%Y-%m-%d')
             return True
         except ValueError:
             # 변환이 실패하면 ValueError가 발생, 형식이 맞지 않음
@@ -457,3 +476,22 @@ class MIProphet(MyProphet):
         self._mi_type = mi_type
         self.ticker = getattr(MIs, mi_type)
 
+    @staticmethod
+    def ticker_to_mitype(ticker: str):
+        dict_fields = MIs._asdict()
+        reverse_map = {value: key for key, value in dict_fields.items()}
+        return reverse_map.get(ticker)
+
+    @staticmethod
+    def mitype_to_ticker(mi_type: str):
+        return getattr(MIs, mi_type)
+
+    @staticmethod
+    def bulk_get_latest_data(mi_types: List[str], refresh: bool) -> Dict[str, ProphetLatestData]:
+        ticker_data = MyProphet.bulk_get_latest_data([MIProphet.mitype_to_ticker(mi_type) for mi_type in mi_types],
+                                                     refresh=refresh)
+        mi_data = {}
+        for ticker, data in ticker_data.items():
+            mi_data[MIProphet.ticker_to_mitype(ticker)] = data
+        return mi_data
+

{analyser_hj3415-4.3.2.dist-info → analyser_hj3415-4.4.0.dist-info}/METADATA

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

{analyser_hj3415-4.3.2.dist-info → analyser_hj3415-4.4.0.dist-info}/RECORD

@@ -11,8 +11,8 @@ analyser_hj3415/analyser/eval/red.py,sha256=Abf5HPsNWKnDF4cbZ8xhlNsMF6ljN_sShCAY
 analyser_hj3415/analyser/tsa/__init__.py,sha256=7j-WshikzsDGGo_wuFoMPNmYfY-bLSEMd6o1MKSQKLI,150
 analyser_hj3415/analyser/tsa/common.py,sha256=iRwk88zBtEIqzSmTQWmQIWiiqZ9gN7GLtDUa5rx8IGM,1918
 analyser_hj3415/analyser/tsa/lstm.py,sha256=howngiVaEFKccYif6vrREm6skZiDzxazq44SdY_82yQ,28906
-analyser_hj3415/analyser/tsa/myprophet.py,sha256=QrqTt4hA870v2rJhVjpsspDUdo-aUU2bFfu0q3orIXk,17149
-analyser_hj3415-4.3.2.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
-analyser_hj3415-4.3.2.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
-analyser_hj3415-4.3.2.dist-info/METADATA,sha256=bQ2dBquvHcJvjxL0MoWF7vL1R27Vb15M8HNzwMAO2GQ,6811
-analyser_hj3415-4.3.2.dist-info/RECORD,,
+analyser_hj3415/analyser/tsa/myprophet.py,sha256=QHeIHinCZzMhuLpAqrMdy4E56MFbCGd89i9dR8cGXrw,18976
+analyser_hj3415-4.4.0.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
+analyser_hj3415-4.4.0.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
+analyser_hj3415-4.4.0.dist-info/METADATA,sha256=4m0sEx7lcK10xFNx7-C4gDVvWwWXlcJesac3n7rwPgA,6811
+analyser_hj3415-4.4.0.dist-info/RECORD,,