Add support for custom color palettes #51

mapboxgl/utils.py

@@ -1,7 +1,7 @@
 from .colors import color_ramps
 import geojson
 import json
-
+import re
 
 def row_to_geojson(row, lon, lat):
     """Convert a pandas dataframe row to a geojson format object.  Converts all datetimes to epoch seconds.
@@ -110,17 +110,36 @@ def create_color_stops(breaks, colors='RdYlGn', color_ramps=color_ramps):
     """Convert a list of breaks into color stops using colors from colorBrewer
     see www.colorbrewer2.org for a list of color options to pass
     """
-    num_breaks = len(breaks)
 
-    if colors not in color_ramps.keys():
-        raise ValueError('color does not exist in colorBrewer!')
+    if isinstance(colors, list):
+        ramp = colors
+        for color in color:
+            if not validateColor(color):
+                raise ValueError("Color is in wrong format: " + str(color))
     else:
-        stops = []
+        if colors not in color_ramps.keys():
+            raise ValueError('color does not exist in colorBrewer!')
         try:
+            num_breaks = len(breaks)
             ramp = color_ramps[colors][num_breaks]
         except KeyError:
             raise ValueError("Color ramp {} does not have a {} breaks".format(
                 colors, num_breaks))
-        for i, b in enumerate(breaks):
-            stops.append([b, ramp[i]])
-        return stops
+    stops = []
+    for i, b in enumerate(breaks):
+        stops.append([b, ramp[i]])
+    return stops
+
+def validateColor(color):
+    if not isinstance(color, str):
+        return False
+    regex = "rgb\([0-9]{1,3},[0-9]{1,3},[0-9]{1,3}\)"
+    if not re.fullmatch(regex, color):
+        return False
+    color = color.replace("rgb(", "")
+    color = color.replace(")", "")
+    color = color.split(",")
+    for value in color:
+        if not 0 <= int(value) <= 255:
+            return False
+    return True