2018-10-29 13:35:18 +00:00
|
|
|
from django.db import models
|
|
|
|
from django_kepi import object_type_registry
|
2018-10-29 14:28:13 +00:00
|
|
|
from django_kepi.cache_model import Cache
|
2018-10-29 15:18:23 +00:00
|
|
|
from django.contrib.contenttypes.fields import GenericForeignKey
|
|
|
|
from django.contrib.contenttypes.models import ContentType
|
2018-10-29 13:35:18 +00:00
|
|
|
|
|
|
|
#######################
|
|
|
|
|
2018-10-29 14:28:13 +00:00
|
|
|
class SomethingManager(models.Manager):
|
2018-10-29 13:35:18 +00:00
|
|
|
|
2018-10-29 14:28:13 +00:00
|
|
|
f_type = None
|
2018-10-29 13:35:18 +00:00
|
|
|
|
|
|
|
def get_queryset(self):
|
|
|
|
|
2018-10-29 15:18:23 +00:00
|
|
|
# This is an empty QuerySet; it's here because
|
|
|
|
# it sets the type of the objects in the result.
|
2018-10-29 14:39:52 +00:00
|
|
|
result = models.QuerySet(
|
|
|
|
model = self.parent_model,
|
|
|
|
)
|
2018-10-29 14:28:13 +00:00
|
|
|
|
2018-10-29 15:18:23 +00:00
|
|
|
cache_query = Cache.objects.filter(f_type=self.f_type)
|
|
|
|
cache_query.annotate(
|
|
|
|
is_local = models.Value(
|
|
|
|
False
|
|
|
|
),
|
|
|
|
content_type = models.Value(
|
|
|
|
ContentType.objects.get_for_model(Cache),
|
|
|
|
),
|
|
|
|
)
|
|
|
|
|
2018-10-29 14:39:52 +00:00
|
|
|
handlers = [
|
2018-10-29 15:18:23 +00:00
|
|
|
cache_query,
|
2018-10-29 14:39:52 +00:00
|
|
|
]
|
2018-10-29 14:28:13 +00:00
|
|
|
|
2018-10-29 15:18:23 +00:00
|
|
|
for h in object_type_registry[self.f_type]:
|
|
|
|
query = h.objects.all()
|
|
|
|
query.annotate(
|
|
|
|
is_local = models.Value(
|
|
|
|
True,
|
|
|
|
),
|
|
|
|
content_type = models.Value(
|
|
|
|
ContentType.objects.get_for_model(h),
|
|
|
|
),
|
|
|
|
)
|
|
|
|
handlers.append(query)
|
2018-10-29 13:35:18 +00:00
|
|
|
|
2018-10-29 15:18:23 +00:00
|
|
|
return result.union(*handlers)
|
2018-10-29 13:35:18 +00:00
|
|
|
|
|
|
|
#######################
|
|
|
|
|
|
|
|
class Person(models.Model):
|
|
|
|
|
|
|
|
f_type = 'Person'
|
2018-10-29 14:28:13 +00:00
|
|
|
objects = SomethingManager()
|
|
|
|
objects.f_type = f_type
|
2018-10-29 13:35:18 +00:00
|
|
|
|
2018-10-29 15:18:23 +00:00
|
|
|
content_type = models.ForeignKey(
|
|
|
|
ContentType,
|
|
|
|
on_delete=models.CASCADE,
|
|
|
|
),
|
|
|
|
|
|
|
|
content_object = GenericForeignKey(
|
|
|
|
'content_type',
|
|
|
|
'url',
|
|
|
|
),
|
|
|
|
|
2018-10-29 13:35:18 +00:00
|
|
|
class Meta:
|
2018-10-29 14:28:13 +00:00
|
|
|
managed = False
|
2018-10-29 14:39:52 +00:00
|
|
|
|
|
|
|
Person.objects.parent_model = Person # this is ugly
|
|
|
|
|