"""测试 Flask API 日期参数"""
import os
from datasource.flask_api_source import FlaskAPIDataSource

# 设置环境变量
os.environ['FLASK_API_URL'] = 'https://k3s.tokenpluse.xyz'

api = FlaskAPIDataSource()

# 测试获取 2020-2024 年的数据
print("测试 1: 获取 2020-2024 年数据")
df = api.fetch("399006.SZ", "2020-01-01", "2024-12-31")
if df is not None:
    print(f"  数据量: {len(df)}")
    print(f"  日期范围: {df.index[0]} ~ {df.index[-1]}")
else:
    print("  获取失败")

print("\n测试 2: 获取 2023-2024 年数据")
df2 = api.fetch("399006.SZ", "2023-01-01", "2024-12-31")
if df2 is not None:
    print(f"  数据量: {len(df2)}")
    print(f"  日期范围: {df2.index[0]} ~ {df2.index[-1]}")
else:
    print("  获取失败")