From 367d197cc0db03f060d56351b659007a6aa08e92 Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Sat, 7 Jan 2012 23:30:35 -0400 Subject: [PATCH] Add 'get_for_user' method to the RecentDocument model --- apps/documents/managers.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/apps/documents/managers.py b/apps/documents/managers.py index 8c07b5ec54..189535a67d 100644 --- a/apps/documents/managers.py +++ b/apps/documents/managers.py @@ -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):