Creating Enum from StateMachine Object

[halbow]

Hi 👋

I hope you're doing well and thanks for this lib !

I've seen in the doc that it's possible to crate a StateMachine object from an enum but I would like to see if it would be possible to create an Enum from the StateMachineobject ?

My usecase for this is that I need an Enum for my Database schema but I prefer to keep one single StateMachineobject instead of creating both two different object !

For the moment I use something like this:

class StateMachineMixin(StateMachine):

    @classmethod
    def Enum(cls):
        return Enum(f"{cls.__name__}Enum", {state.value: auto() for state in cls.states})

Let me know what you think, I can open a PR to implement the changes.

Thanks

[fgmacedo]

Hi @halbow , how are you? I liked the idea.

I think It may be implemented the same way we implemented the .from_enum(), on the States class.

An API like:

OrderStatusEnum = OrderStateMachine.states.as_enum()

About the implementation, for compatibility with the from_enum, we should use state.id: state.value as enum name and value.

class States:
    ...
    def as_enum(self):
        return Enum(f"{self.statemachine_name}Enum", {state.id: state.value for state in self})

Note that self.statemachine_name does not exist on the States class, so to accomplish this we will need to initialize the States class informing the name.

What do you think?