init

LabelMaker.py

@@ -1,5 +1,6 @@
 import os
 import logging
+logging.basicConfig(level=logging.INFO)
 from stock.util.LabelChecker import LabelChecker
 
 class LabelMaker:
@@ -24,6 +25,7 @@ class LabelMaker:
         return
 
     def write(self, stock_code, ymd, outFp):
+        logging.info(stock_code + " / " + ymd)
         self.labelChecker.showLabels(stock_code, ymd, outFp)
         return
 
@@ -32,7 +34,7 @@ class LabelMaker:
         for stock_code in stock_codes:
             ymds = self.labelChecker.getDate(stock_code)
             for ymd in ymds:
-                logging.info(stock_code, ymd)
+                logging.info(stock_code + " / " + ymd)
                 bsLine, data = self.labelChecker.makeCandidate(stock_code, ymd)
                 self.labelChecker.updateLabel(self.db_filename, stock_code, bsLine, data, ymd)
         return
@@ -51,11 +53,17 @@ if __name__ == "__main__":
                    '20220729', '20220801', '20220802', '20220803', '20220804',
                    '20220805', '20220808', '20220809', '20220810', '20220811',
                    '20220812', '20220816', '20220817'],
-        # "122630": ['20220801', '20220802', '20220803', '20220804', '20220805', '20220808', '20220809', '20220810', '20220811', '20220812'],
+        "122630": ['20220701', '20220704', '20220705', '20220706', '20220707',
+                   '20220708', '20220711', '20220712', '20220713', '20220714',
+                   '20220715', '20220718', '20220719', '20220720', '20220721',
+                   '20220722', '20220725', '20220726', '20220727', '20220728',
+                   '20220729', '20220801', '20220802', '20220803', '20220804',
+                   '20220805', '20220808', '20220809', '20220810', '20220811',
+                   '20220812', '20220816', '20220817'],
     }
 
 
-    MODE = "UPDATE"
+    MODE = "WRITE"
     if MODE == "UPDATE":
         # 매일 입력하면서 정답 셋 만들기
         stock_code = "252670"
@@ -82,7 +90,8 @@ if __name__ == "__main__":
     elif MODE == "WRITE":
 
         for stock_code in stock_codes:
-            outFp = open("check_"+stock_code+".txt", "w")
+            fileName = os.path.join(RESOURCE_PATH, "tmp", "check_"+stock_code+".txt")
+            outFp = open(fileName, "w")
             for ymd in stock_codes[stock_code]:
                 labelMaker.write(stock_code, ymd, outFp)
             outFp.close()

stock/util/LabelChecker.py

@@ -296,6 +296,7 @@ class LabelChecker (HTS):
             if labels[i] == 2:
                 bsLine['buy'][i] = lows[i]
 
+        if outFp is None:
             #self.draw_simple(stock_code, ymd, data, bsLine)
             self.draw_detail(stock_code, ymd, data, bsLine)
         return