add options to draw control

folium/plugins/draw.py

@@ -1,7 +1,10 @@
 # -*- coding: utf-8 -*-
 
+
 from __future__ import (absolute_import, division, print_function)
 
+import json
+
 from branca.element import CssLink, Element, Figure, JavascriptLink, MacroElement
 
 from jinja2 import Template
@@ -15,23 +18,43 @@ class Draw(MacroElement):
     ----------
     export : bool, default False
         Add a small button that exports the drawn shapes as a geojson file.
-
+    filename : string, default 'data.geojson'
+        Name of geojson file
+    position : string, default 'topleft'
+        Position of control. It can be one of 'topleft', 'toprigth', 'bottomleft', 'bottomright'
+        See https://leafletjs.com/reference-1.4.0.html#control
+    draw_options : dict, optional
+        The options used to configure the draw toolbar
+        See http://leaflet.github.io/Leaflet.draw/docs/leaflet-draw-latest.html#drawoptions
+    edit_options : dict, optional
+        The options used to configure the edit toolbar
+        See https://leaflet.github.io/Leaflet.draw/docs/leaflet-draw-latest.html#editpolyoptions
     Examples
     --------
     >>> m = folium.Map()
-    >>> Draw(export=True).add_to(m)
+    >>> Draw(
+    ...     export=True,
+    ...     filename='my_data.geojson',
+    ...     position='topleft',
+    ...     draw_options={'polyline': {'allowIntersection': False}},
+    ...     edit_options={'poly': {'allowIntersection': False}}
+    ... ).add_to(m)
 
     For more info please check
     https://leaflet.github.io/Leaflet.draw/docs/leaflet-draw-latest.html
 
     """
     _template = Template(u"""
             {% macro script(this, kwargs) %}
+            var options = {
+              position: "{{this.position}}",
+              draw: {{this.draw_options}},
+              edit: {{this.edit_options}}
+            }
             // FeatureGroup is to store editable layers.
             var drawnItems = new L.featureGroup().addTo({{this._parent.get_name()}});
-            var {{this.get_name()}} = new L.Control.Draw({
-                "edit": {"featureGroup": drawnItems}
-                }).addTo({{this._parent.get_name()}})
+            options.edit.featureGroup = drawnItems
+            var {{this.get_name()}} = new L.Control.Draw(options).addTo({{this._parent.get_name()}})
             {{this._parent.get_name()}}.on(L.Draw.Event.CREATED, function (event) {
               var layer = event.layer,
                   type = event.layerType,
@@ -43,35 +66,40 @@ class Draw(MacroElement):
                 });
                drawnItems.addLayer(layer);
              });
-
-        {{this._parent.get_name()}}.on('draw:created', function(e) {
-            drawnItems.addLayer(e.layer);
-        });
-
-        document.getElementById('export').onclick = function(e) {
-            var data = drawnItems.toGeoJSON();
-            var convertedData = 'text/json;charset=utf-8,' + encodeURIComponent(JSON.stringify(data));
-            document.getElementById('export').setAttribute('href', 'data:' + convertedData);
-            document.getElementById('export').setAttribute('download','data.geojson');
-        }
+            {{this._parent.get_name()}}.on('draw:created', function(e) {
+                drawnItems.addLayer(e.layer);
+            });
+            document.getElementById('export').onclick = function(e) {
+              var data = drawnItems.toGeoJSON();
+              var convertedData = 'text/json;charset=utf-8,' + encodeURIComponent(JSON.stringify(data));
+              document.getElementById('export').setAttribute('href', 'data:' + convertedData);
+              document.getElementById('export').setAttribute(
+                'download',
+                "{{this.filename}}"
+              );
+            }
             {% endmacro %}
             """)
 
-    def __init__(self, export=False):
+    def __init__(self, export=False, filename='data.geojson',
+                 position='topleft', draw_options=None, edit_options=None):
         super(Draw, self).__init__()
         self._name = 'DrawControl'
         self.export = export
+        self.filename = filename
+        self.position = position
+        self.draw_options = json.dumps(draw_options or {})
+        self.edit_options = json.dumps(edit_options or {})
 
     def render(self, **kwargs):
-        super(Draw, self).render()
+        super(Draw, self).render(**kwargs)
 
         figure = self.get_root()
         assert isinstance(figure, Figure), ('You cannot render this Element '
                                             'if it is not in a Figure.')
 
         figure.header.add_child(
             JavascriptLink('https://cdnjs.cloudflare.com/ajax/libs/leaflet.draw/1.0.2/leaflet.draw.js'))  # noqa
-
         figure.header.add_child(
             CssLink('https://cdnjs.cloudflare.com/ajax/libs/leaflet.draw/1.0.2/leaflet.draw.css'))  # noqa