diff --git a/hts/util/DataDownloader.py b/hts/util/DataDownloader.py index b6121cf..bc5038e 100644 --- a/hts/util/DataDownloader.py +++ b/hts/util/DataDownloader.py @@ -52,7 +52,7 @@ class DataDownloader (HTS): outfp.write("%d,%s,%d,%d,%d,%d,%d\n" % (day, str(time).zfill(4), start, high, low, close, vol)) outfp.close() - return + return size if __name__ == "__main__": @@ -64,12 +64,27 @@ if __name__ == "__main__": stock_code = "252670" dataDownloader = DataDownloader() - for i in range(1, 1000): + zero_count_m, zero_count_S = 0, 0 + for i in range(1, 100000): given_day = datetime.today() - timedelta(i) - print(given_day) # '차트 주가 - 월(M), 주(W), 일(D), 시(H), 분(m), 초(S) 차트 요청 - dataDownloader.writeStockData(stock_code, given_day.strftime('%Y%m%d'), 'm', RESOURCE_DIR) - dataDownloader.writeStockData(stock_code, given_day.strftime('%Y%m%d'), 'S', RESOURCE_DIR) + if zero_count_m < 10: + count_m = dataDownloader.writeStockData(stock_code, given_day.strftime('%Y%m%d'), 'm', RESOURCE_DIR) + 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'), 'm', RESOURCE_DIR) + 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 print ("done...")