From f977c07f21c6496cf8b17baf13ef297a21ca267f Mon Sep 17 00:00:00 2001
From: dsyoon <dosangyoon@gmail.com>
Date: Thu, 2 Feb 2023 16:09:28 +0900
Subject: [PATCH] init

---
 HTS_etf.py    |  5 +++--
 HTS_stocks.py | 10 ++++++----
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/HTS_etf.py b/HTS_etf.py
index b8d5741..1a7ad44 100644
--- a/HTS_etf.py
+++ b/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...")
diff --git a/HTS_stocks.py b/HTS_stocks.py
index 7c93ac7..45e76ff 100644
--- a/HTS_stocks.py
+++ b/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...")