added is color

1 files changed, 6 insertions, 1 deletions

diff --git a/gradient.py b/gradient.py
index 6d2e15f..4d01f9b 100755
--- a/gradient.py
+++ b/gradient.py
@@ -18,6 +18,11 @@ import sha
 #logger.setLevel(logging.WARNING)

 #}}}

 

+def is_color(s):

+  if re.match('(rgba?\([0-9]+,[0-9]+,[0-9]+\))|([a-zA-Z]+)|(\#[A-Ha-h0-9]+)', s):

+    return s

+  else:

+    raise ValueError

 PARAM_LIST = [ 

     "width", "height", 

     "color1", "color2", 

@@ -98,7 +103,7 @@ class Gradient:
         for key in PARAM_LIST:

             if key in form:

                 if key in ['color1', 'color2']:

-                    params[key] = form[key]

+                    params[key] = is_color(form[key])

                 else:

                     params[key] = sanitize(form[key])