如何在Django中合并/嵌套EnumFields？

Question

Django-EnumFields lets you combine Enum fields in Django (the clue was in the title).

Can you combine nest these?

Here's an example that plays off the docs :

from django.db import models
from django_enumfield import enum

class BeerStyle(enum.Enum):
    LAGER = 0
    STOUT = 1
    WEISSBIER = 2

class SoftDrinkStyle(enum.Enum):
    COKE = 3
    LEMONADE = 4

class Drink(models.Model):
    style = enum.EnumField(????, default=BeerStyle.LAGER)

I don't know what would go in place of ???? , or if there is a better way to get this nested/combination to play out with Django. I'm mainly asking as I want Enum behaviour, with the ability to probe different types, eg in a save method, check for User age if the Drink is or type Beer .

Is this possible? Having played with this for a bit I don't see how.

Answer 1

Having looked over how Python Enums work, this looks like the best behaviour to mock up 'subclasses'

from django.db import models
from django_enumfield import enum

class DrinkStyle(enum.Enum):
    LAGER = (0, 'Beer')
    STOUT = (1, 'Beer')
    WEISSBIER = (2, 'Beer')
    COKE = (3, 'SoftDrink')
    LEMONADE = (4, 'SoftDrink')

    def __init__(self, id, drink_type):
        self.id = id    
        self.type = drink_type

    @property
    def type(self):
        return self.drink_type

class Drink(models.Model):
    style = enum.EnumField(DrinkStyle, default=DrinkStyle.LAGER)

Then use DrinkStyle.COKE.type to return the type.