python-oxdjango/oxdjango/fields.py

52 lines
1.6 KiB
Python
Raw Normal View History

2010-01-22 23:06:33 +00:00
from django.db import models
from django.utils import simplejson as json
class DictField(models.TextField):
"""DictField is a textfield that contains JSON-serialized dictionaries."""
# Used so to_python() is called
__metaclass__ = models.SubfieldBase
def to_python(self, value):
"""Convert our string value to python after we load it from the DB"""
if isinstance(value, dict):
return value
2010-02-08 15:04:18 +00:00
try:
value = json.loads(value)
except: #this is required to load fixtures
value = eval(value)
2010-01-22 23:06:33 +00:00
assert isinstance(value, dict)
return value
def get_db_prep_save(self, value):
"""Convert our JSON object to a string before we save"""
assert isinstance(value, dict)
value = json.dumps(value)
return super(DictField, self).get_db_prep_save(value)
class TupleField(models.TextField):
"""TupleField is a textfield that contains JSON-serialized tuples."""
# Used so to_python() is called
__metaclass__ = models.SubfieldBase
def to_python(self, value):
"""Convert our string value to JSON after we load it from the DB"""
if isinstance(value, tuple):
return value
2010-02-08 15:04:18 +00:00
try:
value = json.loads(value)
except: #this is required to load fixtures
value = eval(value)
2010-01-22 23:06:33 +00:00
assert isinstance(value, list)
return tuple(value)
def get_db_prep_save(self, value):
"""Convert our JSON object to a string before we save"""
assert isinstance(value, tuple)
value = json.dumps(value)
return super(TupleField, self).get_db_prep_save(value)