Add 'get_for_user' method to the RecentDocument model
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user