init

HTS_etf.py

@@ -84,7 +84,6 @@ class HTS_etf (HTS):
         return result
 
     def buyRealTime(self, today, stocks, analyzed_day=1000):
-        self.connect2DB("hts.db")
 
         print ("START...")
         THIS_TIME = datetime.now()
@@ -192,7 +191,6 @@ class HTS_etf (HTS):
             time.sleep(10)
             THIS_TIME = datetime.now()
 
-        self.disconnect()
         return
 
     def updteTodayStock(self, stock_code, today_str):
@@ -215,10 +213,13 @@ if __name__ == "__main__":
     ]
 
     hts = HTS_etf(RESOURCE_PATH)
+    hts.connect2DB("hts.db")
+
     today_str = today.strftime('%Y%m%d')
     hts.buyRealTime(today_str, stocks, analyzed_day=1000)
 
     db_filename = os.path.join(RESOURCE_PATH, "hts.db")
     hts.insertStockData(today_str, stocks)
 
+    hts.disconnect()
     print ("done...")

HTS_stocks.py

@@ -97,8 +97,6 @@ class HTS_Stocks (HTS):
         return all_stocks, valid_company
 
     def buyRealTime(self, today, n = 200):
-        self.connect2DB("hts.db")
-        self.connect2StockDB()
 
         print ("START...")
         THIS_TIME = datetime.now()
@@ -205,8 +203,6 @@ class HTS_Stocks (HTS):
             time.sleep(10)
             THIS_TIME = datetime.now()
 
-        self.disconnectStockDB()
-        self.disconnect()
         return
 
     def updteTodayStock(self, db_filename, stock_code, today_str):
@@ -222,7 +218,13 @@ if __name__ == "__main__":
     RESOURCE_PATH = os.path.join(PROJECT_HOME, "resources")
 
     hts = HTS_Stocks(RESOURCE_PATH)
+    hts.connect2DB("hts.db")
+    hts.connect2StockDB()
+
     today_str = today.strftime('%Y%m%d')
     hts.buyRealTime(today_str)
 
+    hts.disconnectStockDB()
+    hts.disconnect()
+
     print ("done...")