init

hts/HTS.py

@@ -1,10 +1,10 @@
-#import win32com.client
+import win32com.client
 import time
 import os
 from datetime import datetime, timedelta
 import pandas as pd
 from enum import Enum
-import plotly.graph_objects as go
+#import plotly.graph_objects as go
 from stockpredictor.analysis.Common import Common
 
 # enum 주문 상태 세팅용
@@ -781,7 +781,7 @@ class HTS:
         size = len(data["Close"])
         STOCK = []
         for i in range(size):
-            STOCK.append({'volume': data['Volume'][i], 'close': data["Close"][i], 'open': data["Open"][i], 'high': data["High"][i], 'low': data["Low"][i], 'avg5': data["avg5"][i], 'avg20': data["avg20"][i], 'avg60': data["avg60"][i], 'avg120': data["avg120"][i], 'avg240': data["avg240"][i]})
+            STOCK.append({'volume': data['Volume'][i], 'close': data["Close"][i], 'open': data["Open"][i], 'high': data["High"][i], 'low': data["Low"][i], 'avg2': data["avg2"][i], 'avg5': data["avg5"][i], 'avg10': data["avg10"][i], 'avg20': data["avg20"][i], 'avg30': data["avg30"], 'avg40': data["avg40"], 'avg50': data["avg50"], 'avg60': data["avg60"][i]})
 
         bsLine = {}
         bsLine['buy']  = [-1 for i in range(size)]
@@ -960,7 +960,7 @@ class HTS:
         while datetime.strptime(given_day + " 083000", '%Y%m%d %H%M%S') < datetime.now() < datetime.strptime(given_day + " 15200", '%Y%m%d %H%M%S'):
             second = datetime.now().strftime('%Y%m%d %H%M%S')
 
-            if datetime.now() < datetime.strptime(given_day + " 144000", '%Y%m%d %H%M%S'):
+            if datetime.now() < datetime.strptime(given_day + " 151000", '%Y%m%d %H%M%S'):
                 if second in timecheck and not timecheck[second]:
                     print("TIMECHECK", second)
                     logFp.write("%s,%s,\n" %("TIMECHECK", second))
@@ -971,7 +971,6 @@ class HTS:
 
                     # 분석을 통해서 볼린저밴드 상/하단을 계산한다.
                     data, upper, lower = self.analyze(result)
-                    final_price = data["Close"][len(data["Close"])- 1]
                     # 사야 할 시점/가격과 팔아야 할 시점/가격을 체크한다.
                     bs_buy_price, bs_sell_price = self.checkTransaction_Realtime(data, upper, lower)
 
@@ -1068,7 +1067,7 @@ if __name__ == "__main__":
 
     given_day = datetime.today().strftime('%Y%m%d')
     #hts.writeStockData(stock_codes, given_day)
-    hts.simulate(stock_codes[0], given_day)
-    #hts.buyRealTime(stock_codes[0], given_day)
+    #hts.simulate(stock_codes[0], given_day)
+    hts.buyRealTime(stock_codes[0], given_day)
 
     print ("done...")