import asyncio
import argparse
from packages.core.database import SessionLocal
from packages.core.logger import app_logger
from packages.connectors.mexico import MexicoAduanasConnector

async def run_mexico_job(start_period: str = None, end_period: str = None):
    """
    运行墨西哥海关数据抓取与解析任务
    """
    app_logger.info("Starting Mexico aduanas job...")
    async with SessionLocal() as session:
        connector = MexicoAduanasConnector(
            session=session,
            start_period=start_period,
            end_period=end_period
        )
        await connector.run()
    app_logger.info("Mexico aduanas job finished.")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Run Mexico 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_mexico_job(start_period=args.start, end_period=args.end))