2569718930@qq.com
106 lines
3.4 KiB
Python
106 lines
3.4 KiB
Python
"""Market scan API service functions."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from inspect import Parameter, signature
|
|
from typing import Any, Dict
|
|
|
|
from fastapi import HTTPException, Request
|
|
from fastapi.concurrency import run_in_threadpool
|
|
|
|
import web.routes as legacy_routes
|
|
from web.services.request_timing import ServerTimingRecorder
|
|
|
|
|
|
def _supports_timing_recorder(func: Any) -> bool:
|
|
try:
|
|
params = signature(func).parameters.values()
|
|
except (TypeError, ValueError):
|
|
return True
|
|
return any(
|
|
param.name == "timing_recorder" or param.kind == Parameter.VAR_KEYWORD
|
|
for param in params
|
|
)
|
|
|
|
|
|
def _supports_keyword(func: Any, name: str) -> bool:
|
|
try:
|
|
params = signature(func).parameters.values()
|
|
except (TypeError, ValueError):
|
|
return True
|
|
return any(
|
|
param.name == name or param.kind == Parameter.VAR_KEYWORD
|
|
for param in params
|
|
)
|
|
|
|
|
|
async def get_scan_terminal_payload(
|
|
request: Request,
|
|
*,
|
|
scan_mode: str = "tradable",
|
|
min_price: float = 0.05,
|
|
max_price: float = 0.95,
|
|
min_edge_pct: float = 2.0,
|
|
min_liquidity: float = 500.0,
|
|
high_liquidity_only: bool = False,
|
|
market_type: str = "maxtemp",
|
|
time_range: str = "today",
|
|
limit: int = 25,
|
|
force_refresh: bool = False,
|
|
diff: bool = False,
|
|
since_snapshot_id: str | None = None,
|
|
region: str = "",
|
|
timezone_offset_seconds: int | None = None,
|
|
) -> Dict[str, Any]:
|
|
timer = ServerTimingRecorder(
|
|
request,
|
|
log_name="scan_terminal_timing",
|
|
prefix="scan_terminal",
|
|
state_attr="scan_terminal_server_timing",
|
|
)
|
|
outcome = "ok"
|
|
status_code = 200
|
|
try:
|
|
timer.measure("assert_entitlement", lambda: legacy_routes._assert_entitlement(request))
|
|
filters: Dict[str, Any] = {
|
|
"scan_mode": scan_mode,
|
|
"min_price": min_price,
|
|
"max_price": max_price,
|
|
"min_edge_pct": min_edge_pct,
|
|
"min_liquidity": min_liquidity,
|
|
"high_liquidity_only": high_liquidity_only,
|
|
"market_type": market_type,
|
|
"time_range": time_range,
|
|
"limit": limit,
|
|
}
|
|
if timezone_offset_seconds is not None:
|
|
filters["timezone_offset_seconds"] = timezone_offset_seconds
|
|
if region:
|
|
filters["trading_region"] = region
|
|
async def build_payload():
|
|
builder = legacy_routes.build_scan_terminal_payload
|
|
kwargs: Dict[str, Any] = {"force_refresh": force_refresh}
|
|
if _supports_keyword(builder, "diff"):
|
|
kwargs["diff"] = diff
|
|
if _supports_keyword(builder, "since_snapshot_id"):
|
|
kwargs["since_snapshot_id"] = since_snapshot_id
|
|
if _supports_timing_recorder(builder):
|
|
kwargs["timing_recorder"] = timer
|
|
return await run_in_threadpool(builder, filters, **kwargs)
|
|
|
|
return await timer.measure_async("build_payload", build_payload)
|
|
except HTTPException as exc:
|
|
outcome = f"http_{exc.status_code}"
|
|
status_code = exc.status_code
|
|
raise
|
|
except Exception:
|
|
outcome = "exception"
|
|
status_code = 500
|
|
raise
|
|
finally:
|
|
timer.finish(outcome=outcome, status_code=status_code)
|
|
|
|
|
|
async def get_scan_terminal_overview_payload(request: Request) -> Dict[str, Any]:
|
|
return {"overview": [], "available": False}
|