init

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