From 5fd46956d19eba7fbad90ae584b09e345d23ccab Mon Sep 17 00:00:00 2001
From: "dosang.yoon" <dosang.yoon@voithru.com>
Date: Sat, 4 Jun 2022 13:32:05 +0900
Subject: [PATCH] init

---
 hts/util/DataDownloader.py | 57 +++++++++++++++++++-------------------
 1 file changed, 29 insertions(+), 28 deletions(-)

diff --git a/hts/util/DataDownloader.py b/hts/util/DataDownloader.py
index 67aa47b..380aff4 100644
--- a/hts/util/DataDownloader.py
+++ b/hts/util/DataDownloader.py
@@ -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...")