instance.state should return a StateWrapper object

The type of the returned value when the state was not yet set was inconsistent: State instead of StateWrapper
rbarrois · Feb 5, 2018 · 97bb5db · 97bb5db
1 parent 1307bf3
commit 97bb5db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/django_xworkflows/models.py b/django_xworkflows/models.py
@@ -72,7 +72,8 @@ def __get__(self, instance, owner):
             if self.parent_property and hasattr(self.parent_property, '__get__'):
                 # We override a property.
                 return self.parent_property.__get__(instance, owner)
-            return instance.__dict__.get(self.field.name, self.field.workflow.initial_state)
+            default = self.field.to_python(self.field.workflow.initial_state)
+            return instance.__dict__.get(self.field.name, default)
         else:
             return self.field.workflow