from analyze_trade_logs import TradeAnalyzer
import json
from datetime import datetime, timedelta

a = TradeAnalyzer()

# Charger tous les TRADE_OPEN
trade_open_map = {}
with open(a.trades_file, 'r', encoding='utf-8') as f:
    for line in f:
        try:
            entry = json.loads(line.strip())
            if entry['type'] == 'TRADE_OPEN':
                trade_id = entry.get('trade_id')
                trade_open_map[trade_id] = entry
        except:
            pass

print(f"Total TRADE_OPEN dans le fichier: {len(trade_open_map)}")
print(f"Exemples de trade_ids: {list(trade_open_map.keys())[:5]}")

# Charger les TRADE_CLOSE récents
cutoff_time = datetime.now() - timedelta(hours=48)
trades_close = []
with open(a.trades_file, 'r', encoding='utf-8') as f:
    for line in f:
        try:
            entry = json.loads(line.strip())
            if entry['type'] == 'TRADE_CLOSE':
                ts = datetime.fromisoformat(entry['timestamp'])
                if ts >= cutoff_time:
                    trades_close.append(entry)
        except:
            pass

print(f"\nTotal TRADE_CLOSE (48h): {len(trades_close)}")
if trades_close:
    sample_close = trades_close[0]
    print(f"Exemple TRADE_CLOSE trade_id: {sample_close.get('trade_id')}")
    print(f"Présent dans map? {sample_close.get('trade_id') in trade_open_map}")
    if sample_close.get('trade_id') in trade_open_map:
        print(f"Pattern du TRADE_OPEN: {trade_open_map[sample_close.get('trade_id')].get('pattern')}")
