class SimpleState(models.Model):
state = StateEnumField(workflow, default="NEW", null=False)
comment = models.TextField(null=True, blank=True)
def save(self, *args, **kwargs):
super(SimpleState, self).save(*args, **kwargs)
if self.state._to:
self.state.change_state(None, commit=True)
def __str__(self):
return six.text_type(self.state._current_state)
def setUp(self):
self.state_enum = StateEnum(
State(
name="NEW",
next_states=["ASSIGNED", "MODIFIED"],
check_perms=[],
),
State(name="ASSIGNED", next_states=["MODIFIED"]),
State(name="MODIFIED", next_states=[]),
)
self.state_enum.set_state("NEW")
self.field = StateEnumField(self.state_enum, default='NEW')
