Add save and create_map methods

Author: Martin Journois

folium/element.py

@@ -13,7 +13,7 @@
 import json
 import base64
 
-from .six import urlopen
+from .six import urlopen, text_type, binary_type
 from .utilities import _camelify, _parse_size
 
 class Element(object):
@@ -80,6 +80,27 @@ def render(self, **kwargs):
         """TODO : docstring here."""
         return self._template.render(this=self, kwargs=kwargs)
 
+    def save(self, outfile, close_file=True, **kwargs):
+        """Saves an Element into a file.
+
+        Parameters
+        ----------
+        outfile : str or file object
+            The file (or filename) where you want to ouput the html.
+        close_file : bool, default True
+            Whether the file has to be closed after write.
+        """
+        if isinstance(outfile,text_type) or isinstance(outfile,binary_type):
+            fid = open(outfile, 'wb')
+        else:
+            fid = outfile
+
+        root = self.get_root()
+        html = root.render(**kwargs)
+        fid.write(html.encode('utf8'))
+        if close_file:
+            fid.close()
+
 class Link(Element):
     def get_code(self):
         if self.code is None:

folium/folium.py

@@ -30,6 +30,23 @@ class Map(_Map):
     """This class inherits from the map.Map object in order to provide bindings to
     former folium API.
     """
+    def create_map(self, path='map.html', plugin_data_out=True, template=None):
+        """Write Map output to HTML.
+
+        Parameters:
+        -----------
+        path: string, default 'map.html'
+            Path for HTML output for map
+        plugin_data_out: boolean, default True
+            Deprecated, not used anymore
+        template: string, default None
+            Deprecated, not used anymore
+        """
+        warnings.warn("%s is deprecated. Use %s instead" % ("Map.create_map",
+                                                            "Map.save"),
+                      FutureWarning, stacklevel=2)
+        self.save(path)
+
     def add_wms_layer(self, wms_name=None, wms_url=None, wms_format=None,
                       wms_layers=None, wms_transparent=True):
         """Adds a simple tile layer.

tests/test_folium.py

@@ -563,6 +563,7 @@ def test_create_map(self):
 
         # Test write.
         map._parent.render()
+        map.save('map.html')
 
     def test_line(self):
         """Test line."""