47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from uuid import UUID
|
||
|
|
|
||
|
|
from pydantic import Field, model_validator
|
||
|
|
|
||
|
|
from .common import ApiEnvelope, JsonObject, JsonValue, RouteContract, SchemaModel
|
||
|
|
|
||
|
|
HTTP_METHOD = 'POST'
|
||
|
|
ROUTE_PATH = '/api/crop-simulation/growth/'
|
||
|
|
|
||
|
|
|
||
|
|
class CropSimulationGrowthRequest(SchemaModel):
|
||
|
|
plant_name: str
|
||
|
|
dynamic_parameters: list[str] = Field(min_length=1)
|
||
|
|
farm_uuid: UUID | None = None
|
||
|
|
weather: JsonValue | None = None
|
||
|
|
soil_parameters: JsonObject | None = None
|
||
|
|
site_parameters: JsonObject | None = None
|
||
|
|
crop_parameters: JsonObject | None = None
|
||
|
|
agromanagement: JsonObject | None = None
|
||
|
|
page_size: int | None = Field(default=None, ge=1, le=50)
|
||
|
|
|
||
|
|
@model_validator(mode='after')
|
||
|
|
def validate_farm_or_weather(self) -> 'CropSimulationGrowthRequest':
|
||
|
|
if self.farm_uuid is None and self.weather is None:
|
||
|
|
raise ValueError('Either farm_uuid or weather must be provided.')
|
||
|
|
return self
|
||
|
|
|
||
|
|
|
||
|
|
class CropSimulationGrowthResponseData(SchemaModel):
|
||
|
|
task_id: str
|
||
|
|
status_url: str
|
||
|
|
plant_name: str
|
||
|
|
|
||
|
|
|
||
|
|
class CropSimulationGrowthResponse(ApiEnvelope[CropSimulationGrowthResponseData]):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
CONTRACT = RouteContract(
|
||
|
|
method=HTTP_METHOD,
|
||
|
|
path=ROUTE_PATH,
|
||
|
|
request_model=CropSimulationGrowthRequest.__name__,
|
||
|
|
response_model=CropSimulationGrowthResponse.__name__,
|
||
|
|
)
|