additional layers are templates

Author: Martin Journois

folium/plugins/__init__.py

@@ -3,4 +3,5 @@
 from .marker_cluster import MarkerCluster
 from .scroll_zoom_toggler import ScrollZoomToggler
 from .terminator import Terminator
-from .boat_marker import BoatMarker
+from .boat_marker import BoatMarker
+from .layer import Layer, LayerControl

folium/plugins/layer.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+
+from .plugin import Plugin
+
+class Layer(Plugin):
+    """Adds a layer to the map."""
+    def __init__(self, layer_name = None, tile_url=None, min_zoom=1, max_zoom=18, attribution=''):
+        super(Layer, self).__init__()
+        self.plugin_name = 'Layer'
+        self.tile_url = tile_url if tile_url is not None else '//otile1.mqcdn.com/tiles/1.0.0/osm/{z}/{x}/{y}.png'
+        self.attribution = attribution
+        self.min_zoom = min_zoom
+        self.max_zoom = max_zoom
+        self.object_id = self.object_name
+        if layer_name is not None:
+            self.object_name = layer_name
+
+    def render_js(self, nb):
+        return """
+        var layer_"""+self.object_id+""" = L.tileLayer('"""+self.tile_url+"""', {
+            maxZoom: """+str(self.max_zoom)+""",
+            minZoom: """+str(self.min_zoom)+""",
+            attribution: '"""+str(self.attribution)+"""'
+            });
+        layer_"""+self.object_id+""".addTo(map);
+        """
+
+class LayerControl(Plugin):
+    """Adds a layer control to the map."""
+    def __init__(self):
+        super(LayerControl, self).__init__()
+        self.plugin_name = 'LayerControl'
+
+    def render_js(self, nb):
+        return """
+        var baseLayer = {
+          "Base Layer": base_tile,"""+\
+        ",".join(['"%s" : layer_%s ' % (x.object_name,x.object_id) for x in self.map.plugins['Layer']])+\
+        """};
+
+        L.control.layers(baseLayer, layer_list).addTo(map);
+        """
+    
+    
+    

folium/plugins/plugin.py

@@ -13,6 +13,7 @@ def __init__(self):
     def add_to_map(self, map):
         """Adds the plugin on a folium.map object."""
         map.plugins.setdefault(self.plugin_name,[]).append(self)
+        self.map = map
 
     def render_html(self, nb):
         """Generates the HTML part of the plugin."""

folium/templates/fol_template.html

@@ -131,7 +131,7 @@
                                        layers: [base_tile]
                                      });
 
-      L.control.layers(baseLayer, layer_list).addTo(map);
+      // L.control.layers(baseLayer, layer_list).addTo(map);
 
       //cluster group
       var clusteredmarkers = L.markerClusterGroup();