RegularPolygonMarker and CircleMarker are instances of Marker

Author: Martin Journois

folium/features.py

@@ -12,7 +12,7 @@
                         _locations_mirror, _locations_tolist, image_to_url)
 
 from .element import Element, Figure, JavascriptLink, CssLink, MacroElement
-from .map import TileLayer, Icon
+from .map import TileLayer, Icon, Marker
 
 
 class WmsTileLayer(TileLayer):
@@ -48,7 +48,7 @@ def __init__(self, url, name=None,
         """)  # noqa
 
 
-class RegularPolygonMarker(MacroElement):
+class RegularPolygonMarker(Marker):
     def __init__(self, location, color='black', opacity=1, weight=2,
                  fill_color='blue', fill_opacity=1,
                  number_of_sides=4, rotation=0, radius=15, popup=None):
@@ -85,9 +85,8 @@ def __init__(self, location, color='black', opacity=1, weight=2,
 
         For more information, see https://humangeo.github.io/leaflet-dvf/
         """
-        super(RegularPolygonMarker, self).__init__()
+        super(RegularPolygonMarker, self).__init__(location, popup=popup)
         self._name = 'RegularPolygonMarker'
-        self.location = location
         self.color = color
         self.opacity = opacity
         self.weight = weight
@@ -96,8 +95,6 @@ def __init__(self, location, color='black', opacity=1, weight=2,
         self.number_of_sides = number_of_sides
         self.rotation = rotation
         self.radius = radius
-        if popup is not None:
-            self.add_children(popup)
 
         self._template = Template(u"""
             {% macro script(this, kwargs) %}
@@ -420,22 +417,19 @@ def __init__(self, html=None, icon_size=None, icon_anchor=None,
             """)  # noqa
 
 
-class CircleMarker(MacroElement):
+class CircleMarker(Marker):
     def __init__(self, location, radius=500, color='black',
                  fill_color='black', fill_opacity=0.6, popup=None):
         """
         TODO docstring here
 
         """
-        super(CircleMarker, self).__init__()
+        super(CircleMarker, self).__init__(location, popup=popup)
         self._name = 'CircleMarker'
-        self.location = location
         self.radius = radius
         self.color = color
         self.fill_color = fill_color
         self.fill_opacity = fill_opacity
-        if popup is not None:
-            self.add_children(popup)
 
         self._template = Template(u"""
             {% macro script(this, kwargs) %}

folium/map.py

@@ -363,11 +363,8 @@ def __init__(self, location, popup=None, icon=None):
         ----------
         location: tuple or list, default None
             Latitude and Longitude of Marker (Northing, Easting)
-        popup: string or tuple, default 'Pop Text'
-            Input text or visualization for object. Can pass either text,
-            or a tuple of the form (Vincent object, 'vis_path.json')
-            It is possible to adjust the width of text/HTML popups
-            using the optional keywords `popup_width` (default is 300px).
+        popup: string or folium.Popup, default None
+            Input text or visualization for object.
         icon: Icon plugin
             the Icon plugin to use to render the marker.
 
@@ -378,15 +375,16 @@ def __init__(self, location, popup=None, icon=None):
         Example
         -------
         >>>map.simple_marker(location=[45.5, -122.3], popup='Portland, OR')
-        >>>map.simple_marker(location=[45.5, -122.3], popup=(vis, 'vis.json'))
-
+        >>>map.simple_marker(location=[45.5, -122.3], popup=folium.Popup('Portland, OR'))
         """
         super(Marker, self).__init__()
         self._name = 'Marker'
         self.location = location
         if icon is not None:
             self.add_children(icon)
-        if popup is not None:
+        if isinstance(popup, text_type) or isinstance(popup, binary_type):
+            self.add_children(Popup(popup))
+        elif popup is not None:
             self.add_children(popup)
 
         self._template = Template(u"""

tests/test_features.py

@@ -6,7 +6,7 @@
 """
 
 import os
-from folium import Map
+from folium import Map, Popup
 from folium import features
 from folium.six import text_type
 from folium.element import Element
@@ -79,6 +79,15 @@ def test_figure_double_rendering():
     out2 = f.render()
     assert out == out2
 
+def test_marker_popups():
+    m = Map()
+    features.Marker([45,-180],popup='-180').add_to(m)
+    features.Marker([45,-120],popup=Popup('-120')).add_to(m)
+    features.RegularPolygonMarker([45,-60],popup='-60').add_to(m)
+    features.RegularPolygonMarker([45,0],popup=Popup('0')).add_to(m)
+    features.CircleMarker([45,60],popup='60').add_to(m)
+    features.CircleMarker([45,120],popup=Popup('120')).add_to(m)
+    m._repr_html_()
 
 # DivIcon.
 def test_divicon():