diff options
| author | Rob Austein <sra@hactrn.net> | 2012-11-10 14:30:31 +0000 |
|---|---|---|
| committer | Rob Austein <sra@hactrn.net> | 2012-11-10 14:30:31 +0000 |
| commit | 9c01ce4d4118f88172665343c933c185c8def5f3 (patch) | |
| tree | 4d14c2bd0715ea5cd36caf52d98f9bbc4b6dcffe | |
| parent | a29f391f804e43ff2bbf0c5550fae44a5c8d85a0 (diff) | |
Stylistic nit.
svn path=/branches/tk316/; revision=4839
| -rw-r--r-- | rpkid/rpki/irdb/models.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/rpkid/rpki/irdb/models.py b/rpkid/rpki/irdb/models.py index 455affb7..010ba635 100644 --- a/rpkid/rpki/irdb/models.py +++ b/rpkid/rpki/irdb/models.py @@ -77,9 +77,8 @@ class EnumField(django.db.models.PositiveSmallIntegerField): __metaclass__ = django.db.models.SubfieldBase def __init__(self, *args, **kwargs): - if 'choices' in kwargs: - if isinstance(kwargs["choices"], (tuple, list)) and isinstance(kwargs["choices"][0], str): - kwargs["choices"] = tuple(enumerate(kwargs["choices"], 1)) + if isinstance(kwargs.get("choices"), (tuple, list)) and isinstance(kwargs["choices"][0], str): + kwargs["choices"] = tuple(enumerate(kwargs["choices"], 1)) django.db.models.PositiveSmallIntegerField.__init__(self, *args, **kwargs) self.enum_i2s = dict(self.flatchoices) self.enum_s2i = dict((v, k) for k, v in self.flatchoices) |