use new class FitSelected

folium/features.py

@@ -1929,3 +1929,56 @@ def __init__(
             out.setdefault(cm(color), []).append([[lat1, lng1], [lat2, lng2]])
         for key, val in out.items():
             self.add_child(PolyLine(val, color=key, weight=weight, opacity=opacity))
+
+
+class FitSelected(MacroElement):
+    """Fit the bounds of the maps to the enabled overlays.
+
+    Parameters
+    ----------
+    padding: int, default 25
+        Sets the amount of padding in the top left and bottom right corners of
+        a map container that shouldn't be accounted for when setting the view
+        to fit bounds.
+    max_zoom: int, optional
+        The maximum possible zoom to use when fitting to the bounds.
+    fly: bool, default False
+        Use a smoother, longer animation.
+    fit_on_map_load: bool, default True
+        Apply the fit when initially loading the map.
+    """
+
+    _template = Template(
+        """
+        {% macro script(this, kwargs) %}
+        function customFlyToBounds() {
+            let bounds = L.latLngBounds([]);
+            {{ this._parent.get_name() }}.eachLayer(function(layer) {
+                if (typeof layer.getBounds === 'function') {
+                    bounds.extend(layer.getBounds());
+                }
+            });
+            if (bounds.isValid()) {
+                {{ this._parent.get_name() }}.{{ this.method }}(bounds, {{ this.options|tojson }});
+            }
+        }
+        {{ this._parent.get_name() }}.on('overlayadd', customFlyToBounds);
+        {%- if this.fit_on_map_load %}
+        customFlyToBounds();
+        {%- endif %}
+        {% endmacro %}
+    """
+    )
+
+    def __init__(
+        self,
+        padding: int = 25,
+        max_zoom: Optional[int] = None,
+        fly: bool = False,
+        fit_on_map_load: bool = True,
+    ):
+        super().__init__()
+        self._name = "FlyToSelected"
+        self.method = "flyToBounds" if fly else "fitBounds"
+        self.fit_on_map_load = fit_on_map_load
+        self.options = parse_options(padding=(padding, padding), max_zoom=max_zoom)

folium/map.py

@@ -154,22 +154,6 @@ class LayerControl(MacroElement):
             {{ val }}.remove();
             {%- endfor %}
 
-            {%- if this.fly_to_selected %}
-            function customFlyToBounds() {
-                let bounds = L.latLngBounds([]);
-                {{ this._parent.get_name() }}.eachLayer(function(layer) {
-                    if (typeof layer.getBounds === 'function') {
-                        bounds.extend(layer.getBounds());
-                    }
-                });
-                if (bounds.isValid()) {
-                    {{ this._parent.get_name() }}.flyToBounds(bounds, {padding: [25, 25]});
-                }
-            }
-            {{ this._parent.get_name() }}.on('overlayadd', customFlyToBounds);
-            customFlyToBounds();
-            {%- endif %}
-
         {% endmacro %}
         """
     )
@@ -179,15 +163,13 @@ def __init__(
         position: str = "topright",
         collapsed: bool = True,
         autoZIndex: bool = True,
-        fly_to_selected: bool = True,
         **kwargs: TypeJsonValue,
     ):
         super().__init__()
         self._name = "LayerControl"
         self.options = parse_options(
             position=position, collapsed=collapsed, autoZIndex=autoZIndex, **kwargs
         )
-        self.fly_to_selected = fly_to_selected
         self.base_layers: OrderedDict[str, str] = OrderedDict()
         self.overlays: OrderedDict[str, str] = OrderedDict()
         self.layers_untoggle: OrderedDict[str, str] = OrderedDict()