diff --git a/stock/crawler/StockCrawler.py b/stock/crawler/StockCrawler.py index 851f27f..4d366e9 100644 --- a/stock/crawler/StockCrawler.py +++ b/stock/crawler/StockCrawler.py @@ -254,7 +254,6 @@ class StockCrawler: 'GOOG': 'Alphabet C', 'AMZN': 'Amazon.com', 'AVGO': 'Broadcom', - 'FB': 'Meta Platforms', 'NVDA': 'NVIDIA', 'UNH': 'UnitedHealth', 'TSM': 'Taiwan Semiconductor', @@ -310,6 +309,8 @@ class StockCrawler: yfin.pdr_override() data = pdr.get_data_yahoo(ticker, start_day, end_day, auto_adjust=True) + data['datetime'] = data.index.strftime("%Y.%m.%d") + data.set_index('datetime', inplace=True) us_sotck_data[ticker] = { 'close': data['Close'].to_dict(), @@ -322,7 +323,8 @@ class StockCrawler: cursor.close() conn.close() - dateList = list(us_sotck_data['SQQQ']['close']) + dateList = list(us_sotck_data['^KS11']['close']) + #dateList = [temp.strftime("%Y-%m-%d") for temp in dateList] for idx, item_code in enumerate(us_sotck_data): stock_data = [] stock = us_sotck_data[item_code] @@ -339,7 +341,7 @@ class StockCrawler: continue stock_data.append({ - 'CODE':item_code, 'NAME':special_stocks[item_code], 'ymd': ymd.strftime('%Y.%m.%d'), + 'CODE':item_code, 'NAME':special_stocks[item_code], 'ymd': ymd, 'close': round(stock['close'][ymd], 2), 'diff': round(diff, 2), 'open': round(stock['open'][ymd], 2), 'high': round(stock['high'][ymd], 2), 'low': round(stock['low'][ymd], 2), 'volume': stock['volume'][ymd] }) @@ -545,8 +547,9 @@ class StockCrawler: return if __name__ == "__main__": - stockCrawler = StockCrawler() + START_DATE = "2000.01.01" + stockCrawler = StockCrawler(START_DATE) - PROJECT_HOME = os.path.join(os.path.dirname(os.path.join(os.path.dirname(os.path.join(os.path.dirname(os.path.join(os.path.dirname(__file__)))))))) + PROJECT_HOME = os.path.join(os.path.dirname(os.path.join(os.path.dirname(os.path.join(os.path.dirname(__file__)))))) stockFileName = PROJECT_HOME + '/resources/stock.db' - stockCrawler.crawl_us_sotck_datas(stockFileName) + stockCrawler.crawl_special_stocks(stockFileName)