#!/usr/bin/env python3
"""
测试最终修复的监控器查询
"""

import sys
import os
import datetime
sys.path.append(os.path.dirname(__file__))

from influx_service import InfluxService, InfluxConnectionParams
from config_model import AppConfig
from pathlib import Path

def test_current_time_query():
    """测试当前时间的查询"""
    print("🧪 测试当前时间查询")
    
    # 加载配置
    config_path = Path("default.json")
    config = AppConfig.load(config_path)
    
    # 创建InfluxDB服务
    params = InfluxConnectionParams(
        url=config.influx.url,
        org=config.influx.org,
        token=config.influx.token
    )
    
    service = InfluxService(params)
    
    bucket = getattr(config.influx, 'bucket', 'PCM')
    measurement = getattr(config.influx, 'measurement', 'experiment_status')
    fields = ['status']
    filters = {}
    
    # 模拟监控器的时间处理（当前时间）
    start_time = datetime.datetime.now()  # 无时区的当前时间
    
    print(f"原始时间: {start_time}")
    
    # 应用监控器的时间处理逻辑
    if hasattr(start_time, 'isoformat'):
        if start_time.tzinfo is None:
            start_time_utc = start_time.replace(tzinfo=datetime.timezone.utc)
        else:
            start_time_utc = start_time.astimezone(datetime.timezone.utc)
        start_iso = start_time_utc.isoformat()
    else:
        start_iso = "-1h"
    
    time_range = start_iso
    
    print(f"处理后时间范围: {time_range}")
    
    # 生成查询
    flux = service.build_flux(
        bucket=bucket,
        measurement=measurement,
        fields=fields,
        filters=filters,
        time_range=time_range
    )
    
    print("生成的查询:")
    for i, line in enumerate(flux.split('\n'), 1):
        if line.strip():
            print(f"  {i}: {line}")
    
    # 执行查询
    try:
        df = service.query(
            bucket=bucket,
            measurement=measurement,
            fields=fields,
            filters=filters,
            time_range=time_range
        )
        
        print(f"✅ 查询成功! 返回 {len(df)} 行数据")
        if not df.empty:
            print("数据预览:")
            print(df.head())
        
    except Exception as e:
        print(f"❌ 查询失败: {e}")
        
        # 尝试使用相对时间作为备选
        print("\n🔄 尝试使用相对时间...")
        try:
            df_fallback = service.query(
                bucket=bucket,
                measurement=measurement,
                fields=fields,
                filters=filters,
                time_range="-1h"
            )
            print(f"✅ 相对时间查询成功! 返回 {len(df_fallback)} 行数据")
        except Exception as e2:
            print(f"❌ 相对时间查询也失败: {e2}")

def main():
    print("监控器最终修复测试")
    print("=" * 30)
    test_current_time_query()

if __name__ == "__main__":
    main()