Hopiu/django-model-utils
1
0
Fork 0
mirror of https://github.com/Hopiu/django-model-utils.git synced 2026-03-17 04:10:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix StatusField bug

Browse source 
Fixes #29
This commit is contained in:
Trey Hunner 2013-04-09 14:28:53 -07:00
parent f0b56c0ca3
commit 59e484e17f
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
model_utils/fields.py
View file

@ -54,14 +54,17 @@ class StatusField(models.CharField):
self.check_for_status = not kwargs.pop('no_check_for_status', False)
super(StatusField, self).__init__(*args, **kwargs)
def contribute_to_class(self, cls, name):
if not cls._meta.abstract and self.check_for_status:
assert hasattr(cls, 'STATUS'), \
def prepare_class(self, sender, **kwargs):
if not sender._meta.abstract and self.check_for_status:
assert hasattr(sender, 'STATUS'), \
"To use StatusField, the model '%s' must have a STATUS choices class attribute." \
% cls.__name__
self._choices = cls.STATUS
% sender.__name__
self._choices = sender.STATUS
if not self.has_default():
self.default = tuple(cls.STATUS)[0][0] # set first as default
self.default = tuple(sender.STATUS)[0][0] # set first as default
def contribute_to_class(self, cls, name):
models.signals.class_prepared.connect(self.prepare_class, sender=cls)
super(StatusField, self).contribute_to_class(cls, name)