From 8677d39f64869de0b0a6dcb58f61597d6041a056 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期四, 04 七月 2024 18:45:28 +0800
Subject: [PATCH] 策略修改

---
 test/test_api.py |   19 ++++++++++++++++---
 1 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/test/test_api.py b/test/test_api.py
index 48cee32..8de3447 100644
--- a/test/test_api.py
+++ b/test/test_api.py
@@ -3,7 +3,7 @@
 from code_attribute import history_k_data_util, target_codes_manager
 from third_data import kpl_api, kpl_data_manager, kpl_block_util, kpl_util
 from third_data.kpl_data_manager import KPLCodeJXBlockManager
-from trade import huaxin_trade_api
+from trade import huaxin_trade_api, buy_strategy
 from trade.trade_manager import CodeTradeStateManager
 from utils import output_data_util, tool
 
@@ -28,8 +28,21 @@
                 sse_other_codes.append(code.encode())
     return sse_stock_codes, sse_other_codes, szse_codes
 
+
 if __name__ == "__main__":
-    underlying_codes = target_codes_manager.get_subscript_underlying_codes()
-    print(__split_codes(underlying_codes))
+    code = "000333"
+    print(CodeTradeStateManager().get_trade_state(code,buy_strategy.STRATEGY_TYPE_LIMIT_UP))
+    print(CodeTradeStateManager().get_trade_state(code, buy_strategy.STRATEGY_TYPE_RISE_HIGH))
+
+    CodeTradeStateManager().set_trade_state(code, buy_strategy.STRATEGY_TYPE_LIMIT_UP,
+                                            CodeTradeStateManager.TRADE_STATE_ALREADY_BUY)
+    print(CodeTradeStateManager().get_trade_state(code,buy_strategy.STRATEGY_TYPE_LIMIT_UP))
+    print(CodeTradeStateManager().get_trade_state(code, buy_strategy.STRATEGY_TYPE_RISE_HIGH))
+
+    CodeTradeStateManager().set_trade_state(code, buy_strategy.STRATEGY_TYPE_RISE_HIGH,
+                                            CodeTradeStateManager.TRADE_STATE_ALREADY_BUY)
+    print(CodeTradeStateManager().get_trade_state(code,buy_strategy.STRATEGY_TYPE_LIMIT_UP))
+    print(CodeTradeStateManager().get_trade_state(code, buy_strategy.STRATEGY_TYPE_RISE_HIGH))
+
 
 

--
Gitblit v1.8.0