From 606cc5c65211af6023679c163e888e3ea0410bf0 Mon Sep 17 00:00:00 2001 From: zhiyong Date: Thu, 1 May 2025 00:21:46 +0800 Subject: [PATCH] update example --- src/example_trade.py | 36 ++++++++++++++++++------------------ 1 file changed, 18 insertions(+), 18 deletions(-) diff --git a/src/example_trade.py b/src/example_trade.py index 5466b46..557bca9 100644 --- a/src/example_trade.py +++ b/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}