Fix project schemas types

po8klasie · Mar 13, 2023 · ec9a721 · ec9a721
1 parent 8d75187
commit ec9a721
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/po8klasie_fastapi/app/project/schemas.py b/po8klasie_fastapi/app/project/schemas.py
@@ -1,12 +1,18 @@
 from typing import Any, Dict, List
 
-from pydantic import BaseModel
+from pydantic import BaseModel, validator
 
 
 class MapOptionsSchema(BaseModel):
-    center: [float, float]
+    center: List[float]
     zoom: float
 
+    @validator("center")
+    def check_len_eq_2(cls, value):
+        if len(value) != 2:
+            raise ValueError("map_options.center should have len == 2")
+        return value
+
 
 class DefaultQuerySchema(BaseModel):
     project_id: str