Improve the frontend 5
This commit is contained in:
22
services/external/app/jobs/initialize_data.py
vendored
22
services/external/app/jobs/initialize_data.py
vendored
@@ -16,18 +16,30 @@ logger = structlog.get_logger()
|
||||
|
||||
|
||||
async def main(months: int = 24):
|
||||
"""Initialize historical data for all enabled cities"""
|
||||
"""Initialize historical data for all enabled cities and seed calendars"""
|
||||
logger.info("Starting data initialization job", months=months)
|
||||
|
||||
try:
|
||||
manager = DataIngestionManager()
|
||||
success = await manager.initialize_all_cities(months=months)
|
||||
|
||||
if success:
|
||||
logger.info("✅ Data initialization completed successfully")
|
||||
# Initialize weather and traffic data
|
||||
weather_traffic_success = await manager.initialize_all_cities(months=months)
|
||||
|
||||
# Seed school calendars
|
||||
logger.info("Proceeding to seed school calendars...")
|
||||
calendar_success = await manager.seed_school_calendars()
|
||||
|
||||
# Both must succeed
|
||||
overall_success = weather_traffic_success and calendar_success
|
||||
|
||||
if overall_success:
|
||||
logger.info("✅ Data initialization completed successfully (weather, traffic, calendars)")
|
||||
sys.exit(0)
|
||||
else:
|
||||
logger.error("❌ Data initialization failed")
|
||||
if not weather_traffic_success:
|
||||
logger.error("❌ Weather/traffic initialization failed")
|
||||
if not calendar_success:
|
||||
logger.error("❌ Calendar seeding failed")
|
||||
sys.exit(1)
|
||||
|
||||
except Exception as e:
|
||||
|
||||
Reference in New Issue
Block a user