X-Git-Url: https://git.stderr.nl/gitweb?a=blobdiff_plain;f=tools%2Fmisc.py;h=ddc6cee7490dc6fc3492376f5e004821827b2952;hb=381c81a8ffac8c4b3abf6a762d031a7e261e8eaa;hp=1edf4725f59ad248f3f28ac11f87819cf54b74d4;hpb=f1833394374f434a61ab1ab9a551f2bded207182;p=matthijs%2Fprojects%2Fxerxes.git

diff --git a/tools/misc.py b/tools/misc.py
index 1edf472..ddc6cee 100644
--- a/tools/misc.py
+++ b/tools/misc.py
@@ -14,3 +14,46 @@ def make_iter(value):
             pass
     return [value]
 # vim: set sts=4 sw=4 expandtab:
+
+"""
+Decarator that catches any exception raised by the decorated function,
+prints it to stdout and raises it again.
+"""
+def log_error(func):
+    def show(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except Exception, e:
+            import traceback
+            traceback.print_exc()
+            raise e
+    return show
+
+def make_choices(objects):
+    """
+    Transforms a list (or iteratable) of model objects to a list
+    suitable to be used as a list of choices in form widgets like
+    Select.
+
+    This fullfills a similar (but simpler) function as
+    django.forms.models.ModelChoiceIterator, but that one requires a
+    FormField and is not public.
+    """
+    return [(o.pk, o) for o in objects]
+
+def filter_choices(choices, filter):
+    """
+    Returns the given choices list with only the choices with the names
+    in filter left. For example, when a model defines
+    A   = 'A'
+    B   = 'B'
+    FOO_CHOICES = (
+        (A, "Foo A"),
+        (B, "Foo B")
+    )
+
+    you can later define a modelfield using
+
+    foo = ChoiceField(choices=filter_choices(Foo.FOO_CHOICES, [Foo.A]))
+    """
+    return [(name, value) for (name, value) in choices if name in filter]