Geo form: fix point 3D creation and edition

1 files changed, 13 insertions, 2 deletions

diff --git a/ishtar_common/forms_common.py b/ishtar_common/forms_common.py
index 53cca91d5..2caf6c1af 100644
--- a/ishtar_common/forms_common.py
+++ b/ishtar_common/forms_common.py
@@ -2821,13 +2821,16 @@ class GISForm(forms.ModelForm, CustomForm, ManageOldType):
                 for geo_fields in self.GEO_FIELDS:
                     if geo_fields != keys:
                         for geo_field in geo_fields:
-                            self.fields.pop(geo_field)
                             if geo_field == "x":
                                 self.fields.pop("estimated_error_x")
                                 self.fields.pop("y")
                                 self.fields.pop("estimated_error_y")
                             if geo_field == "z":
+                                if geo_keys[0] == "point_3d":
+                                    geo_keys = list(geo_keys) + ["z"]
+                                    continue
                                 self.fields.pop("estimated_error_z")
+                            self.fields.pop(geo_field)
                 break
         return geo_keys
 
@@ -2849,7 +2852,15 @@ class GISForm(forms.ModelForm, CustomForm, ManageOldType):
             raise forms.ValidationError(
                 _("If you set a buffer type set a buffer.")
             )
-
+        if cleaned_data.get("point_3d"):
+            if "z" not in cleaned_data:
+                raise forms.ValidationError(
+                    _("A value is expected for Z.")
+                )
+            value = cleaned_data["point_3d"].ewkt
+            if "POINT Z" not in value:
+                value = value.replace("POINT", "POINT Z")
+                cleaned_data["point_3d"] = f'{value.split(")")[0]} {cleaned_data["z"]})'
 
         if "x" not in self.geo_keys:
             # reverse...