Files
trex-core/trex_core/application/safety_sell.py
T
genichinandClaude Opus 4.8 7d727824dc feat(slot): BAT-030 — 자산 중심 cutover 완성 (dcy-020 BAT-δ, REQ-056·057·062·064·065)
M1 literal inactive 전면화: SlotState 5-state(idle/closed 별칭 제거),
마이그레이션 010(CHECK 교체+행 rewrite), ops_checks 를 transition_log 기반 재작성.
M2 집행 매도 수량 브로커 정본(D3): dispatch 매도 qty=broker hldg_qty,
미러 fallback 없음, 브로커 0 → skip+WARN, 리스트 기반 dispatch(head-block 방지).
M3 컬럼 drop + legacy pool 제거: 마이그레이션 011(trading_date·entry_count DROP),
repo 계약 instance 스코프 개편(집행 큐 포함), preallocation 폐기(데몬 7스레드),
admission 자산 중심 단일 경로(slot_id 정본), capacity 재정의(available=M−active),
daily_symbol_entries 코드 경로 제거.
M4 cutover 도구(D1·D4): trex-core-cutover(전체 rebuild, preflight/브로커 carry-over/
--allow-unmapped/--promote-unique/dry-run) + PG 리허설 테스트 8건.
M5 baseline 마감: data-model/runtime-flows/project-structure/interface-contract
자산 중심 최종 갱신 + run-trex-core-daemon 스킬 정합.

검증: 단위 367 passed + ruff clean; PG-gated 29 passed(010/011 round-trip·드리프트
가드·cutover 리허설); RabbitMQ 21 passed; run-the-app e2e(test DB+실 MQ, M=1):
reconcile 슬롯 생성→slot_id 매수→dedup→capacity_full→fill→already_active→
매도→inactive(자산 바인딩 보존)→영구 슬롯 재활성 전 구간 green.
실 cutover 는 운영자 실행(REL 단계, D4).

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-10 15:16:15 +09:00

289 lines
13 KiB
Python

