Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 10 additions & 7 deletions src/validation/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
# Third party imports
from geojson_pydantic import Feature, FeatureCollection, MultiPolygon, Polygon
from pydantic import BaseModel as PydanticModel
from pydantic import Field, validator
from pydantic import ConfigDict, Field, validator

# Reader imports
from src.config import (
Expand Down Expand Up @@ -256,13 +256,14 @@ class SnapshotResponse(BaseModel):
task_id: str
track_link: str

class Config:
json_schema_extra = {
model_config = ConfigDict(
json_schema_extra={
"example": {
"task_id": "aa539af6-83d4-4aa3-879e-abf14fffa03f",
"track_link": "/tasks/status/aa539af6-83d4-4aa3-879e-abf14fffa03f/",
}
}
)


class SnapshotTaskResult(BaseModel):
Expand All @@ -279,8 +280,8 @@ class SnapshotTaskResponse(BaseModel):
status: str
result: SnapshotTaskResult

class Config:
json_schema_extra = {
model_config = ConfigDict(
json_schema_extra={
"example": {
"id": "3fded368-456f-4ef4-a1b8-c099a7f77ca4",
"status": "SUCCESS",
Expand All @@ -294,13 +295,15 @@ class Config:
},
}
}
)


class StatusResponse(BaseModel):
last_updated: str

class Config:
json_schema_extra = {"example": {"lastUpdated": "2022-06-27 19:59:24+05:45"}}
model_config = ConfigDict(
json_schema_extra={"example": {"lastUpdated": "2022-06-27 19:59:24+05:45"}}
)


class StatsRequestParams(BaseModel, GeometryValidatorMixin):
Expand Down