Custom form: operations forms

1 files changed, 13 insertions, 5 deletions

diff --git a/ishtar_common/forms.py b/ishtar_common/forms.py
index 8b97cfbab..eebd912ea 100644
--- a/ishtar_common/forms.py
+++ b/ishtar_common/forms.py
@@ -336,18 +336,26 @@ def get_data_from_formset(data):
 
 
 class FieldType(object):
-    def __init__(self, key, model, is_multiple=False):
+    def __init__(self, key, model, is_multiple=False, extra_args=None):
         self.key = key
         self.model = model
         self.is_multiple = is_multiple
+        self.extra_args = extra_args
 
     def get_choices(self, initial=None):
-        return self.model.get_types(
-            empty_first=not self.is_multiple,
-            initial=initial)
+        args = {
+            'empty_first': not self.is_multiple,
+            'initial': initial
+        }
+        if self.extra_args:
+            args.update(self.extra_args)
+        return self.model.get_types(**args)
 
     def get_help(self):
-        return self.model.get_help()
+        args = {}
+        if self.extra_args:
+            args.update(self.extra_args)
+        return self.model.get_help(**args)
 
 
 class ManageOldType(object):