"""Safety-sell trigger service (REQ-037, REQ-038, REQ-039 / BAT-011).
A single service that the `trex-safety-sell` daemon worker drives once per
tick. It evaluates three autonomous sell triggers on `holding` slots:
- forced_close_at expiry (REQ-037) — watchdog-gated
- stop-loss / take-profit price (REQ-038) — watchdog-gated, KIS polling
- market-close liquidation (REQ-039) — watchdog-INDEPENDENT, once/day
Watchdog gating: forced_close and price checks only act on slots whose
watchdog has expired (strategy module is absent). During the grace period
no expiry-gated trigger fires (REQ-034). Market-close liquidation ignores
the watchdog and the grace period entirely (REQ-039 AC-2/AC-6).
All sells go through `SlotService.sell()` (REQ-036), which is idempotent on
`pending_sell` — so concurrent/duplicate triggers never double-submit.
Each trigger is exception-isolated: a single slot's failure (KIS error,
transition race) is logged and the loop continues (REQ-024 degrade policy).
"""
from __future__ import annotations
import logging
from datetime import date as date_cls
from datetime import datetime, time as time_cls, timedelta, timezone
from trex_core.application.broker_port import BrokerPort
from trex_core.application.watchdog import WatchdogStore
from trex_core.domain.slot import SlotRepository, SlotService
from trex_core.domain.slot.models import Slot
from trex_core.runtime.config import SafetySellConfig
_log = logging.getLogger(__name__)
_KST = timezone(timedelta(hours=9))
# Conservative KIS current-price quota guard (REQ-038 AC-8). When the
# estimated hourly polling volume crosses this, we log a warning so the
# operator can widen TREX_STOP_LOSS_POLL_INTERVAL.
_KIS_PRICE_QUOTA_PER_HOUR = 1000
_ACTOR = "trex-safety-sell"
def _parse_kst_time(spec: str) -> time_cls:
parts = spec.split(":")
return time_cls(hour=int(parts[0]), minute=int(parts[1]) if len(parts) > 1 else 0)
class SafetySellService:
"""Evaluates safety-sell triggers for one daemon tick (REQ-037/038/039)."""
def __init__(
self,
*,
slot_repo: SlotRepository,
slot_service: SlotService,
broker: BrokerPort,
watchdog: WatchdogStore,
config: SafetySellConfig,
instance_key: str = "default",
) -> None:
self._repo = slot_repo
self._svc = slot_service
self._broker = broker
self._watchdog = watchdog
self._cfg = config
self._instance_key = instance_key # dcy-017 FR-5: multi-instance label
self._last_liquidation_date: date_cls | None = None
self._last_forced_close_at: datetime | None = None
self._last_price_check_at: datetime | None = None
# BAT-014 / REQ-052: per-slot retry cooldown for persistent rejection.
self._last_attempt: dict = {} # slot_id -> last sell-attempt datetime
self._attempt_count: dict = {} # slot_id -> consecutive attempt count
# dcy-017 / REQ-037 (issue #12): watchdog-ownership visibility.
self._owned: set = set() # holding slot_ids currently watchdog-owned
self.counters: dict[str, int] = {} # phase/event counters (in-process metrics)
# ----------------------------------------------- dcy-017 visibility helpers
def _count(self, key: str) -> None:
self.counters[key] = self.counters.get(key, 0) + 1
# WARN-level fire reasons (attention events); others log at INFO.
_WARN_REASONS = frozenset({"forced_close_at", "stop_loss"})
def _log_fire(self, slot: Slot, reason: str) -> None:
"""Visibility for an actual safety-sell fire (dcy-017 FR-1, OQ-2)."""
self._count(f"fire:{reason}")
level = logging.WARNING if reason in self._WARN_REASONS else logging.INFO
_log.log(
level,
"safety-sell fired: phase=%s slot=%s portfolio_id=%s reason=%s instance_key=%s",
reason, slot.slot_id, slot.portfolio_id, reason, self._instance_key,
)
def _update_ownership(self, now: datetime) -> None:
"""Log watchdog-ownership entry (strategy absent) / exit (returned).
dcy-017 FR-1: WARN on entry (a holding slot's strategy went silent so the
safety net now owns it), INFO on exit. Counters track both. Best-effort —
never raises into the tick (NFR-1)."""
try:
if self._in_grace():
return # no expiry judgement during grace (REQ-034)
current = {
s.slot_id for s in self._holding(now) if self._watchdog_expired(s)
}
by_id = {s.slot_id: s for s in self._holding(now)}
for sid in current - self._owned:
self._count("owned:enter")
s = by_id.get(sid)
_log.warning(
"watchdog ownership entered (strategy absent): slot=%s "
"portfolio_id=%s phase=watchdog_owned instance_key=%s",
sid, (s.portfolio_id if s else None), self._instance_key,
)
for sid in self._owned - current:
self._count("owned:exit")
_log.info(
"watchdog ownership released (strategy returned): slot=%s "
"phase=watchdog_released instance_key=%s",
sid, self._instance_key,
)
self._owned = current
except Exception: # noqa: BLE001 - visibility must not break the tick
return
@property
def base_tick_sec(self) -> float:
"""Worker loop sleep — the finest of the configured check cadences."""
return float(max(
1,
min(self._cfg.forced_close_check_interval_sec,
self._cfg.stop_loss_poll_interval_sec),
))
# ---------------------------------------------------------------- helpers
def _holding(self, now: datetime) -> list[Slot]:
# dcy-020 (BAT-030): permanent asset slots — instance-scoped holding
# sweep (the day-scoped pool listing was removed with trading_date).
return self._repo.list_holding_slots(self._instance_key)
def _watchdog_expired(self, slot: Slot) -> bool:
return self._watchdog.is_expired(str(slot.slot_id), self._cfg.watchdog_timeout_sec)
def _in_grace(self) -> bool:
return self._watchdog.in_grace_period(self._cfg.watchdog_grace_period_sec)
def _safe_sell(self, slot: Slot, reason: str) -> bool:
try:
self._svc.sell(slot.slot_id, actor=_ACTOR, reason=reason)
return True
except Exception: # noqa: BLE001 - one slot's failure must not stop the loop
_log.warning("safety-sell failed (slot=%s reason=%s)",
slot.slot_id, reason, exc_info=True)
return False
# ------------------------------------------------ REQ-052 retry cooldown
def _in_cooldown(self, slot_id, now: datetime) -> bool:
"""True if the slot was attempted within min_retry_interval_sec (skip retry)."""
last = self._last_attempt.get(slot_id)
if last is None:
return False
return (now - last).total_seconds() < self._cfg.min_retry_interval_sec
def _attempt_sell(self, slot: Slot, reason: str, now: datetime) -> bool:
"""Record a sell attempt (cooldown already cleared by caller) + warn on repeat.
Updates last_attempt/attempt_count regardless of success (dcy-010 FR-2).
"""
sid = slot.slot_id
count = self._attempt_count.get(sid, 0) + 1
if count >= self._cfg.max_retry_warn_count:
_log.warning(
"safety-sell repeated rejection: slot=%s count=%d reason=%s last_attempt_at=%s",
sid, count, reason, self._last_attempt.get(sid),
)
self._last_attempt[sid] = now
self._attempt_count[sid] = count
return self._safe_sell(slot, reason)
def _prune_cooldown(self, now: datetime) -> None:
"""Drop cooldown state for slots no longer holding (dcy-010 FR-4)."""
holding_ids = {s.slot_id for s in self._holding(now)}
for sid in list(self._last_attempt):
if sid not in holding_ids:
self._last_attempt.pop(sid, None)
self._attempt_count.pop(sid, None)
def estimated_hourly_kis_calls(self, n_slots: int) -> int:
"""REQ-038 AC-8: worst-case current-price polls per hour for n_slots."""
interval = max(1, self._cfg.stop_loss_poll_interval_sec)
return n_slots * (3600 // interval)
# ---------------------------------------------------- REQ-037 forced_close
def check_forced_close(self, now: datetime) -> list:
"""Sell holding+watchdog-expired slots whose forced_close_at has passed."""
if self._in_grace():
return []
fired = []
for s in self._holding(now):
if s.forced_close_at is None:
continue
if not self._watchdog_expired(s):
continue # strategy module active — do not interfere (AC-3)
if s.forced_close_at > now:
continue
if self._in_cooldown(s.slot_id, now):
continue # REQ-052: persistent-rejection backoff
if self._attempt_sell(s, "forced_close_at", now):
self._log_fire(s, "forced_close_at")
fired.append(s.slot_id)
return fired
# ----------------------------------------------- REQ-038 stop-loss/take-profit
def check_prices(self, now: datetime) -> list:
"""Poll KIS current price for eligible slots and trigger SL/TP sells."""
if self._in_grace():
return []
candidates = [
s for s in self._holding(now)
if (s.stop_loss_price is not None or s.take_profit_price is not None)
and self._watchdog_expired(s) # no KIS call for watchdog-active (AC-4)
]
if self.estimated_hourly_kis_calls(len(candidates)) > _KIS_PRICE_QUOTA_PER_HOUR:
_log.warning(
"safety-sell KIS price polling may exceed quota: ~%d calls/hour "
"(%d slots, interval=%ds). Consider widening TREX_STOP_LOSS_POLL_INTERVAL.",
self.estimated_hourly_kis_calls(len(candidates)),
len(candidates), self._cfg.stop_loss_poll_interval_sec,
)
fired = []
for s in candidates:
if self._in_cooldown(s.slot_id, now):
continue # REQ-052: skip KIS poll + retry during cooldown
price = self._broker.get_current_price(s.symbol)
if price is None:
continue # unavailable — skip, retry next tick (no false trigger)
if s.stop_loss_price is not None and price <= s.stop_loss_price:
if self._attempt_sell(s, "stop_loss", now):
self._log_fire(s, "stop_loss")
fired.append(s.slot_id)
elif s.take_profit_price is not None and price >= s.take_profit_price:
if self._attempt_sell(s, "take_profit", now):
self._log_fire(s, "take_profit")
fired.append(s.slot_id)
return fired
# ------------------------------------------------ REQ-039 market-close liquidation
def check_market_close(self, now: datetime) -> list:
"""Liquidate ALL holding slots at the configured KST time, once per day."""
spec = self._cfg.market_close_liquidation_time
if not spec:
return [] # feature disabled (AC-3)
now_kst = now.astimezone(_KST)
today = now_kst.date()
if self._last_liquidation_date == today:
return [] # already fired this trading day (AC-4)
if now_kst.time() < _parse_kst_time(spec):
return [] # not yet
fired = []
for s in self._holding(now): # watchdog/grace ignored (AC-2/AC-6)
if self._safe_sell(s, "market_close_liquidation"):
self._log_fire(s, "market_close_liquidation")
fired.append(s.slot_id)
self._last_liquidation_date = today
return fired
# ----------------------------------------------------------------- run_once
def run_once(self, now: datetime | None = None) -> None:
"""One worker tick — market-close, then cadence-gated forced_close + prices."""
now = now or datetime.now(tz=timezone.utc)
self._prune_cooldown(now) # REQ-052 FR-4: drop state for non-holding slots
self._update_ownership(now) # dcy-017 FR-1: watchdog ownership enter/exit visibility
try:
self.check_market_close(now)
except Exception: # noqa: BLE001
_log.warning("market-close liquidation tick failed", exc_info=True)
if self._due(self._last_forced_close_at, now, self._cfg.forced_close_check_interval_sec):
self._last_forced_close_at = now
self.check_forced_close(now)
if self._due(self._last_price_check_at, now, self._cfg.stop_loss_poll_interval_sec):
self._last_price_check_at = now
self.check_prices(now)
@staticmethod
def _due(last: datetime | None, now: datetime, interval_sec: int) -> bool:
if last is None:
return True
return (now - last).total_seconds() >= interval_sec