82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
from __future__ import absolute_import
|
|
|
|
import datetime
|
|
|
|
from django.db import models
|
|
from django.contrib.auth.models import User
|
|
from django.core.urlresolvers import reverse
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.contrib.contenttypes import generic
|
|
from django.utils.simplejson import loads
|
|
from django.utils.http import urlencode
|
|
|
|
from .managers import RecentSearchManager, IndexableObjectManager
|
|
|
|
|
|
class RecentSearch(models.Model):
|
|
"""
|
|
Keeps a list of the n most recent search keywords for a given user
|
|
"""
|
|
user = models.ForeignKey(User, verbose_name=_(u'user'), editable=False)
|
|
query = models.TextField(verbose_name=_(u'query'), editable=False)
|
|
datetime_created = models.DateTimeField(verbose_name=_(u'datetime created'), editable=False)
|
|
hits = models.IntegerField(verbose_name=_(u'hits'), editable=False)
|
|
|
|
objects = RecentSearchManager()
|
|
|
|
def __unicode__(self):
|
|
return self.form_string()
|
|
|
|
def form_string(self):
|
|
if self.is_advanced():
|
|
return u'%s (%s)' % (self.get_query(), self.hits)
|
|
else:
|
|
return u'%s (%s)' % (u' '.join(self.get_query().get('q')), self.hits)
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.datetime_created = datetime.datetime.now()
|
|
super(RecentSearch, self).save(*args, **kwargs)
|
|
|
|
def get_query(self):
|
|
try:
|
|
return loads(self.query)
|
|
except ValueError:
|
|
return {}
|
|
|
|
def is_advanced(self):
|
|
return 'q' not in self.get_query()
|
|
|
|
def get_absolute_url(self):
|
|
return '?'.join([reverse('search'), urlencode(self.get_query(), doseq=True)])
|
|
|
|
class Meta:
|
|
ordering = ('-datetime_created',)
|
|
verbose_name = _(u'recent search')
|
|
verbose_name_plural = _(u'recent searches')
|
|
|
|
|
|
|
|
class IndexableObject(models.Model):
|
|
"""
|
|
Store a list of object links that have been modified and are
|
|
meant to be indexed in the next search index update
|
|
"""
|
|
datetime = models.DateTimeField(verbose_name=_(u'date time'))
|
|
content_type = models.ForeignKey(ContentType)
|
|
object_id = models.PositiveIntegerField()
|
|
content_object = generic.GenericForeignKey('content_type', 'object_id')
|
|
|
|
objects = IndexableObjectManager()
|
|
|
|
def __unicode__(self):
|
|
return unicode(self.content_object)
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.datetime = datetime.datetime.now()
|
|
super(IndexableObject, self).save(*args, **kwargs)
|
|
|
|
class Meta:
|
|
verbose_name = _(u'indexable object')
|
|
verbose_name_plural = _(u'indexable objects')
|