fix(postgis type): completes the geometry type with z when three dime…

…nsions are required
APSL · May 14, 2024 · a9d3add · a9d3add
1 parent ca5e7a7
commit a9d3add
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/base_geoengine/fields.py b/base_geoengine/fields.py
@@ -36,7 +36,12 @@ class GeoField(fields.Field):
 
     @property
     def column_type(self):
-        return ("geometry", f"geometry({self.geo_type.upper()}, {self.srid})")
+        postgis_geom_type = self.geo_type.upper() if self.geo_type else "GEOMETRY"
+        if self.dim == "3":
+            postgis_geom_type += "Z"
+        elif self.dim == "4":
+            postgis_geom_type += "ZM"
+        return ("geometry", f"geometry({postgis_geom_type}, {self.srid})")
 
     def convert_to_column(self, value, record, values=None):
         """Convert value to database format