vent_agent_2.0/scripts/refresh_main_fan_mines.py

#!/usr/bin/env python3
"""执行主通风机批次固定首步：刷新并保存正常生产煤矿清单，不调用模型。"""
import asyncio
import json
from pathlib import Path
import sys

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))


async def refresh():
    import config
    from extensions.main_fan import mine_catalog
    from extensions.shaanxi.client import GatewayClient
    from extensions.shaanxi.settings import GatewaySettings

    gateway = GatewayClient(GatewaySettings.from_config())
    try:
        snapshot = await mine_catalog.refresh(gateway=gateway)
    finally:
        await gateway.close()
    print(json.dumps({
        'snapshot_id': snapshot['snapshot_id'], 'stage': snapshot['stage'],
        'counts': snapshot['counts'], 'status_counts': snapshot['status_counts'],
        'selected_mines_file': str(config.ROOT / 'data/main_fan/mine_catalog'
                                   / snapshot['snapshot_id'] / 'selected-mines.json'),
    }, ensure_ascii=False, indent=2))


if __name__ == '__main__':
    asyncio.run(refresh())