Move the RecentDocumentManager to the models.py file

To be able to return a QuerySet and not a list of recent documents
This commit is contained in:
Roberto Rosario
2012-02-05 23:27:03 -04:00
parent 2147923eb0
commit c0d8e6f667
2 changed files with 19 additions and 22 deletions
-20
View File
@@ -1,29 +1,9 @@
from __future__ import absolute_import
from ast import literal_eval
from datetime import datetime
from django.db import models
from .conf.settings import RECENT_COUNT
class RecentDocumentManager(models.Manager):
def add_document_for_user(self, user, document):
if user.is_authenticated():
self.model.objects.filter(user=user, document=document).delete()
new_recent = self.model(user=user, document=document, datetime_accessed=datetime.now())
new_recent.save()
to_delete = self.model.objects.filter(user=user)[RECENT_COUNT:]
for recent_to_delete in to_delete:
recent_to_delete.delete()
def get_for_user(self, user):
if user.is_authenticated():
return [recent_document.document for recent_document in self.model.objects.filter(user=user)]
else:
return []
class DocumentPageTransformationManager(models.Manager):
def get_for_document_page(self, document_page):