48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
from __future__ import absolute_import
|
|
|
|
from urlparse import urlparse, parse_qs
|
|
from urllib import unquote_plus
|
|
|
|
from django.utils.simplejson import dumps
|
|
from django.db.models import Manager
|
|
from django.contrib.auth.models import AnonymousUser
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.utils.encoding import smart_str
|
|
|
|
|
|
class RecentSearchManager(Manager):
|
|
def add_query_for_user(self, search_view):
|
|
from .settings import RECENT_COUNT
|
|
|
|
query_dict = parse_qs(unquote_plus(smart_str(urlparse(search_view.request.get_full_path()).query)))
|
|
|
|
if query_dict and not isinstance(search_view.request.user, AnonymousUser):
|
|
# If the URL query has at least one variable with a value
|
|
new_recent, created = self.model.objects.get_or_create(user=search_view.request.user, query=dumps(query_dict), defaults={'hits': 0})
|
|
new_recent.hits = search_view.results.count()
|
|
new_recent.save()
|
|
to_delete = self.model.objects.filter(user=search_view.request.user)[RECENT_COUNT:]
|
|
for recent_to_delete in to_delete:
|
|
recent_to_delete.delete()
|
|
|
|
def get_for_user(self, user):
|
|
return [entry for entry in self.model.objects.filter(user=user) if entry.get_query()]
|
|
|
|
|
|
class IndexableObjectManager(Manager):
|
|
def get_indexables(self, datetime=None):
|
|
if datetime:
|
|
return self.model.objects.filter(datetime__gte=datetime)
|
|
else:
|
|
return self.model.objects.all()
|
|
|
|
def get_indexables_pk_list(self, datetime=None):
|
|
return self.get_indexables(datetime).values_list('object_id', flat=True)
|
|
|
|
def mark_indexable(self, obj):
|
|
content_type = ContentType.objects.get_for_model(obj)
|
|
self.model.objects.get_or_create(content_type=content_type, object_id=obj.pk)
|
|
|
|
def clear_all(self):
|
|
self.model.objects.all().delete()
|