Fixed the documents app and updated to latest app registry api

This commit is contained in:
Roberto Rosario
2012-09-10 22:17:30 -04:00
parent e63b596ff5
commit 4d7999cadb
10 changed files with 98 additions and 84 deletions

View File

@@ -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 []