From a9d211f32f4ff0b1b9f3e49fff364bbc863694f7 Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Mon, 21 Nov 2011 05:40:41 -0400 Subject: [PATCH] Added size and page_count properties to the document model --- apps/documents/models.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/apps/documents/models.py b/apps/documents/models.py index d4bb53f274..6272e5635b 100644 --- a/apps/documents/models.py +++ b/apps/documents/models.py @@ -194,6 +194,10 @@ class Document(models.Model): self.save() return detected_pages + + @property + def page_count(self): + return self.documentpage_set.count() def save_to_file(self, filepath, buffer_size=1024 * 1024): """ @@ -274,6 +278,13 @@ class Document(models.Model): def delete(self, *args, **kwargs): super(Document, self).delete(*args, **kwargs) return self.file.storage.delete(self.file.path) + + @property + def size(self): + if self.exists(): + return self.file.storage.size(self.file.path) + else: + return None class DocumentTypeFilename(models.Model):