Fixed the documents app and updated to latest app registry api
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
from __future__ import absolute_import
|
||||
|
||||
from ast import literal_eval
|
||||
from datetime import datetime
|
||||
|
||||
from django.db import models
|
||||
|
||||
@@ -24,3 +25,23 @@ class DocumentPageTransformationManager(models.Manager):
|
||||
warnings.append(e)
|
||||
|
||||
return transformations, warnings
|
||||
|
||||
|
||||
class RecentDocumentManager(models.Manager):
|
||||
def add_document_for_user(self, user, document):
|
||||
from .settings import RECENT_COUNT
|
||||
|
||||
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):
|
||||
document_model = models.get_model('documents', 'Document')
|
||||
if user.is_authenticated():
|
||||
return document_model.objects.filter(recentdocument__user=user)
|
||||
else:
|
||||
return []
|
||||
|
||||
Reference in New Issue
Block a user