mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2025-11-25 06:46:25 +00:00
e7b43782a4
Pydantic deprecates using extra keyword arguments on Field. Used json_schema_extra instead. Deprecated in Pydantic V2.0 to be removed in V3.0. Signed-off-by: Bobby Noelte <b0661n0e17e@gmail.com>
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""Weather forecast module for weather predictions."""
|
|
|
|
from typing import Optional
|
|
|
|
from pydantic import Field, field_validator
|
|
|
|
from akkudoktoreos.config.configabc import SettingsBaseModel
|
|
from akkudoktoreos.prediction.prediction import get_prediction
|
|
from akkudoktoreos.prediction.weatherabc import WeatherProvider
|
|
from akkudoktoreos.prediction.weatherimport import WeatherImportCommonSettings
|
|
|
|
prediction_eos = get_prediction()
|
|
|
|
# Valid weather providers
|
|
weather_providers = [
|
|
provider.provider_id()
|
|
for provider in prediction_eos.providers
|
|
if isinstance(provider, WeatherProvider)
|
|
]
|
|
|
|
|
|
class WeatherCommonProviderSettings(SettingsBaseModel):
|
|
"""Weather Forecast Provider Configuration."""
|
|
|
|
WeatherImport: Optional[WeatherImportCommonSettings] = Field(
|
|
default=None,
|
|
json_schema_extra={"description": "WeatherImport settings", "examples": [None]},
|
|
)
|
|
|
|
|
|
class WeatherCommonSettings(SettingsBaseModel):
|
|
"""Weather Forecast Configuration."""
|
|
|
|
provider: Optional[str] = Field(
|
|
default=None,
|
|
json_schema_extra={
|
|
"description": "Weather provider id of provider to be used.",
|
|
"examples": ["WeatherImport"],
|
|
},
|
|
)
|
|
|
|
provider_settings: WeatherCommonProviderSettings = Field(
|
|
default_factory=WeatherCommonProviderSettings,
|
|
json_schema_extra={
|
|
"description": "Provider settings",
|
|
"examples": [
|
|
# Example 1: Empty/default settings (all providers None)
|
|
{
|
|
"WeatherImport": None,
|
|
},
|
|
],
|
|
},
|
|
)
|
|
|
|
# Validators
|
|
@field_validator("provider", mode="after")
|
|
@classmethod
|
|
def validate_provider(cls, value: Optional[str]) -> Optional[str]:
|
|
if value is None or value in weather_providers:
|
|
return value
|
|
raise ValueError(
|
|
f"Provider '{value}' is not a valid weather provider: {weather_providers}."
|
|
)
|