import sys
import os
from datetime import datetime

# Adjust path if needed to import TradingStats and TradingEngine
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

from src.stats.trading_stats import TradingStats
from src.trading.trading_engine import TradingEngine

def update_db_with_latest_mark_prices():
    engine = TradingEngine()
    stats = TradingStats()
    exchange_positions = engine.get_positions() or []
    print(f"Fetched {len(exchange_positions)} positions from exchange.")
    updated = 0
    for pos in exchange_positions:
        symbol = pos.get('symbol')
        mark_price = pos.get('markPrice') or pos.get('markPx')
        if symbol and mark_price is not None:
            # Find the open trade in DB
            open_positions = stats.get_open_positions()
            for db_pos in open_positions:
                if db_pos.get('symbol') == symbol:
                    lifecycle_id = db_pos.get('trade_lifecycle_id')
                    try:
                        stats.trade_manager.update_trade_market_data(
                            lifecycle_id,
                            mark_price=float(mark_price)
                        )
                        updated += 1
                        print(f"Updated {symbol} (Lifecycle: {lifecycle_id}) with mark_price={mark_price}")
                    except Exception as e:
                        print(f"Failed to update {symbol}: {e}")
    print(f"Updated {updated} positions in DB with latest mark prices.")

def print_open_positions_mark_prices():
    stats = TradingStats()
    open_positions = stats.get_open_positions()
    print(f"\nDB now has {len(open_positions)} open positions:")
    for pos in open_positions:
        symbol = pos.get('symbol', 'N/A')
        entry_price = pos.get('entry_price', 'N/A')
        mark_price = pos.get('mark_price', 'N/A')
        print(f"Symbol: {symbol}, Entry Price: {entry_price}, Mark Price: {mark_price}")

if __name__ == "__main__":
    update_db_with_latest_mark_prices()
    print_open_positions_mark_prices()