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

init

Browse Source
This commit is contained in:
dosang.yoon
2022-06-04 13:32:05 +09:00
parent 14d98b0610
commit 5fd46956d1
1 changed files with 29 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
hts/util/DataDownloader.py
View File

@@ -58,39 +58,40 @@ if __name__ == "__main__":
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"
#stock_code = "252670"
stock_code = "122630"
stock_codes = ["252670", "122630"]
dataDownloader = DataDownloader()
outFp_m = open(RESOURCE_DIR + "/" + 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 + "/" + stock_code + "_S.csv", mode="w", encoding="utf-8")
outFp_S.write("%s,%s,%s,%s,%s,%s,%s\n" % ("날짜", "시간", "시가", "고가", "저가", "종가", "거래량"))
for stock_code in stock_codes:
outFp_m = open(RESOURCE_DIR + "/" + 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 + "/" + stock_code + "_S.csv", mode="w", encoding="utf-8")
outFp_S.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)
zero_count_m, zero_count_S = 0, 0
for i in range(1, 100000):
given_day = datetime.today() - timedelta(i)
# '차트 주가 - 월(M), 주(W), 일(D), 시(H), 분(m), 초(S) 차트 요청
if zero_count_m < 10:
count_m = dataDownloader.writeStockData(stock_code, given_day.strftime('%Y%m%d'), 'm', outFp_m)
if count_m == 0:
zero_count_m += 1
else:
zero_count_m = 0
print(given_day, "m", count_m)
# '차트 주가 - 월(M), 주(W), 일(D), 시(H), 분(m), 초(S) 차트 요청
if zero_count_m < 10:
count_m = dataDownloader.writeStockData(stock_code, given_day.strftime('%Y%m%d'), 'm', outFp_m)
if count_m == 0:
zero_count_m += 1
else:
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_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:
break
if zero_count_m > 10 and zero_count_S > 10:
break
outFp_m.close()
outFp_S.close()
outFp_m.close()
outFp_S.close()
print ("done...")