Consolidate reset functionality

gee-community · Aug 19, 2023 · 85e2dd0 · 85e2dd0
1 parent b2857c8
commit 85e2dd0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/geemap/geemap.py b/geemap/geemap.py
@@ -75,9 +75,7 @@ def handle_draw(_, action, geo_json):
                 self._roi_end = True
                 self._roi_start = False
             except Exception as e:
-                self.geometries = []
-                self.properties = []
-                self.last_geometry = None
+                self.reset(clear_draw_control=False)
                 self._roi_start = False
                 self._roi_end = False
                 print("There was an error creating Earth Engine Feature.")

diff --git a/geemap/map_widgets.py b/geemap/map_widgets.py
@@ -434,15 +434,16 @@ def last_feature(self):
     def count(self):
         return len(self.geometries)
 
-    def reset(self):
+    def reset(self, clear_draw_control=True):
         """Resets the draw controls."""
         if self.layer is not None:
             self.host_map.remove_layer(self.layer)
         self.geometries = []
         self.properties = []
         self.last_geometry = None
         self.layer = None
-        self._clear_draw_control()
+        if clear_draw_control:
+            self._clear_draw_control()
 
     def remove_geometry(self, geometry):
         index = self.geometries.index(geometry)