From 82baf564c6814e408fbdb7b397d24571ff01ecd1 Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Thu, 24 May 2012 02:31:37 -0400 Subject: [PATCH] Add document model get_content method --- apps/documents/models.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/apps/documents/models.py b/apps/documents/models.py index 01e8d6c809..57b2f2be51 100644 --- a/apps/documents/models.py +++ b/apps/documents/models.py @@ -280,6 +280,9 @@ class Document(models.Model): @property def content(self): return self.latest_version.content + + def get_content(self, *args, **kwargs): + return self.latest_version.get_content(*args, **kwargs) filename = property(_get_filename, _set_filename) @@ -542,10 +545,15 @@ class DocumentVersion(models.Model): @property def content(self): + return self.get_content() + + def get_content(self, add_page_number=False): content = [] for page in self.document.pages.all(): if page.content: content.append(page.content) + if add_page_number: + content.append(u'\n\n\n - Page %s - \n\n\n' % page.page_number) return u''.join(content)