init

HTS_etf.py

@@ -84,6 +84,7 @@ class HTS_etf (HTS):
         return result
 
     def buyRealTime(self, today, stocks, analyzed_day=1000):
+        self.connect2DB("hts.db")
 
         print ("START...")
         THIS_TIME = datetime.now()
@@ -191,11 +192,12 @@ class HTS_etf (HTS):
             time.sleep(10)
             THIS_TIME = datetime.now()
 
+        self.disconnect()
         return
 
-    def updteTodayStock(self, db_filename, stock_code, today_str):
+    def updteTodayStock(self, stock_code, today_str):
         bsLine, data = self.labelChecker.makeCandidate(stock_code, today_str)
-        self.labelChecker.updateLabel(db_filename, stock_code, bsLine, data, today_str)
+        self.labelChecker.updateLabel(stock_code, bsLine, data, today_str)
         return
 
 
@@ -209,7 +211,7 @@ if __name__ == "__main__":
     # KODEX 인버스 * 2
     stocks = [
         {"stock_code": "252670", "stock_name": "KODEX 200선물인버스2X"},
-        #{"stock_code": "122630", "stock_name": "KODEX 레버리지"}
+        {"stock_code": "122630", "stock_name": "KODEX 레버리지"}
     ]
 
     hts = HTS_etf(RESOURCE_PATH)
@@ -217,6 +219,6 @@ if __name__ == "__main__":
     hts.buyRealTime(today_str, stocks, analyzed_day=1000)
 
     db_filename = os.path.join(RESOURCE_PATH, "hts.db")
-    hts.insertStockData(db_filename, today_str, stocks)
+    hts.insertStockData(today_str, stocks)
 
     print ("done...")