import asyncio
from packages.connectors.brazil import BrazilComexStatConnector
from packages.core.database import AsyncSessionLocal

async def main():
    print("开始测试抓取...")
    async with AsyncSessionLocal() as session:
        connector = BrazilComexStatConnector(
            session=session,
            start_period="2026-05",
            end_period="2026-05",
            max_rows_per_slice=10  # 只抓 10 条
        )
        connector.batch_no = "TEST_BATCH_001"
        await connector.run()
    print("测试抓取结束.")

asyncio.run(main())