OPS Live Trading Operations

from datetime import datetime

class DailyChecklist:
    """Automated daily operations checklist."""
    
    def __init__(self, engine, connector):
        self.engine = engine
        self.connector = connector
    
    def pre_session_check(self):
        """Run before trading session."""
        checks = {}
        
        # 1. Connection
        checks['connection'] = self.connector.is_connected()
        
        # 2. Account balance
        account = self.connector.get_account_info()
        checks['balance_ok'] = account.balance > 0
        checks['margin_level'] = account.margin_level
        
        # 3. All positions have SL/TP
        positions = self.connector.get_positions()
        checks['all_sl_set'] = all(p.sl > 0 for p in positions)
        checks['all_tp_set'] = all(p.tp > 0 for p in positions)
        
        # 4. Risk state
        dd_pct = (account.starting_balance - account.equity) / account.starting_balance * 100
        checks['dd_zone'] = 'NORMAL' if dd_pct < 4 else 'CAUTION' if dd_pct < 6 else 'DANGER'
        
        # 5. Model loaded
        checks['models_loaded'] = self.engine.models_loaded
        
        # Report
        print(f"=== PRE-SESSION CHECK ({datetime.now():%Y-%m-%d %H:%M}) ===")
        all_ok = True
        for check, value in checks.items():
            status = "[PASS]" if value not in [False, 'DANGER', 'CRITICAL'] else "[FAIL]"
            if value == 'CAUTION':
                status = "[WARN]"
            print(f"  {status} {check}: {value}")
            if status == "[FAIL]":
                all_ok = False
        
        print(f"\n{'[PASS] CLEARED FOR TRADING' if all_ok else '[FAIL] ISSUES DETECTED — RESOLVE BEFORE TRADING'}")
        return all_ok