customs-data / apps /worker /run_extended.py
3v324v23's picture
Initial commit for Hugging Face Space
3834ddb
Raw
History Blame Contribute Delete
856 Bytes
import asyncio
import sys
import os
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
from packages.core.logger import app_logger
from packages.connectors.mock.extended_mock import ExtendedMockConnector, COUNTRY_CONFIGS
from packages.core.database import AsyncSessionLocal
async def run_extended_mock_jobs():
"""
循环跑批处理 7 个新国家的采集任务
"""
async with AsyncSessionLocal() as session:
for country_code in COUNTRY_CONFIGS.keys():
app_logger.info(f"--- Starting Sync Job for {country_code} ---")
connector = ExtendedMockConnector(country_code=country_code, session=session)
await connector.run()
app_logger.info(f"--- Finished Sync Job for {country_code} ---")
if __name__ == "__main__":
asyncio.run(run_extended_mock_jobs())