From 045a5aa6434da6e83c3d850b17e7e58cd7b55ef5 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期一, 15 五月 2023 15:46:32 +0800
Subject: [PATCH] 开盘啦板块影响交易逻辑

---
 trade/trade_manager.py |   16 +++++++++++++---
 1 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/trade/trade_manager.py b/trade/trade_manager.py
index d3d1714..0027416 100644
--- a/trade/trade_manager.py
+++ b/trade/trade_manager.py
@@ -78,7 +78,7 @@
 
     @classmethod
     def add_code(cls, code):
-        cls.__get_redis().sadd(cls.__key,code)
+        cls.__get_redis().sadd(cls.__key, code)
 
     @classmethod
     def remove_code(cls, code):
@@ -91,7 +91,6 @@
     @classmethod
     def clear(cls):
         cls.__get_redis().delete(cls.__key)
-
 
 
 # 鑾峰彇浜ゆ槗鐘舵��
@@ -117,6 +116,17 @@
     if keys is not None:
         for key in keys:
             if int(redis.get(key)) == state:
+                codes.append(key.replace("trade-state-", ''))
+    return codes
+
+
+def get_codes_by_trade_states(states):
+    redis = __redis_manager.getRedis()
+    keys = redis.keys("trade-state-*")
+    codes = []
+    if keys is not None:
+        for key in keys:
+            if int(redis.get(key)) in states:
                 codes.append(key.replace("trade-state-", ''))
     return codes
 
@@ -477,4 +487,4 @@
 
 
 if __name__ == "__main__":
-    set_trade_state("002351",TRADE_STATE_BUY_DELEGATED)
+    set_trade_state("002351", TRADE_STATE_BUY_DELEGATED)

--
Gitblit v1.8.0