2010-02-24 20:06:59 +01:00
|
|
|
from django.db import models
|
2019-07-23 14:04:06 +02:00
|
|
|
from django.db.models.query_utils import DeferredAttribute
|
2014-03-14 23:18:20 +01:00
|
|
|
|
|
|
|
|
|
|
|
class CustomTypedField(models.TextField):
|
|
|
|
def db_type(self, connection):
|
|
|
|
return 'custom_field'
|
2019-07-23 14:04:06 +02:00
|
|
|
|
|
|
|
|
|
|
|
class CustomDeferredAttribute(DeferredAttribute):
|
|
|
|
def __get__(self, instance, cls=None):
|
|
|
|
self._count_call(instance, 'get')
|
|
|
|
return super().__get__(instance, cls)
|
|
|
|
|
|
|
|
def __set__(self, instance, value):
|
|
|
|
self._count_call(instance, 'set')
|
|
|
|
instance.__dict__[self.field.attname] = value
|
|
|
|
|
|
|
|
def _count_call(self, instance, get_or_set):
|
|
|
|
count_attr = '_%s_%s_count' % (self.field.attname, get_or_set)
|
|
|
|
count = getattr(instance, count_attr, 0)
|
|
|
|
setattr(instance, count_attr, count + 1)
|
|
|
|
|
|
|
|
|
|
|
|
class CustomDescriptorField(models.CharField):
|
|
|
|
descriptor_class = CustomDeferredAttribute
|