Django LowerCaseCharField
Мы реализовали LowerCaseCharField. Мы были бы рады услышать лучшие предложения реализации.
from django.db.models.fields import CharField
class LowerCaseCharField(CharField):
"""
Defines a charfield which automatically converts all inputs to
lowercase and saves.
"""
def pre_save(self, model_instance, add):
"""
Converts the string to lowercase before saving.
"""
current_value = getattr(model_instance, self.attname)
setattr(model_instance, self.attname, current_value.lower())
return getattr(model_instance, self.attname)
На самом деле мы любим иметь:
> modelinstance.field_name="TEST"
> print modelinstance.field_name
'test'
текущая реализация только преобразовывает в нижний регистр, когда это сохраняется.
3 ответа
Вы можете переопределить to_python , что позволит вам сравнивать нестрочные строки при поиске в базе данных. Фактический метод - get_prep_value , но поскольку он вызывает to_python для CharField , его удобнее переопределить:
def to_python(self, value):
value = super(LowerCaseCharField, self).to_python(value)
if isinstance(value, basestring):
return value.lower()
return value
Теперь вы можете выполнять такие запросы, как:
MyModel.objects.filter(lccf="MiXeD")
Edit:
Перечитывая ваш вопрос, похоже, вы хотите, чтобы понижение вступило в силу немедленно. Для этого вам необходимо создать дескриптор (объект python нового стиля с методами __ get __ и __ set __ , см. документацию python и django код для связанных моделей) и переопределите calculate_to_class в поле, чтобы установить поле модели для вашего дескриптора.
Вот полный пример, который у меня в голове, который можно использовать повторно для всех полей, которые хотят изменить значение при настройке.
class ModifyingFieldDescriptor(object):
""" Modifies a field when set using the field's (overriden) .to_python() method. """
def __init__(self, field):
self.field = field
def __get__(self, instance, owner=None):
if instance is None:
raise AttributeError('Can only be accessed via an instance.')
return instance.__dict__[self.field.name]
def __set__(self, instance, value):
instance.__dict__[self.field.name] = self.field.to_python(value)
class LowerCaseCharField(CharField):
def to_python(self, value):
value = super(LowerCaseCharField, self).to_python(value)
if isinstance(value, basestring):
return value.lower()
return value
def contribute_to_class(self, cls, name):
super(LowerCaseCharField, self).contribute_to_class(cls, name)
setattr(cls, self.name, ModifyingFieldDescriptor(self))
Использование свойств Python непросто из-за того, как django.models.Models волшебным образом устанавливает свои атрибуты экземпляра на основе атрибутов класса. Для этого есть открытая ошибка .
В итоге я сделал именно то, что делал оригинальный плакат.
Это означает, что вам нужно сделать:
>>> modelinstance.field_name="TEST"
>>> modelinstance.save() # Extra step
>>> print modelinstance.field_name
'test'