Enum class for python