188 lines
7.7 KiB
Python
188 lines
7.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
自适应三分位趋势策略 - 实盘交易
|
|||
|
|
拉取 Bitmart 5/15/60 分钟数据，计算信号，执行开平仓（沿用 交易/bitmart-三分之一策略交易 的浏览器/API 逻辑）
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import time
|
|||
|
|
import datetime
|
|||
|
|
from typing import List, Dict, Optional, Tuple
|
|||
|
|
try:
|
|||
|
|
from loguru import logger
|
|||
|
|
except ImportError:
|
|||
|
|
import logging
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|||
|
|
if ROOT_DIR not in sys.path:
|
|||
|
|
sys.path.insert(0, ROOT_DIR)
|
|||
|
|
|
|||
|
|
from bitmart.api_contract import APIContract
|
|||
|
|
from adaptive_third_strategy.config import (
|
|||
|
|
STEP_5M,
|
|||
|
|
STEP_15M,
|
|||
|
|
STEP_60M,
|
|||
|
|
ATR_PERIOD,
|
|||
|
|
EMA_SHORT,
|
|||
|
|
EMA_MID_FAST,
|
|||
|
|
EMA_MID_SLOW,
|
|||
|
|
EMA_LONG_FAST,
|
|||
|
|
EMA_LONG_SLOW,
|
|||
|
|
BASE_POSITION_PERCENT,
|
|||
|
|
MAX_POSITION_PERCENT,
|
|||
|
|
CONTRACT_SYMBOL,
|
|||
|
|
)
|
|||
|
|
from adaptive_third_strategy.indicators import get_ema_atr_from_klines, align_higher_tf_ema
|
|||
|
|
from adaptive_third_strategy.strategy_core import check_trigger, build_volume_ma
|
|||
|
|
from adaptive_third_strategy.data_fetcher import fetch_klines, _format_bar
|
|||
|
|
|
|||
|
|
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AdaptiveThirdStrategyTrader:
|
|||
|
|
"""实盘：获取多周期K线 -> 策略信号 -> 开平仓（可接浏览器或纯 API）"""
|
|||
|
|
|
|||
|
|
def __init__(
|
|||
|
|
self,
|
|||
|
|
api_key: str,
|
|||
|
|
secret_key: str,
|
|||
|
|
memo: str = "合约交易",
|
|||
|
|
symbol: str = CONTRACT_SYMBOL,
|
|||
|
|
bit_id: Optional[str] = None,
|
|||
|
|
):
|
|||
|
|
self.api_key = api_key
|
|||
|
|
self.secret_key = secret_key
|
|||
|
|
self.memo = memo
|
|||
|
|
self.symbol = symbol
|
|||
|
|
self.bit_id = bit_id
|
|||
|
|
self.contractAPI = APIContract(api_key, secret_key, memo, timeout=(5, 15))
|
|||
|
|
self.check_interval = 3
|
|||
|
|
self.last_trigger_kline_id: Optional[int] = None
|
|||
|
|
self.last_trigger_direction: Optional[str] = None
|
|||
|
|
self.last_trade_kline_id: Optional[int] = None
|
|||
|
|
self.position_direction: int = 0 # 1 多 -1 空 0 无
|
|||
|
|
|
|||
|
|
def get_klines_multi(self) -> Tuple[List[Dict], List[Dict], List[Dict]]:
|
|||
|
|
"""拉取当前 5/15/60 分钟 K 线（最近约 3 小时足够算 EMA/ATR）"""
|
|||
|
|
end_ts = int(time.time())
|
|||
|
|
start_ts = end_ts - 3600 * 3
|
|||
|
|
k5 = fetch_klines(self.contractAPI, STEP_5M, start_ts, end_ts, self.symbol)
|
|||
|
|
k15 = fetch_klines(self.contractAPI, STEP_15M, start_ts, end_ts, self.symbol)
|
|||
|
|
k60 = fetch_klines(self.contractAPI, STEP_60M, start_ts, end_ts, self.symbol)
|
|||
|
|
return k5, k15, k60
|
|||
|
|
|
|||
|
|
def get_position_status(self) -> bool:
|
|||
|
|
"""查询当前持仓，更新 self.position_direction"""
|
|||
|
|
try:
|
|||
|
|
response = self.contractAPI.get_position(contract_symbol=self.symbol)[0]
|
|||
|
|
if response.get("code") != 1000:
|
|||
|
|
return False
|
|||
|
|
positions = response.get("data", [])
|
|||
|
|
if not positions:
|
|||
|
|
self.position_direction = 0
|
|||
|
|
return True
|
|||
|
|
self.position_direction = 1 if positions[0].get("position_type") == 1 else -1
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"持仓查询异常: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def check_realtime_signal(
|
|||
|
|
self,
|
|||
|
|
klines_5m: List[Dict],
|
|||
|
|
klines_15m: List[Dict],
|
|||
|
|
klines_60m: List[Dict],
|
|||
|
|
) -> Tuple[Optional[str], Optional[float], Optional[Dict]]:
|
|||
|
|
"""
|
|||
|
|
基于当前 K 线（最后一根未收盘）检测实时信号。
|
|||
|
|
返回 (方向, 触发价, 当前K线) 或 (None, None, None)。
|
|||
|
|
"""
|
|||
|
|
if len(klines_5m) < ATR_PERIOD + 2:
|
|||
|
|
return None, None, None
|
|||
|
|
from adaptive_third_strategy.indicators import get_ema_atr_from_klines
|
|||
|
|
from adaptive_third_strategy.strategy_core import check_trigger
|
|||
|
|
ema_5m, atr_5m = get_ema_atr_from_klines(klines_5m, EMA_SHORT, ATR_PERIOD)
|
|||
|
|
ema_15m_align = align_higher_tf_ema(klines_5m, klines_15m, EMA_MID_FAST, EMA_MID_SLOW)
|
|||
|
|
ema_60m_align = align_higher_tf_ema(klines_5m, klines_60m, EMA_LONG_FAST, EMA_LONG_SLOW)
|
|||
|
|
volume_ma = build_volume_ma(klines_5m)
|
|||
|
|
curr_idx = len(klines_5m) - 1
|
|||
|
|
direction, trigger_price, _, _ = check_trigger(
|
|||
|
|
klines_5m, curr_idx, atr_5m, ema_5m, ema_15m_align, ema_60m_align, volume_ma, use_confirm=True
|
|||
|
|
)
|
|||
|
|
curr = klines_5m[curr_idx]
|
|||
|
|
curr_id = curr["id"]
|
|||
|
|
if direction is None:
|
|||
|
|
return None, None, None
|
|||
|
|
if self.last_trigger_kline_id == curr_id and self.last_trigger_direction == direction:
|
|||
|
|
return None, None, None
|
|||
|
|
if self.last_trade_kline_id == curr_id:
|
|||
|
|
return None, None, None
|
|||
|
|
if (direction == "long" and self.position_direction == 1) or (direction == "short" and self.position_direction == -1):
|
|||
|
|
return None, None, None
|
|||
|
|
return direction, trigger_price, curr
|
|||
|
|
|
|||
|
|
def run_loop(self, ding_callback=None):
|
|||
|
|
"""
|
|||
|
|
主循环：拉数据 -> 检测信号 -> 若需交易则调用开平仓（需外部实现或注入）。
|
|||
|
|
ding_callback(msg, error=False) 可选，用于钉钉通知。
|
|||
|
|
"""
|
|||
|
|
def ding(msg: str, error: bool = False):
|
|||
|
|
if ding_callback:
|
|||
|
|
ding_callback(msg, error)
|
|||
|
|
else:
|
|||
|
|
if error:
|
|||
|
|
logger.error(msg)
|
|||
|
|
else:
|
|||
|
|
logger.info(msg)
|
|||
|
|
|
|||
|
|
logger.info("自适应三分位趋势策略实盘启动，按 Bitmart 方式拉取 5/15/60 分钟数据")
|
|||
|
|
while True:
|
|||
|
|
try:
|
|||
|
|
k5, k15, k60 = self.get_klines_multi()
|
|||
|
|
if len(k5) < ATR_PERIOD + 2:
|
|||
|
|
time.sleep(self.check_interval)
|
|||
|
|
continue
|
|||
|
|
if not self.get_position_status():
|
|||
|
|
time.sleep(self.check_interval)
|
|||
|
|
continue
|
|||
|
|
direction, trigger_price, curr = self.check_realtime_signal(k5, k15, k60)
|
|||
|
|
if direction and trigger_price is not None and curr is not None:
|
|||
|
|
curr_id = curr["id"]
|
|||
|
|
ding(f"信号: {direction} @ {trigger_price:.2f} 当前K线 id={curr_id}")
|
|||
|
|
# 这里接入你的开平仓实现：平仓 + 开单（可调用 交易/bitmart-三分之一策略交易 的 平仓/开单 或 API 下单）
|
|||
|
|
# 示例：仅记录，不实际下单
|
|||
|
|
self.last_trigger_kline_id = curr_id
|
|||
|
|
self.last_trigger_direction = direction
|
|||
|
|
self.last_trade_kline_id = curr_id
|
|||
|
|
time.sleep(self.check_interval)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.exception(e)
|
|||
|
|
time.sleep(60)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
import argparse
|
|||
|
|
parser = argparse.ArgumentParser(description="自适应三分位趋势策略实盘")
|
|||
|
|
parser.add_argument("--api-key", default="", help="Bitmart API Key")
|
|||
|
|
parser.add_argument("--secret-key", default="", help="Bitmart Secret Key")
|
|||
|
|
parser.add_argument("--bit-id", default="", help="比特浏览器 ID（若用浏览器下单）")
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
api_key = args.api_key or os.environ.get("BITMART_API_KEY", "a0fb7b98464fd9bcce67e7c519d58ec10d0c38a8")
|
|||
|
|
secret_key = args.secret_key or os.environ.get("BITMART_SECRET_KEY", "4eaeba78e77aeaab1c2027f846a276d164f264a44c2c1bb1c5f3be50c8de1ca5")
|
|||
|
|
trader = AdaptiveThirdStrategyTrader(api_key, secret_key, bit_id=args.bit_id or None)
|
|||
|
|
try:
|
|||
|
|
from 交易.tools import send_dingtalk_message
|
|||
|
|
def ding(msg, error=False):
|
|||
|
|
prefix = "❌自适应三分位：" if error else "🔔自适应三分位："
|
|||
|
|
send_dingtalk_message(f"{prefix}{msg}")
|
|||
|
|
except Exception:
|
|||
|
|
def ding(msg, error=False):
|
|||
|
|
logger.info(msg)
|
|||
|
|
trader.run_loop(ding_callback=ding)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|