bakery-ia/services/external/app/models/__init__.py

"""
External Service Models Package

Import all models to ensure they are registered with SQLAlchemy Base.
"""

# Import all models to register them with the Base metadata
from .traffic import (
    TrafficData,
    TrafficMeasurementPoint,
    TrafficDataBackgroundJob,
)

from .weather import (
    WeatherData,
    WeatherForecast,
)

from .city_weather import CityWeatherData
from .city_traffic import CityTrafficData

# List all models for easier access
__all__ = [
    # Traffic models
    "TrafficData",
    "TrafficMeasurementPoint",
    "TrafficDataBackgroundJob",
    # Weather models
    "WeatherData",
    "WeatherForecast",
    # City-based models (new)
    "CityWeatherData",
    "CityTrafficData",
]