from django.db.models.fields import subclassing
from django.core import exceptions
from django.conf import settings
from django.db import models
class TrueNoneField(models.NullBooleanField):
__metaclass__ = subclassing.SubfieldBase
def db_type(self):
if 'postgresql' in settings.DATABASE_ENGINE:
return "boolean CHECK (%s <> false)" % self.get_attname_column()[1]
else:
return super(TrueNoneField, self).db_type()
def to_python(self, value):
if value in (True, 1):
return True
elif value in (False, 0, None):
return None
raise exceptions.ValidationError("This value must be either True or None.")
def get_db_prep_save(self, value):
return super(TrueNoneField, self).get_db_prep_save(self.to_python(value))
Comments