Add deconstruct() method to django fields.

wsmith323 · Nov 24, 2016 · 50356c0 · 50356c0
1 parent 7137126
commit 50356c0
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/staticmodel/django/fields.py b/staticmodel/django/fields.py
@@ -21,6 +21,19 @@ def __init__(self, *args, **kwargs):
 
         super(StaticModelFieldMixin, self).__init__(*args, **kwargs)
 
+    def deconstruct(self):
+        name, path, args, kwargs = super(StaticModelFieldMixin, self).deconstruct()
+
+        kwargs.update(dict(
+            static_model=self._static_model,
+            value_field_name=self._value_field_name,
+            display_field_name=self._display_field_name,
+        ))
+
+        del kwargs["choices"]
+
+        return name, path, args, kwargs
+
     def _validate_values(self):
         for member in self._static_model.members.all():
             value = getattr(member, self._value_field_name)