Feat : Fixed time error

Author: NE7K

upstock/config.py

@@ -17,15 +17,16 @@
 load_dotenv()
 
 logging.basicConfig(
-    level=logging.INFO,
+    level=logging.INFO, # info 이상 log print
     format='%(asctime)s [%(levelname)s] %(name)s - %(message)s',
-    stream=sys.stdout
+    stream=sys.stdout   # import sys 터미널에 로그 출력
 )
 
 @dataclass(frozen=True) # 불변
 class SupabaseConfig:
     url: str = os.getenv('SUPABASE_URL')
     key: str = os.getenv('SUPABASE_KEY')
+    bucket_name: str = 'sentiment_file' # download model file
 
     @property
     def client(self) -> Client:

upstock/nodes/predict.py

@@ -47,11 +47,12 @@ def run_predict():
     # print(news_df['Date'].head(10))
     today = datetime.date.today() # today
 
-    # string to date time
-    news_df['parsed_date'] = pd.to_datetime(
-        news_df['Date'],
-        errors='coerce'
+    news_df["parsed_date"] = pd.to_datetime(
+        today.strftime("%Y-%m-%d ") + news_df["Date"],
+        format="%Y-%m-%d %I:%M%p",
+        errors="coerce"
     )
+    
     today_news = news_df[news_df['parsed_date'].dt.date == today] # today == parse data date
     predict_texts = today_news['Title'].tolist() # insert pare data
 

upstock/storage/downloader.py

@@ -2,18 +2,16 @@
 from Supabase storage 
 """
 
-
 import os
 import time
 import logging
 
-from upstock.config import supabase
+from upstock.config import supabase, SupabaseConfig
 
 logger = logging.getLogger(__name__)
 
 # supabase storage
 def download_model_file():
-    bucket_name = 'sentiment_file'
 
     file_paths = [
         'upstock_sentiment_model.keras',
@@ -22,7 +20,7 @@ def download_model_file():
     ]
 
     os.makedirs('SaveModel', exist_ok=True) # exist no error
-    bucket = supabase.storage.from_(bucket_name)
+    bucket = supabase.storage.from_(SupabaseConfig.bucket_name)
 
     for file_path in file_paths: