init

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...")