537 lines
19 KiB
Python
537 lines
19 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
BitMart 基于开盘价的五分之一策略交易
|
||
|
||
策略规则（1111）：
|
||
- 做多触发价 = 当前K线开盘价 + 实体/5
|
||
- 做空触发价 = 当前K线开盘价 - 实体/5
|
||
- 基于前一根有效K线（实体 >= 0.1）
|
||
|
||
执行：触及做多/做空触发价则开仓或反手，同根K线只交易一次
|
||
|
||
运行：python open_fifth_strategy/main.py（在项目根目录 lm_code 下）
|
||
"""
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
_root = Path(__file__).resolve().parent.parent
|
||
if str(_root) not in sys.path:
|
||
sys.path.insert(0, str(_root))
|
||
|
||
import random
|
||
import time
|
||
from concurrent.futures import ThreadPoolExecutor
|
||
|
||
from loguru import logger
|
||
from bit_tools import openBrowser
|
||
from DrissionPage import ChromiumPage
|
||
from DrissionPage import ChromiumOptions
|
||
|
||
from bitmart.api_contract import APIContract
|
||
from 交易.tools import send_dingtalk_message
|
||
|
||
from open_fifth_strategy.config import (
|
||
API_KEY,
|
||
SECRET_KEY,
|
||
MEMO,
|
||
CONTRACT_SYMBOL,
|
||
KLINE_STEP,
|
||
MIN_BODY_SIZE,
|
||
CHECK_INTERVAL,
|
||
LEVERAGE,
|
||
OPEN_TYPE,
|
||
RISK_PERCENT,
|
||
BIT_ID,
|
||
)
|
||
|
||
ding_executor = ThreadPoolExecutor(max_workers=2, thread_name_prefix="dingtalk")
|
||
|
||
|
||
class OpenBasedFifthStrategy:
|
||
"""基于开盘价的五分之一策略"""
|
||
|
||
def __init__(self, bit_id=None):
|
||
self.page = None
|
||
self.api_key = API_KEY
|
||
self.secret_key = SECRET_KEY
|
||
self.memo = MEMO
|
||
self.contract_symbol = CONTRACT_SYMBOL
|
||
self.contractAPI = APIContract(
|
||
self.api_key, self.secret_key, self.memo, timeout=(5, 15)
|
||
)
|
||
|
||
self.start = 0 # 持仓: -1空, 0无, 1多
|
||
self.open_avg_price = None
|
||
self.current_amount = None
|
||
self.position_cross = None
|
||
self.bit_id = bit_id or BIT_ID
|
||
|
||
self.min_body_size = MIN_BODY_SIZE
|
||
self.kline_step = KLINE_STEP
|
||
self.check_interval = CHECK_INTERVAL
|
||
self.leverage = LEVERAGE
|
||
self.open_type = OPEN_TYPE
|
||
self.risk_percent = RISK_PERCENT
|
||
|
||
self.last_trigger_kline_id = None
|
||
self.last_trigger_direction = None
|
||
self.last_trade_kline_id = None
|
||
|
||
# ==================== 策略核心 ====================
|
||
|
||
def get_body_size(self, candle):
|
||
return abs(float(candle["open"]) - float(candle["close"]))
|
||
|
||
def find_valid_prev_bar(self, all_data, current_idx):
|
||
"""找前一根有效K线（实体>=min_body_size）"""
|
||
if current_idx <= 0:
|
||
return None, None
|
||
for i in range(current_idx - 1, -1, -1):
|
||
prev = all_data[i]
|
||
if self.get_body_size(prev) >= self.min_body_size:
|
||
return i, prev
|
||
return None, None
|
||
|
||
def get_trigger_levels(self, prev, curr_open):
|
||
"""
|
||
基于当前K线开盘价计算触发价（1111）
|
||
做多触发 = 当前K线开盘价 + 实体/5
|
||
做空触发 = 当前K线开盘价 - 实体/5
|
||
"""
|
||
body = self.get_body_size(prev)
|
||
if body < 0.001:
|
||
return None, None
|
||
curr_o = float(curr_open)
|
||
return curr_o + body / 5, curr_o - body / 5
|
||
|
||
def get_1m_bars_for_3m_bar(self, bar_3m):
|
||
"""获取当前3分钟K线对应的3根1分钟K线"""
|
||
try:
|
||
start_ts = int(bar_3m["id"])
|
||
end_ts = start_ts + 3 * 60
|
||
response = self.contractAPI.get_kline(
|
||
contract_symbol=self.contract_symbol,
|
||
step=1,
|
||
start_time=start_ts,
|
||
end_time=end_ts,
|
||
)[0]
|
||
if response.get("code") != 1000:
|
||
return []
|
||
data = response.get("data", [])
|
||
out = []
|
||
for k in data:
|
||
out.append(
|
||
{
|
||
"id": int(k["timestamp"]),
|
||
"open": float(k["open_price"]),
|
||
"high": float(k["high_price"]),
|
||
"low": float(k["low_price"]),
|
||
"close": float(k["close_price"]),
|
||
}
|
||
)
|
||
out.sort(key=lambda x: x["id"])
|
||
return out
|
||
except Exception as e:
|
||
logger.warning(f"获取1分钟K线失败: {e}")
|
||
return []
|
||
|
||
def determine_trigger_order_by_1m(self, bars_1m, long_trigger, short_trigger):
|
||
"""同根K线多空都触及时，用1分钟K线判断先后"""
|
||
if not bars_1m:
|
||
return None
|
||
for bar in bars_1m:
|
||
high = float(bar["high"])
|
||
low = float(bar["low"])
|
||
open_price = float(bar["open"])
|
||
long_ok = high >= long_trigger
|
||
short_ok = low <= short_trigger
|
||
if long_ok and not short_ok:
|
||
return "long"
|
||
if short_ok and not long_ok:
|
||
return "short"
|
||
if long_ok and short_ok:
|
||
d_long = abs(long_trigger - open_price)
|
||
d_short = abs(short_trigger - open_price)
|
||
return "short" if d_short < d_long else "long"
|
||
return None
|
||
|
||
def check_trigger(self, kline_data):
|
||
"""
|
||
检测当前K线是否触发信号（1111：当前开盘价±实体/5）
|
||
返回：(方向, 触发价, 有效前一根, 当前K线) 或 (None,...)
|
||
"""
|
||
if len(kline_data) < 2:
|
||
return None, None, None, None
|
||
|
||
curr = kline_data[-1]
|
||
curr_kline_id = curr["id"]
|
||
curr_open = curr["open"]
|
||
_, prev = self.find_valid_prev_bar(kline_data, len(kline_data) - 1)
|
||
if prev is None:
|
||
return None, None, None, None
|
||
|
||
long_trigger, short_trigger = self.get_trigger_levels(prev, curr_open)
|
||
if long_trigger is None:
|
||
return None, None, None, None
|
||
|
||
c_high = float(curr["high"])
|
||
c_low = float(curr["low"])
|
||
long_triggered = c_high >= long_trigger
|
||
short_triggered = c_low <= short_trigger
|
||
|
||
direction = None
|
||
trigger_price = None
|
||
|
||
if long_triggered and short_triggered:
|
||
bars_1m = self.get_1m_bars_for_3m_bar(curr)
|
||
if bars_1m:
|
||
direction = self.determine_trigger_order_by_1m(
|
||
bars_1m, long_trigger, short_trigger
|
||
)
|
||
trigger_price = long_trigger if direction == "long" else short_trigger
|
||
if direction is None:
|
||
c_open_f = float(curr["open"])
|
||
d_long = abs(long_trigger - c_open_f)
|
||
d_short = abs(short_trigger - c_open_f)
|
||
direction = "short" if d_short <= d_long else "long"
|
||
trigger_price = long_trigger if direction == "long" else short_trigger
|
||
elif short_triggered:
|
||
direction = "short"
|
||
trigger_price = short_trigger
|
||
elif long_triggered:
|
||
direction = "long"
|
||
trigger_price = long_trigger
|
||
|
||
if direction is None:
|
||
return None, None, None, None
|
||
if (
|
||
self.last_trigger_kline_id == curr_kline_id
|
||
and self.last_trigger_direction == direction
|
||
):
|
||
return None, None, None, None
|
||
|
||
return direction, trigger_price, prev, curr
|
||
|
||
# ==================== BitMart API ====================
|
||
|
||
def get_klines(self):
|
||
try:
|
||
end_time = int(time.time())
|
||
response = self.contractAPI.get_kline(
|
||
contract_symbol=self.contract_symbol,
|
||
step=self.kline_step,
|
||
start_time=end_time - 3600 * 3,
|
||
end_time=end_time,
|
||
)[0]["data"]
|
||
formatted = []
|
||
for k in response:
|
||
formatted.append(
|
||
{
|
||
"id": int(k["timestamp"]),
|
||
"open": float(k["open_price"]),
|
||
"high": float(k["high_price"]),
|
||
"low": float(k["low_price"]),
|
||
"close": float(k["close_price"]),
|
||
}
|
||
)
|
||
formatted.sort(key=lambda x: x["id"])
|
||
return formatted
|
||
except Exception as e:
|
||
if "429" in str(e) or "too many requests" in str(e).lower():
|
||
logger.warning(f"API限流，等待60秒: {e}")
|
||
time.sleep(60)
|
||
else:
|
||
logger.error(f"获取K线异常: {e}")
|
||
self.ding("获取K线异常", error=True)
|
||
return None
|
||
|
||
def get_available_balance(self):
|
||
try:
|
||
response = self.contractAPI.get_assets_detail()[0]
|
||
if response["code"] == 1000:
|
||
data = response["data"]
|
||
if isinstance(data, dict):
|
||
return float(data.get("available_balance", 0))
|
||
for asset in data:
|
||
if asset.get("currency") == "USDT":
|
||
return float(asset.get("available_balance", 0))
|
||
return None
|
||
except Exception as e:
|
||
logger.error(f"余额查询异常: {e}")
|
||
return None
|
||
|
||
def get_position_status(self):
|
||
try:
|
||
response = self.contractAPI.get_position(
|
||
contract_symbol=self.contract_symbol
|
||
)[0]
|
||
if response["code"] == 1000:
|
||
positions = response["data"]
|
||
if not positions:
|
||
self.start = 0
|
||
self.open_avg_price = None
|
||
self.current_amount = None
|
||
self.position_cross = None
|
||
return True
|
||
self.start = 1 if positions[0]["position_type"] == 1 else -1
|
||
self.open_avg_price = positions[0]["open_avg_price"]
|
||
self.current_amount = positions[0]["current_amount"]
|
||
self.position_cross = positions[0]["position_cross"]
|
||
return True
|
||
return False
|
||
except Exception as e:
|
||
logger.error(f"持仓查询异常: {e}")
|
||
return False
|
||
|
||
def set_leverage(self):
|
||
try:
|
||
response = self.contractAPI.post_submit_leverage(
|
||
contract_symbol=self.contract_symbol,
|
||
leverage=self.leverage,
|
||
open_type=self.open_type,
|
||
)[0]
|
||
if response["code"] == 1000:
|
||
logger.success(f"全仓 {self.leverage}x 杠杆设置成功")
|
||
return True
|
||
logger.error(f"杠杆设置失败: {response}")
|
||
return False
|
||
except Exception as e:
|
||
logger.error(f"设置杠杆异常: {e}")
|
||
return False
|
||
|
||
# ==================== 浏览器 ====================
|
||
|
||
def _open_browser(self):
|
||
try:
|
||
bit_port = openBrowser(id=self.bit_id)
|
||
co = ChromiumOptions()
|
||
co.set_local_port(port=bit_port)
|
||
self.page = ChromiumPage(addr_or_opts=co)
|
||
return True
|
||
except Exception:
|
||
return False
|
||
|
||
def close_extra_tabs(self):
|
||
try:
|
||
for idx, tab in enumerate(self.page.get_tabs()):
|
||
if idx > 0:
|
||
tab.close()
|
||
return True
|
||
except Exception:
|
||
return False
|
||
|
||
def click_safe(self, xpath, sleep=0.5):
|
||
try:
|
||
ele = self.page.ele(xpath)
|
||
if not ele:
|
||
return False
|
||
ele.scroll.to_see(center=True)
|
||
time.sleep(sleep)
|
||
ele.click(by_js=True)
|
||
return True
|
||
except Exception:
|
||
return False
|
||
|
||
def 平仓(self):
|
||
logger.info("执行平仓...")
|
||
self.click_safe('x://span[normalize-space(text()) ="市价"]')
|
||
time.sleep(0.5)
|
||
self.ding("执行平仓操作")
|
||
|
||
def 开单(self, marketPriceLongOrder=0, size=None):
|
||
if size is None or size <= 0:
|
||
logger.warning("开单金额无效")
|
||
return False
|
||
direction_str = "做多" if marketPriceLongOrder == 1 else "做空"
|
||
logger.info(f"执行{direction_str}，金额: {size}")
|
||
size = max(1, min(25, int(size)))
|
||
try:
|
||
self.click_safe('x://button[normalize-space(text()) ="市价"]')
|
||
self.page.ele('x://*[@id="size_0"]').input(str(size))
|
||
if marketPriceLongOrder == -1:
|
||
self.click_safe('x://span[normalize-space(text()) ="卖出/做空"]')
|
||
else:
|
||
self.click_safe('x://span[normalize-space(text()) ="买入/做多"]')
|
||
self.ding(f"执行{direction_str}，金额: {size}")
|
||
return True
|
||
except Exception as e:
|
||
logger.error(f"开单异常: {e}")
|
||
return False
|
||
|
||
def ding(self, msg, error=False):
|
||
prefix = "❌五分之一：" if error else "🔔五分之一："
|
||
full_msg = f"{prefix}{msg}"
|
||
if error:
|
||
logger.error(msg)
|
||
for _ in range(3):
|
||
ding_executor.submit(self._send_ding_safe, full_msg)
|
||
else:
|
||
logger.info(msg)
|
||
ding_executor.submit(self._send_ding_safe, full_msg)
|
||
|
||
def _send_ding_safe(self, msg):
|
||
try:
|
||
send_dingtalk_message(msg)
|
||
except Exception as e:
|
||
logger.warning(f"消息发送失败: {e}")
|
||
|
||
def _send_position_message(self, latest_kline):
|
||
current_price = float(latest_kline["close"])
|
||
balance = self.get_available_balance()
|
||
self.balance = balance if balance is not None else 0.0
|
||
if self.start != 0:
|
||
open_avg_price = float(self.open_avg_price)
|
||
current_amount = float(self.current_amount)
|
||
position_cross = float(getattr(self, "position_cross", 0) or 0)
|
||
if self.start == 1:
|
||
unrealized_pnl = current_amount * 0.001 * (
|
||
current_price - open_avg_price
|
||
)
|
||
else:
|
||
unrealized_pnl = current_amount * 0.001 * (
|
||
open_avg_price - current_price
|
||
)
|
||
pnl_rate = (
|
||
(current_price - open_avg_price) / open_avg_price * 100
|
||
if self.start == 1
|
||
else (open_avg_price - current_price) / open_avg_price * 100
|
||
)
|
||
direction_str = "空" if self.start == -1 else "多"
|
||
msg = (
|
||
f"【五分之一 {self.contract_symbol}】\n"
|
||
f"方向：{direction_str}\n"
|
||
f"现价：{current_price:.2f}\n"
|
||
f"开仓均价：{open_avg_price:.2f}\n"
|
||
f"浮动盈亏：{unrealized_pnl:+.2f} USDT ({pnl_rate:+.2f}%)\n"
|
||
f"余额：{self.balance:.2f}"
|
||
)
|
||
else:
|
||
msg = (
|
||
f"【五分之一 {self.contract_symbol}】\n"
|
||
f"方向：无\n"
|
||
f"现价：{current_price:.2f}\n"
|
||
f"余额：{self.balance:.2f}"
|
||
)
|
||
self.ding(msg)
|
||
|
||
# ==================== 主循环 ====================
|
||
|
||
def action(self):
|
||
if not self.set_leverage():
|
||
logger.error("杠杆设置失败")
|
||
return
|
||
if not self._open_browser():
|
||
self.ding("打开浏览器失败！", error=True)
|
||
return
|
||
logger.info("浏览器打开成功")
|
||
self.close_extra_tabs()
|
||
self.page.get("https://derivatives.bitmart.com/zh-CN/futures/ETHUSDT")
|
||
time.sleep(2)
|
||
self.click_safe('x://button[normalize-space(text()) ="市价"]')
|
||
logger.info(
|
||
f"五分之一策略（3分钟K线）开始监测，间隔: {self.check_interval}秒"
|
||
)
|
||
|
||
last_report_time = 0
|
||
report_interval = 300
|
||
|
||
while True:
|
||
for _ in range(5):
|
||
if self._open_browser():
|
||
break
|
||
time.sleep(5)
|
||
else:
|
||
self.ding("打开浏览器失败！", error=True)
|
||
return
|
||
self.close_extra_tabs()
|
||
self.page.get("https://derivatives.bitmart.com/zh-CN/futures/ETHUSDT")
|
||
time.sleep(2)
|
||
self.click_safe('x://button[normalize-space(text()) ="市价"]')
|
||
|
||
try:
|
||
kline_data = self.get_klines()
|
||
if not kline_data or len(kline_data) < 3:
|
||
logger.warning("K线数据不足...")
|
||
time.sleep(self.check_interval)
|
||
continue
|
||
|
||
curr = kline_data[-1]
|
||
if not self.get_position_status():
|
||
logger.warning("获取仓位失败，使用缓存")
|
||
|
||
direction, trigger_price, valid_prev, curr_kline = (
|
||
self.check_trigger(kline_data)
|
||
)
|
||
|
||
if direction:
|
||
curr_kline_id = curr_kline["id"]
|
||
if self.last_trade_kline_id == curr_kline_id:
|
||
self.last_trigger_kline_id = curr_kline_id
|
||
self.last_trigger_direction = direction
|
||
time.sleep(self.check_interval)
|
||
continue
|
||
|
||
if (direction == "long" and self.start == 1) or (
|
||
direction == "short" and self.start == -1
|
||
):
|
||
self.last_trigger_kline_id = curr_kline_id
|
||
self.last_trigger_direction = direction
|
||
time.sleep(self.check_interval)
|
||
continue
|
||
|
||
balance = self.get_available_balance()
|
||
trade_size = (balance or 0) * self.risk_percent
|
||
executed = False
|
||
|
||
if direction == "long":
|
||
if self.start == -1:
|
||
self.平仓()
|
||
time.sleep(1)
|
||
self.开单(marketPriceLongOrder=1, size=trade_size)
|
||
executed = True
|
||
elif self.start == 0:
|
||
self.开单(marketPriceLongOrder=1, size=trade_size)
|
||
executed = True
|
||
elif direction == "short":
|
||
if self.start == 1:
|
||
self.平仓()
|
||
time.sleep(1)
|
||
self.开单(marketPriceLongOrder=-1, size=trade_size)
|
||
executed = True
|
||
elif self.start == 0:
|
||
self.开单(marketPriceLongOrder=-1, size=trade_size)
|
||
executed = True
|
||
|
||
self.last_trigger_kline_id = curr_kline_id
|
||
self.last_trigger_direction = direction
|
||
if executed:
|
||
self.last_trade_kline_id = curr_kline_id
|
||
self.get_position_status()
|
||
self._send_position_message(curr_kline)
|
||
last_report_time = time.time()
|
||
|
||
if time.time() - last_report_time >= report_interval:
|
||
if self.get_position_status():
|
||
self._send_position_message(kline_data[-1])
|
||
last_report_time = time.time()
|
||
|
||
time.sleep(self.check_interval)
|
||
|
||
except Exception as e:
|
||
logger.error(f"主循环异常: {e}")
|
||
time.sleep(self.check_interval)
|
||
time.sleep(3)
|
||
if random.randint(1, 10) > 7:
|
||
self.page.close()
|
||
time.sleep(15)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
OpenBasedFifthStrategy(bit_id=BIT_ID).action()
|
||
except KeyboardInterrupt:
|
||
logger.info("程序被用户中断")
|
||
finally:
|
||
ding_executor.shutdown(wait=True)
|
||
logger.info("已退出")
|