dekun
469 lines
17 KiB
Python
469 lines
17 KiB
Python
"""OKX 期权模块：Flask 路由注册。"""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import threading
|
|
from typing import Any
|
|
|
|
from flask import Flask, jsonify, redirect, request, url_for
|
|
from jinja2 import ChoiceLoader, FileSystemLoader
|
|
|
|
from lib.options.options_db import init_options_tables
|
|
from lib.options.options_monitor_lib import options_monitor_loop
|
|
from lib.options.options_pricing_lib import (
|
|
calc_order_size,
|
|
ct_mult_from_meta,
|
|
min_sz_from_meta,
|
|
premium_per_sheet,
|
|
total_premium,
|
|
)
|
|
|
|
|
|
def _env_bool(key: str, default: bool = False) -> bool:
|
|
raw = (os.getenv(key) or "").strip().lower()
|
|
if not raw:
|
|
return default
|
|
return raw in ("1", "true", "yes", "on")
|
|
|
|
|
|
def _env_float(key: str, default: float) -> float:
|
|
try:
|
|
return float(os.getenv(key, str(default)))
|
|
except (TypeError, ValueError):
|
|
return default
|
|
|
|
|
|
def attach_options_templates(app: Flask, repo_root: str) -> None:
|
|
tpl_dir = os.path.join(repo_root, "lib", "options", "templates")
|
|
if not os.path.isdir(tpl_dir):
|
|
return
|
|
existing = app.jinja_loader
|
|
loaders = [FileSystemLoader(tpl_dir)]
|
|
if existing is not None:
|
|
if isinstance(existing, ChoiceLoader):
|
|
loaders = list(existing.loaders) + loaders
|
|
else:
|
|
loaders.insert(0, existing)
|
|
app.jinja_loader = ChoiceLoader(loaders)
|
|
|
|
|
|
def install_options_trading(app: Flask, repo_root: str, app_module: Any) -> None:
|
|
enabled = _env_bool("OKX_OPTIONS_ENABLED", False)
|
|
attach_options_templates(app, repo_root)
|
|
cfg = _build_cfg(app_module)
|
|
app.extensions["options_cfg"] = cfg
|
|
register_options_routes(app, cfg)
|
|
if enabled:
|
|
_start_monitor_thread(app, cfg)
|
|
|
|
|
|
def _build_cfg(app_module: Any) -> dict[str, Any]:
|
|
from lib.exchange.okx_options_lib import (
|
|
build_option_chain,
|
|
estimate_usdt_to_usdc,
|
|
execute_convert,
|
|
fetch_option_positions,
|
|
fetch_options_balances,
|
|
format_position_row,
|
|
options_api_ready,
|
|
place_option_limit_order,
|
|
quote_option_contract,
|
|
transfer_ccy,
|
|
)
|
|
|
|
return {
|
|
"enabled": _env_bool("OKX_OPTIONS_ENABLED", False),
|
|
"get_db": app_module.get_db,
|
|
"login_required": app_module.login_required,
|
|
"exchange_options": getattr(app_module, "exchange_options", None),
|
|
"send_wechat": app_module.send_wechat_msg,
|
|
"render_main_page": app_module.render_main_page,
|
|
"trade_budget": _env_float("OKX_OPTIONS_TRADE_BUDGET_USDC", 10.0),
|
|
"budget_buffer": _env_float("OKX_OPTIONS_BUDGET_BUFFER", 0.95),
|
|
"default_underly": (os.getenv("OKX_OPTIONS_DEFAULT_UNDERLY") or "ETH").strip().upper(),
|
|
"max_dte_days": _env_float("OKX_OPTIONS_MAX_DTE_DAYS", 2.0),
|
|
"itm_max_dist": _env_float("OKX_OPTIONS_ITM_MAX_DIST_USD", 30.0),
|
|
"td_mode": (os.getenv("OKX_OPTIONS_TD_MODE") or "cross").strip(),
|
|
"allow_market_close": _env_bool("OKX_OPTIONS_ALLOW_MARKET_CLOSE", False),
|
|
"profit_ratio": _env_float("OKX_OPTIONS_PROFIT_ALERT_RATIO", 1.0),
|
|
"poll_seconds": _env_float("OKX_OPTIONS_POLL_SECONDS", 15.0),
|
|
"account_label": (os.getenv("OKX_OPTIONS_ACCOUNT_LABEL") or "OKX期权").strip(),
|
|
"build_option_chain": build_option_chain,
|
|
"quote_option_contract": quote_option_contract,
|
|
"place_option_limit_order": place_option_limit_order,
|
|
"fetch_option_positions": fetch_option_positions,
|
|
"fetch_options_balances": fetch_options_balances,
|
|
"format_position_row": format_position_row,
|
|
"estimate_usdt_to_usdc": estimate_usdt_to_usdc,
|
|
"execute_convert": execute_convert,
|
|
"transfer_ccy": transfer_ccy,
|
|
"options_api_ready": options_api_ready,
|
|
}
|
|
|
|
|
|
def _require_options_ex(cfg: dict[str, Any]):
|
|
if not cfg.get("enabled"):
|
|
return None, "期权模块未启用，请在 .env 设置 OKX_OPTIONS_ENABLED=true 并重启 PM2"
|
|
ex = cfg.get("exchange_options")
|
|
ok, reason = cfg["options_api_ready"](ex)
|
|
if not ok:
|
|
return None, reason or "期权 API 未配置"
|
|
return ex, ""
|
|
|
|
|
|
def register_options_routes(app: Flask, cfg: dict[str, Any]) -> None:
|
|
lr = cfg["login_required"]
|
|
|
|
@app.route("/api/options/balances")
|
|
@lr
|
|
def api_options_balances():
|
|
ex, err = _require_options_ex(cfg)
|
|
if ex is None:
|
|
return jsonify({"ok": False, "msg": err})
|
|
bal = cfg["fetch_options_balances"](ex)
|
|
return jsonify({"ok": True, **bal, "trade_budget": cfg["trade_budget"]})
|
|
|
|
@app.route("/api/options/chain")
|
|
@lr
|
|
def api_options_chain():
|
|
ex, err = _require_options_ex(cfg)
|
|
if ex is None:
|
|
return jsonify({"ok": False, "msg": err})
|
|
u = (request.args.get("underlying") or cfg["default_underly"]).upper()
|
|
chain = cfg["build_option_chain"](
|
|
ex,
|
|
u,
|
|
max_dte_days=cfg["max_dte_days"],
|
|
itm_only=True,
|
|
itm_max_dist_usd=cfg["itm_max_dist"],
|
|
)
|
|
return jsonify({"ok": True, **chain})
|
|
|
|
@app.route("/api/options/quote")
|
|
@lr
|
|
def api_options_quote():
|
|
ex, err = _require_options_ex(cfg)
|
|
if ex is None:
|
|
return jsonify({"ok": False, "msg": err})
|
|
inst_id = (request.args.get("inst_id") or "").strip()
|
|
if not inst_id:
|
|
return jsonify({"ok": False, "msg": "缺少 inst_id"})
|
|
q = cfg["quote_option_contract"](ex, inst_id)
|
|
if not q.get("ok"):
|
|
return jsonify(q)
|
|
ask = q.get("ask")
|
|
ct_mult = q.get("ct_mult") or 0.01
|
|
min_sz = q.get("min_sz") or 1
|
|
mode = (request.args.get("mode") or "budget_full").strip()
|
|
budget = cfg["trade_budget"]
|
|
eth_amount = None
|
|
try:
|
|
if request.args.get("eth_amount"):
|
|
eth_amount = float(request.args.get("eth_amount"))
|
|
except (TypeError, ValueError):
|
|
pass
|
|
if ask is None or ask <= 0:
|
|
return jsonify({**q, "ok": False, "msg": "暂无卖一价"})
|
|
sizing = calc_order_size(
|
|
quote_per_unit=float(ask),
|
|
ct_mult=float(ct_mult),
|
|
min_sz=int(min_sz),
|
|
budget_usdc=budget if mode != "eth_amount" else None,
|
|
budget_buffer=cfg["budget_buffer"],
|
|
eth_amount=eth_amount if mode == "eth_amount" else None,
|
|
budget_cap=cfg["trade_budget"],
|
|
)
|
|
return jsonify(
|
|
{
|
|
**q,
|
|
"quote_per_unit": ask,
|
|
"premium_per_sheet": premium_per_sheet(float(ask), float(ct_mult)),
|
|
"sizing": sizing,
|
|
}
|
|
)
|
|
|
|
@app.route("/api/options/open", methods=["POST"])
|
|
@lr
|
|
def api_options_open():
|
|
ex, err = _require_options_ex(cfg)
|
|
if ex is None:
|
|
return jsonify({"ok": False, "msg": err})
|
|
data = request.get_json(silent=True) or {}
|
|
inst_id = (data.get("inst_id") or "").strip()
|
|
mode = (data.get("mode") or "budget_full").strip()
|
|
signal_note = (data.get("signal_note") or "").strip()
|
|
if not inst_id:
|
|
return jsonify({"ok": False, "msg": "缺少 inst_id"})
|
|
q = cfg["quote_option_contract"](ex, inst_id)
|
|
if not q.get("ok"):
|
|
return jsonify(q)
|
|
ask = q.get("ask")
|
|
if ask is None or ask <= 0:
|
|
return jsonify({"ok": False, "msg": "暂无卖一价，无法买入"})
|
|
ct_mult = float(q.get("ct_mult") or 0.01)
|
|
min_sz = int(q.get("min_sz") or 1)
|
|
eth_amount = None
|
|
if mode == "eth_amount":
|
|
try:
|
|
eth_amount = float(data.get("eth_amount"))
|
|
except (TypeError, ValueError):
|
|
return jsonify({"ok": False, "msg": "ETH 数量无效"})
|
|
sizing = calc_order_size(
|
|
quote_per_unit=float(ask),
|
|
ct_mult=ct_mult,
|
|
min_sz=min_sz,
|
|
budget_usdc=cfg["trade_budget"] if mode != "eth_amount" else None,
|
|
budget_buffer=cfg["budget_buffer"],
|
|
eth_amount=eth_amount,
|
|
budget_cap=cfg["trade_budget"],
|
|
)
|
|
if not sizing.get("ok"):
|
|
return jsonify({"ok": False, "msg": sizing.get("msg") or "张数计算失败", "sizing": sizing})
|
|
sheets = int(sizing["sheets"])
|
|
order = cfg["place_option_limit_order"](
|
|
ex,
|
|
inst_id=inst_id,
|
|
side="buy",
|
|
sheets=sheets,
|
|
price=float(ask),
|
|
td_mode=cfg["td_mode"],
|
|
)
|
|
if not order.get("ok"):
|
|
return jsonify(order)
|
|
conn = cfg["get_db"]()
|
|
try:
|
|
init_options_tables(conn)
|
|
meta = q.get("meta") or {}
|
|
u = str(meta.get("uly") or inst_id).split("-")[0]
|
|
conn.execute(
|
|
"""
|
|
INSERT INTO options_trades
|
|
(inst_id, underlying, opt_type, strike, exp_time, sheets, eth_amount,
|
|
open_quote, premium_paid, status, signal_note, exchange_ord_id)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, 'open', ?, ?)
|
|
""",
|
|
(
|
|
inst_id,
|
|
u,
|
|
meta.get("optType"),
|
|
q.get("strike"),
|
|
str(q.get("exp_time") or ""),
|
|
sheets,
|
|
sizing["eth_amount"],
|
|
float(ask),
|
|
sizing["total_premium"],
|
|
signal_note,
|
|
(order.get("data") or {}).get("ordId"),
|
|
),
|
|
)
|
|
conn.commit()
|
|
finally:
|
|
conn.close()
|
|
return jsonify({"ok": True, "order": order, "sizing": sizing})
|
|
|
|
@app.route("/api/options/positions")
|
|
@lr
|
|
def api_options_positions():
|
|
ex, err = _require_options_ex(cfg)
|
|
if ex is None:
|
|
return jsonify({"ok": False, "msg": err})
|
|
raw = cfg["fetch_option_positions"](ex)
|
|
rows = [cfg["format_position_row"](p) for p in raw]
|
|
return jsonify({"ok": True, "positions": rows})
|
|
|
|
@app.route("/api/options/close", methods=["POST"])
|
|
@lr
|
|
def api_options_close():
|
|
ex, err = _require_options_ex(cfg)
|
|
if ex is None:
|
|
return jsonify({"ok": False, "msg": err})
|
|
data = request.get_json(silent=True) or {}
|
|
inst_id = (data.get("inst_id") or "").strip()
|
|
use_market = bool(data.get("market")) and cfg["allow_market_close"]
|
|
if not inst_id:
|
|
return jsonify({"ok": False, "msg": "缺少 inst_id"})
|
|
sheets = data.get("sheets")
|
|
q = cfg["quote_option_contract"](ex, inst_id)
|
|
bid = q.get("bid")
|
|
if not use_market and (bid is None or bid <= 0):
|
|
return jsonify({"ok": False, "msg": "暂无买一价，无法限价平仓"})
|
|
raw_positions = cfg["fetch_option_positions"](ex)
|
|
pos = next((p for p in raw_positions if str(p.get("instId")) == inst_id), None)
|
|
if not pos:
|
|
return jsonify({"ok": False, "msg": "未找到持仓"})
|
|
avail = float(pos.get("availPos") or pos.get("pos") or 0)
|
|
close_sheets = int(sheets) if sheets else int(abs(avail))
|
|
if close_sheets < 1:
|
|
return jsonify({"ok": False, "msg": "可平张数不足"})
|
|
if use_market:
|
|
try:
|
|
resp = ex.private_post_trade_order(
|
|
{
|
|
"instId": inst_id,
|
|
"tdMode": cfg["td_mode"],
|
|
"side": "sell",
|
|
"ordType": "market",
|
|
"sz": str(close_sheets),
|
|
}
|
|
)
|
|
data_rows = (resp or {}).get("data") or []
|
|
if not data_rows or str(data_rows[0].get("sCode")) != "0":
|
|
return jsonify({"ok": False, "msg": data_rows[0].get("sMsg") if data_rows else "市价平仓失败"})
|
|
order = {"ok": True, "data": data_rows[0]}
|
|
except Exception as e:
|
|
return jsonify({"ok": False, "msg": str(e)})
|
|
else:
|
|
order = cfg["place_option_limit_order"](
|
|
ex,
|
|
inst_id=inst_id,
|
|
side="sell",
|
|
sheets=close_sheets,
|
|
price=float(bid),
|
|
td_mode=cfg["td_mode"],
|
|
)
|
|
if not order.get("ok"):
|
|
return jsonify(order)
|
|
prem_recv = total_premium(float(bid or 0), close_sheets * float(q.get("ct_mult") or 0.01))
|
|
conn = cfg["get_db"]()
|
|
try:
|
|
init_options_tables(conn)
|
|
row = conn.execute(
|
|
"SELECT id, premium_paid FROM options_trades WHERE inst_id = ? AND status = 'open' ORDER BY id DESC LIMIT 1",
|
|
(inst_id,),
|
|
).fetchone()
|
|
if row:
|
|
paid = float(row["premium_paid"] or 0)
|
|
pnl = prem_recv - paid
|
|
conn.execute(
|
|
"""
|
|
UPDATE options_trades
|
|
SET status = 'closed', close_quote = ?, premium_received = ?,
|
|
realized_pnl = ?, close_ord_id = ?, closed_at = CURRENT_TIMESTAMP
|
|
WHERE id = ?
|
|
""",
|
|
(
|
|
bid,
|
|
prem_recv,
|
|
pnl,
|
|
(order.get("data") or {}).get("ordId"),
|
|
int(row["id"]),
|
|
),
|
|
)
|
|
conn.commit()
|
|
finally:
|
|
conn.close()
|
|
return jsonify({"ok": True, "order": order, "bid": bid, "sheets": close_sheets})
|
|
|
|
@app.route("/api/options/convert/quote", methods=["POST"])
|
|
@lr
|
|
def api_options_convert_quote():
|
|
ex, err = _require_options_ex(cfg)
|
|
if ex is None:
|
|
return jsonify({"ok": False, "msg": err})
|
|
data = request.get_json(silent=True) or {}
|
|
try:
|
|
amount = float(data.get("amount"))
|
|
except (TypeError, ValueError):
|
|
return jsonify({"ok": False, "msg": "数量无效"})
|
|
return jsonify(cfg["estimate_usdt_to_usdc"](ex, amount))
|
|
|
|
@app.route("/api/options/convert/execute", methods=["POST"])
|
|
@lr
|
|
def api_options_convert_execute():
|
|
ex, err = _require_options_ex(cfg)
|
|
if ex is None:
|
|
return jsonify({"ok": False, "msg": err})
|
|
data = request.get_json(silent=True) or {}
|
|
quote_id = (data.get("quote_id") or "").strip()
|
|
result = cfg["execute_convert"](ex, quote_id)
|
|
if result.get("ok"):
|
|
conn = cfg["get_db"]()
|
|
try:
|
|
init_options_tables(conn)
|
|
conn.execute(
|
|
"""
|
|
INSERT INTO options_convert_log (from_ccy, to_ccy, rfq_sz, received_sz, quote_id, status, message)
|
|
VALUES ('USDT', 'USDC', ?, ?, ?, 'ok', '')
|
|
""",
|
|
(
|
|
data.get("rfq_sz"),
|
|
(result.get("data") or {}).get("baseSz"),
|
|
quote_id,
|
|
),
|
|
)
|
|
conn.commit()
|
|
finally:
|
|
conn.close()
|
|
return jsonify(result)
|
|
|
|
@app.route("/api/options/transfer", methods=["POST"])
|
|
@lr
|
|
def api_options_transfer():
|
|
ex, err = _require_options_ex(cfg)
|
|
if ex is None:
|
|
return jsonify({"ok": False, "msg": err})
|
|
data = request.get_json(silent=True) or {}
|
|
ccy = (data.get("ccy") or "USDC").upper()
|
|
from_acct = (data.get("from") or "funding").strip()
|
|
to_acct = (data.get("to") or "trading").strip()
|
|
try:
|
|
amount = float(data.get("amount"))
|
|
except (TypeError, ValueError):
|
|
return jsonify({"ok": False, "msg": "数量无效"})
|
|
result = cfg["transfer_ccy"](ex, ccy, amount, from_acct, to_acct)
|
|
if result.get("ok"):
|
|
conn = cfg["get_db"]()
|
|
try:
|
|
init_options_tables(conn)
|
|
conn.execute(
|
|
"""
|
|
INSERT INTO options_transfer_log (ccy, amount, from_account, to_account, status, message)
|
|
VALUES (?, ?, ?, ?, 'ok', '')
|
|
""",
|
|
(ccy, amount, from_acct, to_acct),
|
|
)
|
|
conn.commit()
|
|
finally:
|
|
conn.close()
|
|
return jsonify(result)
|
|
|
|
|
|
def _start_monitor_thread(app: Flask, cfg: dict[str, Any]) -> None:
|
|
if app.extensions.get("options_monitor_started"):
|
|
return
|
|
app.extensions["options_monitor_started"] = True
|
|
|
|
def _bid(inst_id: str) -> float | None:
|
|
ex = cfg.get("exchange_options")
|
|
if ex is None:
|
|
return None
|
|
try:
|
|
q = cfg["quote_option_contract"](ex, inst_id)
|
|
return q.get("bid")
|
|
except Exception:
|
|
return None
|
|
|
|
def _positions():
|
|
ex = cfg.get("exchange_options")
|
|
if ex is None:
|
|
return []
|
|
raw = cfg["fetch_option_positions"](ex)
|
|
return [cfg["format_position_row"](p) for p in raw]
|
|
|
|
t = threading.Thread(
|
|
target=options_monitor_loop,
|
|
kwargs={
|
|
"enabled": True,
|
|
"poll_seconds": cfg["poll_seconds"],
|
|
"get_db": cfg["get_db"],
|
|
"fetch_positions": _positions,
|
|
"ticker_bid_fn": _bid,
|
|
"send_wechat": cfg["send_wechat"],
|
|
"account_label": cfg["account_label"],
|
|
"profit_ratio": cfg["profit_ratio"],
|
|
},
|
|
daemon=True,
|
|
name="options-monitor",
|
|
)
|
|
t.start()
|