entrouvert
/
debian-django-multiselectfield
17
0
Fork 0
Code Issues Pull Requests Releases Activity

Merge pull request #21 from vasyabigi/master 

Fixed prep/to_python values of model field.
This commit is contained in:
Pablo Martín 2014-10-13 14:46:10 +02:00
parent a01e3a27c3 cbfe2ee9bd
commit b3ea613558
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
multiselectfield/db/fields.py
View File

@ -98,12 +98,11 @@ class MultiSelectField(models.CharField):
return MultiSelectFormField(**defaults)
def get_prep_value(self, value):
return ",".join(value)
return '' if value is None else ",".join(value)
def to_python(self, value):
if value is not None:
if value:
return value if isinstance(value, list) else value.split(',')
return ''
def contribute_to_class(self, cls, name):
super(MultiSelectField, self).contribute_to_class(cls, name)
@ -112,14 +111,15 @@ class MultiSelectField(models.CharField):
fieldname = name
choicedict = dict(self.choices)
display = []
for value in getattr(obj, fieldname):
item_display = choicedict.get(value, None)
if item_display is None:
try:
item_display = choicedict.get(int(value), value)
except (ValueError, TypeError):
item_display = value
display.append(string_type(item_display))
if getattr(obj, fieldname):
for value in getattr(obj, fieldname):
item_display = choicedict.get(value, None)
if item_display is None:
try:
item_display = choicedict.get(int(value), value)
except (ValueError, TypeError):
item_display = value
display.append(string_type(item_display))
return ", ".join(display)
setattr(cls, 'get_%s_display' % self.name, get_display)