init

hts/HTS.py

@@ -640,6 +640,8 @@ class HTS:
     def buyRealTime(self, stock_code, given_day):
         previous_price = 0
         buy_count = 700
+        total_byu_amt = 0
+
         logFp = open(given_day+".log", "w")
 
         timecheckList = pd.read_csv("timecheck.csv").values.tolist()
@@ -675,10 +677,6 @@ class HTS:
 
                 # 매수신청과 5원 높여서 매도신청
                 if price > 0:
-                    print("BUY", second, price)
-                    logFp.write("%s,%s, %d\n" % ("BUY", second, price))
-                    logFp.flush()
-
                     if previous_price > 0:
                         if previous_price > price:
                             if buy_count < 600:
@@ -695,10 +693,16 @@ class HTS:
                     previous_price = price
 
                     # 매수 주문
-                    self.requestOrder("2", stock_code, buy_count , price)
-                    ## 매도 주문 (아래 잔고를 체킇서 매도를 호출하는 것으로 시도한다.)
-                    #time.sleep(60)
-                    #self.requestOrder("1", stock_code, buy_count , price + 5)
+                    # 현재까지 매입금액이 7백만원 이하일 때만 매수를 한다.
+                    if total_byu_amt < 7000000:
+                        self.requestOrder("2", stock_code, buy_count , price)
+
+                        ## 매도 주문 (아래 잔고를 체킇서 매도를 호출하는 것으로 시도한다.)
+                        #time.sleep(60)
+                        #self.requestOrder("1", stock_code, buy_count , price + 5)
+                        print("BUY", second, price)
+                        logFp.write("%s,%s, %d\n" % ("BUY", second, price))
+                        logFp.flush()
 
                 # 가져온 만큼 데이터를 누적해서 파일로 작성한다.
                 self.write(given_day, result)
@@ -713,6 +717,8 @@ class HTS:
             jangoDic = self.requstJango()
             if jangoDic and len(jangoDic.keys()) > 0:
                 for code in jangoDic:
+                    total_byu_amt = jangoDic[code]['매입금액']
+
                     if jangoDic[code]['매도가능'] > 0:
                         # 가장 마지막 현재가를 가져온다.
                         if len(result['close']) > 0: