Add document model get_content method

This commit is contained in:
Roberto Rosario
2012-05-24 02:31:37 -04:00
parent c2d7ce94b1
commit 82baf564c6

View File

@@ -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)