dsyoon/DeepStock
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init

Browse Source
This commit is contained in:
dsyoon
2023-02-27 13:18:30 +09:00
parent 9b3db7a695
commit 51f8d5a0f2
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
stock/crawler/StockCrawler.py
View File

@@ -254,7 +254,6 @@ class StockCrawler:
'GOOG': 'Alphabet C', 'GOOG': 'Alphabet C',
'AMZN': 'Amazon.com', 'AMZN': 'Amazon.com',
'AVGO': 'Broadcom', 'AVGO': 'Broadcom',
'FB': 'Meta Platforms',
'NVDA': 'NVIDIA', 'NVDA': 'NVIDIA',
'UNH': 'UnitedHealth', 'UNH': 'UnitedHealth',
'TSM': 'Taiwan Semiconductor', 'TSM': 'Taiwan Semiconductor',
@@ -310,6 +309,8 @@ class StockCrawler:
yfin.pdr_override() yfin.pdr_override()
data = pdr.get_data_yahoo(ticker, start_day, end_day, auto_adjust=True) 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] = { us_sotck_data[ticker] = {
'close': data['Close'].to_dict(), 'close': data['Close'].to_dict(),
@@ -322,7 +323,8 @@ class StockCrawler:
cursor.close() cursor.close()
conn.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): for idx, item_code in enumerate(us_sotck_data):
stock_data = [] stock_data = []
stock = us_sotck_data[item_code] stock = us_sotck_data[item_code]
@@ -339,7 +341,7 @@ class StockCrawler:
continue continue
stock_data.append({ 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), '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] 'high': round(stock['high'][ymd], 2), 'low': round(stock['low'][ymd], 2), 'volume': stock['volume'][ymd]
}) })
@@ -545,8 +547,9 @@ class StockCrawler:
return return
if __name__ == "__main__": 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' stockFileName = PROJECT_HOME + '/resources/stock.db'
stockCrawler.crawl_us_sotck_datas(stockFileName) stockCrawler.crawl_special_stocks(stockFileName)