django-model-utils/model_utils/tests/fields.py

27 lines
700 B
Python
Raw Normal View History

2013-08-08 14:02:12 +00:00
from django.db import models
2013-08-12 09:22:34 +00:00
from django.utils.six import with_metaclass, string_types
2013-08-08 14:02:12 +00:00
class MutableField(with_metaclass(models.SubfieldBase, models.TextField)):
2013-08-08 14:02:12 +00:00
def to_python(self, value):
if value == '':
2013-08-08 14:02:12 +00:00
return None
try:
2013-08-12 09:22:34 +00:00
if isinstance(value, string_types):
return [int(i) for i in value.split(',')]
2013-08-08 14:02:12 +00:00
except ValueError:
pass
return value
def get_db_prep_save(self, value, connection):
if value is None:
return ''
2013-08-08 14:02:12 +00:00
if isinstance(value, list):
value = ','.join((str(i) for i in value))
2013-08-08 14:02:12 +00:00
return super(MutableField, self).get_db_prep_save(value, connection)