99 lines
4.5 KiB
Python
99 lines
4.5 KiB
Python
import logging
|
||
from core.base import QuantTrader
|
||
from core.strategy import QuantStrategy
|
||
|
||
logging.basicConfig(level=logging.INFO, format='%(asctime)s %(levelname)s: %(message)s')
|
||
|
||
def main() -> None:
|
||
"""主函数"""
|
||
logging.info("=== 比特币量化交易系统 ===")
|
||
# 导入配置
|
||
try:
|
||
from config import API_KEY, SECRET_KEY, PASSPHRASE, SANDBOX, TRADING_CONFIG, TIME_CONFIG
|
||
except ImportError:
|
||
logging.error("找不到config.py文件，请确保配置文件存在")
|
||
return
|
||
# 检查是否配置了API密钥
|
||
if API_KEY == "your_api_key_here":
|
||
logging.error("请先在config.py中配置你的OKX API密钥！\n1. 登录OKX官网\n2. 进入API管理页面\n3. 创建API Key、Secret Key和Passphrase\n4. 将密钥填入config.py文件中的相应位置")
|
||
return
|
||
# 创建交易器实例
|
||
# sandbox = TRADING_CONFIG.get("sandbox", True)
|
||
symbol = TRADING_CONFIG.get("symbol", "BTC-USDT")
|
||
position_size = TRADING_CONFIG.get("position_size", 0.001)
|
||
trader = QuantTrader(
|
||
API_KEY, SECRET_KEY, PASSPHRASE,
|
||
sandbox=SANDBOX,
|
||
symbol=symbol,
|
||
position_size=position_size
|
||
)
|
||
strategy = QuantStrategy(
|
||
API_KEY, SECRET_KEY, PASSPHRASE,
|
||
sandbox=SANDBOX,
|
||
symbol=symbol,
|
||
position_size=position_size
|
||
)
|
||
# 显示菜单
|
||
while True:
|
||
logging.info("\n请选择操作:\n1. 查看账户余额\n2. 查看当前价格\n3. 执行移动平均线策略\n4. 执行RSI策略\n5. 执行网格交易策略\n6. 运行策略循环\n7. 买入测试\n8. 卖出测试\n9. 获取最小交易量\n0. 退出")
|
||
choice = input("请输入选择 (0-9): ").strip()
|
||
if choice == '0':
|
||
logging.info("退出程序")
|
||
break
|
||
elif choice == '1':
|
||
trader.get_account_balance()
|
||
elif choice == '2':
|
||
trader.get_current_price()
|
||
elif choice == '3':
|
||
sma_short_period = TRADING_CONFIG.get("sma_short_period", 5)
|
||
sma_long_period = TRADING_CONFIG.get("sma_long_period", 20)
|
||
strategy.simple_moving_average_strategy(sma_short_period, sma_long_period)
|
||
elif choice == '4':
|
||
period = TRADING_CONFIG.get("rsi_period", 14)
|
||
rsi_oversold = TRADING_CONFIG.get("rsi_oversold", 30)
|
||
rsi_overbought = TRADING_CONFIG.get("rsi_overbought", 70)
|
||
strategy.rsi_strategy(period, rsi_oversold, rsi_overbought)
|
||
elif choice == '5':
|
||
grid_levels = TRADING_CONFIG.get("grid_levels", 5)
|
||
grid_range = TRADING_CONFIG.get("grid_range", 0.02)
|
||
strategy.grid_trading_strategy(grid_levels, grid_range)
|
||
elif choice == '6':
|
||
strategy_name = input("选择策略 (sma/rsi/grid): ").strip()
|
||
interval = TIME_CONFIG.get("strategy_interval", 30)
|
||
strategy.run_strategy_loop(strategy_name, interval, TRADING_CONFIG)
|
||
elif choice == '7':
|
||
defalt_position_size = 0.01
|
||
input_size = input("请输入买入数量: ")
|
||
if input_size:
|
||
try:
|
||
position_size = float(input_size)
|
||
logging.info(f"买入{position_size}BTC")
|
||
trader.place_market_order('buy', position_size)
|
||
except ValueError:
|
||
logging.warning(f"输入无效，默认买入{defalt_position_size}BTC")
|
||
trader.place_market_order('buy', defalt_position_size)
|
||
else:
|
||
logging.info(f"默认买入{defalt_position_size}BTC")
|
||
trader.place_market_order('buy', defalt_position_size)
|
||
elif choice == '8':
|
||
defalt_position_size = 0.01
|
||
input_size = input("请输入卖出数量: ")
|
||
if input_size:
|
||
try:
|
||
position_size = float(input_size)
|
||
logging.info(f"卖出{position_size}BTC")
|
||
trader.place_market_order('sell', position_size)
|
||
except ValueError:
|
||
logging.warning(f"输入无效，默认卖出{defalt_position_size}BTC")
|
||
trader.place_market_order('sell', defalt_position_size)
|
||
else:
|
||
logging.info(f"默认卖出{defalt_position_size}BTC")
|
||
trader.place_market_order('sell', defalt_position_size)
|
||
elif choice == '9':
|
||
trader.get_minimun_order_size()
|
||
else:
|
||
logging.warning("无效选择，请重新输入")
|
||
|
||
if __name__ == "__main__":
|
||
main()
|