Changeset 170 for trunk/pycha/chart.py

Timestamp:

03/18/09 15:35:07 (3 years ago)

Author:

lgs

Message:

Implement different color schemes. Fixes #29

Files:

• trunk/pycha/chart.py (modified) (3 diffs)

trunk/pycha/chart.py

@@ -17 +17 @@
 
 import copy
+import inspect
 import math
 
@@ -121 +122 @@
         colorSchemeClass = ColorScheme.getColorScheme(name, None)
         if colorSchemeClass is None:
-            raise ValueError('Color scheme is invalid!')
-
+            raise ValueError('Color scheme "%s" is invalid!' % name)
+
+        # Remove invalid args before calling the constructor
         kwargs = dict(self.options.colorScheme.args)
+        validArgs = inspect.getargspec(colorSchemeClass.__init__)[0]
+        kwargs = dict([(k, v) for k, v in kwargs.items() if k in validArgs])
         self.colorScheme = colorSchemeClass(keys, **kwargs)
 
@@ -672 +676 @@
     colorScheme=Option(
         name='gradient',
-        args=Option(initialColor=DEFAULT_COLOR),
+        args=Option(
+            initialColor=DEFAULT_COLOR,
+            colors=[],
+            ),
     ),
     title=None,