import json
from datetime import datetime, timezone

data_path = "/home/aryy/.hermes/profiles/finance/backtests/xauusd_2026/hybrid/data"

def get_context(timestamp_str):
    target_dt = datetime.fromisoformat(timestamp_str.replace("Z", "+00:00"))
    target_ts = int(target_dt.timestamp())
    
    context = {}
    for tf in ["M5"]:
        with open(f"{data_path}/ohlcv_{tf}.json") as f:
            data = json.load(f)
            bars_data = data.get('bars', [])
            
        bars = []
        for b in bars_data:
            ts = b.get('time')
            if ts:
                try:
                    ts_val = int(datetime.fromisoformat(ts.replace("Z", "+00:00")).timestamp())
                    if ts_val <= target_ts:
                        bars.append(b)
                except Exception as e:
                    pass
        
        context[tf] = bars[-12:] if bars else []
        
    return context

items = [
  "2026-03-25T04:00:00Z",
  "2026-03-25T05:45:00Z",
  "2026-03-25T08:00:00Z"
]

for ts in items:
    print(f"\n--- Context for {ts} ---")
    ctx = get_context(ts)
    for tf, bars in ctx.items():
        print(f"  {tf}:")
        for b in bars:
             dt = b.get('time', 'unknown')
             print(f"    {dt} - O:{b.get('open')} H:{b.get('high')} L:{b.get('low')} C:{b.get('close')}")