35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
import urlparse
|
|
|
|
from django.db import models
|
|
from django.utils.http import urlencode
|
|
from django.contrib.auth.models import AnonymousUser
|
|
|
|
from dynamic_search.conf.settings import RECENT_COUNT
|
|
|
|
|
|
class RecentSearchManager(models.Manager):
|
|
def add_query_for_user(self, user, query, hits):
|
|
parsed_query = urlparse.parse_qs(urlencode(dict(query.items())))
|
|
|
|
for key, value in parsed_query.items():
|
|
parsed_query[key] = ' '.join(value)
|
|
|
|
if 'q=' in query:
|
|
# Is a simple query
|
|
if not parsed_query.get('q'):
|
|
# Don't store empty simple searches
|
|
return
|
|
else:
|
|
# Cleanup query string and only store the q parameter
|
|
parsed_query = {'q': parsed_query['q']}
|
|
|
|
if parsed_query and not isinstance(user, AnonymousUser):
|
|
# If the URL query has at least one variable with a value
|
|
new_recent, created = self.model.objects.get_or_create(user=user, query=urlencode(parsed_query), defaults={'hits': hits})
|
|
if not created:
|
|
new_recent.hits = hits
|
|
new_recent.save()
|
|
|
|
for recent_to_delete in self.model.objects.filter(user=user)[RECENT_COUNT:]:
|
|
recent_to_delete.delete()
|