Add geometry discriminator. Related developmentseed#93.

eseglem · Jan 31, 2023 · f15bb60 · f15bb60
1 parent 49af995
commit f15bb60
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/geojson_pydantic/geometries.py b/geojson_pydantic/geometries.py
@@ -3,8 +3,9 @@
 import abc
 from typing import Any, Dict, Iterator, List, Literal, Union
 
-from pydantic import BaseModel, ValidationError, validator
+from pydantic import BaseModel, Field, ValidationError, validator
 from pydantic.error_wrappers import ErrorWrapper
+from typing_extensions import Annotated
 
 from geojson_pydantic.types import (
     LinearRing,
@@ -208,7 +209,10 @@ def _wkt_coordinates(self) -> str:
         )
 
 
-Geometry = Union[Point, MultiPoint, LineString, MultiLineString, Polygon, MultiPolygon]
+Geometry = Annotated[
+    Union[Point, MultiPoint, LineString, MultiLineString, Polygon, MultiPolygon],
+    Field(discriminator="type"),
+]
 
 
 class GeometryCollection(BaseModel):