AttributeError when using Enums (Python 3.5) in Django Model

Question

I'm trying to use Enums on Python 3.5 in my Django model. Why am I getting this when trying to migrate?

field=models.CharField(choices=[('RE', 'Red'), ('GR', 'Green'), ('BL', 'Blue'), ('OR', 'Orange'), ('YE', 'Yellow'), ('PU', 'Purple')], default=users.models.COLOR('BL'), max_length=2),
AttributeError: module 'users.models' has no attribute 'COLOR'

-

class User(AbstractBaseUser):

  class COLOR(enum.Enum):
    RED = 'RE'
    GREEN = 'GR'
    BLUE = 'BL'
    ORANGE = 'OR'
    YELLOW = 'YE'
    PURPLE = 'PU'

  //...
  color = models.CharField(max_length=2, choices=((x.value, x.name.title()) for x in COLOR), default=COLOR.BLUE)

docs.djangoproject.com/en/dev/ref/models/fields/… I think this will help you. — wencakisa
– wencakisa, Commented Mar 28, 2017 at 18:40
@Wencakisa I'm trying to use enums instead just declaring the constants. The error is thrown on the default line, not on the choices line. — aroooo
– aroooo, Commented Mar 28, 2017 at 18:42
Why are you trying to specify an enum value as the default there, anyway? I don't think Django would even recognize what COLOR.BLUE is, even if you specified it correctly. — user2357112
– user2357112, Commented Mar 28, 2017 at 18:43

wim · Accepted Answer · 2017-03-28 18:46:09Z

1

Since you use a CharField on the model, you can not pass an enum instance (default=COLOR.BLUE) as the default value. You should pass a string instead.

The better way to do this is to define a custom field which knows how to clean enum instances. If you rather to keep things simple and stick to the CharField on the model, then just pass the enum value explicitly (default='BL').

edited Mar 28, 2017 at 18:46

answered Mar 28, 2017 at 18:40

wim

368k114 gold badges681 silver badges816 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

AttributeError when using Enums (Python 3.5) in Django Model

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related