update example

src/example_trade.py

@@ -3,9 +3,9 @@ import requests
 # 服务器地址
 URL = "http://trader.biggerfish.tech:9527/yu"
 # 策略名称常量
-STRATEGY = "追梦投资港股ETF"
+STRATEGY = "港股ETF网格"
 
-def buy(code: str, price: float, amount: int, strategy_name: str) -> dict:
+def buy(code: str, price: float, amount: int, strategy_name: str = STRATEGY) -> dict:
     """买入股票
     
     Args:
@@ -23,7 +23,7 @@ def buy(code: str, price: float, amount: int, strategy_name: str) -> dict:
     response = requests.post(f"{URL}/buy", json=data)
     return response.json()
 
-def sell(code: str, price: float, amount: int, strategy_name: str) -> dict:
+def sell(code: str, price: float, amount: int, strategy_name: str = STRATEGY) -> dict:
     """卖出股票
     
     Args:
@@ -41,11 +41,11 @@ def sell(code: str, price: float, amount: int, strategy_name: str) -> dict:
     response = requests.post(f"{URL}/sell", json=data)
     return response.json()
 
-def get_positions(strategy_name: str = "") -> dict:
+def get_positions(strategy_name: str = STRATEGY) -> dict:
     """获取持仓信息
     
     Args:
-        strategy_name: 策略名称，默认为空字符串，表示获取所有持仓
+        strategy_name: 策略名称，默认为STRATEGY常量
     """
     params = {}
     if strategy_name:
@@ -53,11 +53,11 @@ def get_positions(strategy_name: str = "") -> dict:
     response = requests.get(f"{URL}/positions", params=params)
     return response.json()
 
-def get_target_positions(strategy_name: str = "") -> dict:
+def get_target_positions(strategy_name: str = STRATEGY) -> dict:
     """获取目标持仓信息（仅实盘模式有效）
     
     Args:
-        strategy_name: 策略名称，默认为空字符串，表示获取所有策略的目标持仓
+        strategy_name: 策略名称，默认为STRATEGY常量
     """
     params = {"target": "true"}
     if strategy_name:
@@ -65,7 +65,7 @@ def get_target_positions(strategy_name: str = "") -> dict:
     response = requests.get(f"{URL}/positions", params=params)
     return response.json()
 
-def clear_strategy(strategy_name: str) -> dict:
+def clear_strategy(strategy_name: str = STRATEGY) -> dict:
     """清除策略持仓数据
     
     Args:
@@ -122,11 +122,11 @@ def get_order_status() -> dict:
     response = requests.get(f"{URL}/order_status")
     return response.json()
 
-def get_strategy_targets(strategy_name: str = "") -> dict:
+def get_strategy_targets(strategy_name: str = STRATEGY) -> dict:
     """获取策略目标持仓（仅实盘模式）
     
     Args:
-        strategy_name: 策略名称，默认为空字符串，表示获取所有策略的目标持仓
+        strategy_name: 策略名称，默认为STRATEGY常量
         
     Returns:
         字典形式的策略目标持仓信息
@@ -153,7 +153,7 @@ if __name__ == "__main__":
     print("健康状态:", health)
     
     # 示例：查询默认策略的持仓信息
-    positions = get_positions(STRATEGY)
+    positions = get_positions()
     print("当前持仓:", positions)
     # 模拟输出:
     #  {'data': [], 'success': True}
@@ -163,7 +163,7 @@ if __name__ == "__main__":
     print("账户余额:", balance)
     
     # 示例：买入中国银行1000股，价格3.45
-    result = buy("601988.SH", 3.45, 3000, STRATEGY)
+    result = buy("601988.SH", 3.45, 3000)
     print("买入结果:", result)
     # 模拟输出:
     #{'data': {'message': '模拟买入 - 代码: 601988.SH, 价格: 3.45, 数量: 3000', 'order_id': 'simulation'}, 'success': True}
@@ -173,7 +173,7 @@ if __name__ == "__main__":
     print("订单状态:", order_status)
     
     # 示例：卖出中国银行1000股，价格3.48
-    result = sell("601988.SH", 3.48, 1000, STRATEGY)
+    result = sell("601988.SH", 3.48, 1000)
     print("卖出结果:", result)
     # 模拟输出:
     #  {'data': {'message': '模拟卖出 - 代码: 601988.SH, 价格: 3.48, 数量: 1000', 'order_id': 'simulation'}, 'success': True}
@@ -187,17 +187,17 @@ if __name__ == "__main__":
     print("今日委托:", entrusts)
 
     # 示例：再次查询持仓变化
-    positions = get_positions(STRATEGY)
+    positions = get_positions()
     print("交易后持仓:", positions)
     # 模拟输出:
     # {'data': [{'601988.SH': {'closeable_amount': 2000, 'total_amount': 2000}}], 'success': True}
     
     # 示例：获取目标持仓
-    target_positions = get_target_positions(STRATEGY)
+    target_positions = get_target_positions()
     print("目标持仓:", target_positions)
     
     # 示例：获取策略目标持仓
-    strategy_targets = get_strategy_targets(STRATEGY)
+    strategy_targets = get_strategy_targets()
     print("策略目标持仓:", strategy_targets)
     
     # 示例：取消订单（需要真实的委托编号）
@@ -205,13 +205,13 @@ if __name__ == "__main__":
     # print("取消订单结果:", cancel_result)
     
     # 示例：清除策略持仓数据
-    result = clear_strategy(STRATEGY)
+    result = clear_strategy()
     print("清除策略持仓结果:", result)
     # 模拟输出:
     # {'message': "成功清除策略 '追梦投资港股ETF' 的持仓数据", 'success': True}
     
     # 示例：清除后查询持仓
-    positions = get_positions(STRATEGY)
+    positions = get_positions()
     print("清除后持仓:", positions)
     # 模拟输出:
     # {'data': [], 'success': True}