From 2b64792ddf74c31498e4b86f956129fe1a15698f Mon Sep 17 00:00:00 2001 From: "dosang.yoon" Date: Sat, 9 Jul 2022 22:23:40 +0900 Subject: [PATCH] init --- hts/util/DataDownloader.py | 23 ++++++----------------- 1 file changed, 6 insertions(+), 17 deletions(-) diff --git a/hts/util/DataDownloader.py b/hts/util/DataDownloader.py index 44dc7a4..0a414c7 100644 --- a/hts/util/DataDownloader.py +++ b/hts/util/DataDownloader.py @@ -56,20 +56,18 @@ if __name__ == "__main__": today = datetime.today() PROJECT_HOME = os.path.join(os.path.dirname(os.path.join(os.path.dirname(os.path.join(os.path.dirname(__file__)))))) - RESOURCE_DIR = PROJECT_HOME + "/resources" + RESOURCE_DIR = PROJECT_HOME + "/hits/data" stock_codes = ["252670", "122630"] dataDownloader = DataDownloader() for stock_code in stock_codes: - outFp_m = open(RESOURCE_DIR + "/raw/" + stock_code + "_m.csv", mode="w", encoding="utf-8") - outFp_m.write("%s,%s,%s,%s,%s,%s,%s\n" % ("날짜", "시간", "시가", "고가", "저가", "종가", "거래량")) - outFp_S = open(RESOURCE_DIR + "/raw/" + stock_code + "_S.csv", mode="w", encoding="utf-8") - outFp_S.write("%s,%s,%s,%s,%s,%s,%s\n" % ("날짜", "시간", "시가", "고가", "저가", "종가", "거래량")) + outFp = open(RESOURCE_DIR + "/" + stock_code + "_"+today.strftime('%Y%m%d')+".csv", mode="w", encoding="utf-8") + outFp.write("%s,%s,%s,%s,%s,%s,%s\n" % ("날짜", "시간", "시가", "고가", "저가", "종가", "거래량")) zero_count_m, zero_count_S = 0, 0 for i in range(1, 100000): - given_day = datetime.today() - timedelta(i) + given_day = today - timedelta(i) # '차트 주가 - 월(M), 주(W), 일(D), 시(H), 분(m), 초(S) 차트 요청 if zero_count_m < 10: @@ -80,18 +78,9 @@ if __name__ == "__main__": zero_count_m = 0 print(given_day, "m", count_m) - if zero_count_S < 10: - count_S = dataDownloader.writeStockData(stock_code, given_day.strftime('%Y%m%d'), 'S', outFp_S) - if count_S == 0: - zero_count_S += 1 - else: - zero_count_S = 0 - print(given_day, "S", count_S) - - if zero_count_m > 10 and zero_count_S > 10: + if zero_count_m > 10: break - outFp_m.close() - outFp_S.close() + outFp.close() print ("done...")