import asyncio
import argparse
from packages.core.database import SessionLocal
from packages.core.logger import app_logger
from packages.connectors.india import IndiaCustomsConnector

async def run_india_job(start_period: str = None, end_period: str = None):
    """
    运行印度海关商业源数据抓取与解析任务
    """
    app_logger.info("Starting India customs job...")
    async with SessionLocal() as session:
        connector = IndiaCustomsConnector(
            session=session,
            start_period=start_period,
            end_period=end_period
        )
        await connector.run()
    app_logger.info("India customs job finished.")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Run India Customs Data Connector")
    parser.add_argument("--start", type=str, help="Start period in YYYY-MM format")
    parser.add_argument("--end", type=str, help="End period in YYYY-MM format")
    args = parser.parse_args()

    asyncio.run(run_india_job(start_period=args.start, end_period=args.end))