From be73e2b78857adaf006063275726b69c4c60f0d7 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期三, 12 十月 2022 11:54:25 +0800
Subject: [PATCH] 买撤策略修改;加入报警功能

---
 ths_util.py |   27 ++++++++++++++++++++++++++-
 1 files changed, 26 insertions(+), 1 deletions(-)

diff --git a/ths_util.py b/ths_util.py
index f44c855..9b427fc 100644
--- a/ths_util.py
+++ b/ths_util.py
@@ -8,7 +8,11 @@
 import win32con
 import win32gui
 
+import redis_manager
+import tool
 import trade_gui
+
+__redisManager = redis_manager.RedisManager(2)
 
 
 def __click(hwnd):
@@ -118,7 +122,7 @@
 
     if len(wins) < 3:
         for i in range(0, 3):
-            btn = win32gui.GetDlgItem(wins[len(wins)-1], 0x000005ED)
+            btn = win32gui.GetDlgItem(wins[len(wins) - 1], 0x000005ED)
             # 鐐瑰嚮浜嬩欢娣诲姞
             __click(btn)
             time.sleep(0.5)
@@ -171,6 +175,27 @@
     raise Exception("涓撲笟鐗堜笅鍗曟墦寮�澶辫触")
 
 
+def set_ths_dead_state(client_id, dead):
+    redis = __redisManager.getRedis()
+    key = "ths_state_dead_count-{}".format(client_id)
+    if not dead:
+        redis.setex(key, tool.get_expire(), 0)
+    else:
+        redis.incrby(key, 1)
+        redis.expire(key, tool.get_expire())
+
+
+# 鍚岃姳椤烘槸鍚﹀崱姝�
+def is_ths_dead(client_id):
+    key = "ths_state_dead_count-{}".format(client_id)
+    redis = __redisManager.getRedis()
+    val = redis.get(key)
+    if val is not None and int(val) >= 5:
+        return True
+    else:
+        return False
+
+
 if __name__ == "__main__":
     open_trade_gui()
 

--
Gitblit v1.8.0