Add 'get_for_user' method to the RecentDocument model

This commit is contained in:
Roberto Rosario
2012-01-07 23:30:35 -04:00
parent 51b63e86de
commit 367d197cc0

View File

@@ -4,6 +4,7 @@ from ast import literal_eval
from datetime import datetime
from django.db import models
from django.contrib.auth.models import AnonymousUser
from .conf.settings import RECENT_COUNT
@@ -16,6 +17,12 @@ class RecentDocumentManager(models.Manager):
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 not user.is_anonymous():
return [recent_document.document for recent_document in self.model.objects.filter(user=user)]
else:
return []
class DocumentPageTransformationManager(models.Manager):