diff options
Diffstat (limited to 'tbc/models.py')
| -rw-r--r-- | tbc/models.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/tbc/models.py b/tbc/models.py index 74419e4..e2e6940 100644 --- a/tbc/models.py +++ b/tbc/models.py @@ -2,6 +2,7 @@ from django.db import models from django.contrib.auth.models import User from PythonTBC import settings from django.contrib.admin.models import LogEntry +from local import sitemap_path CATEGORY = (("fluid mechanics", "Fluid Mechanics"), @@ -56,7 +57,8 @@ BOOK_PREFERENCE = (("book1","1st Book"), def get_notebook_dir(instance, filename): - return '%s/%s' % (instance.book.title.replace(' ', '_'), filename.replace(' ', '_')) + book_dir = instance.book.title.replace(' ', '_')+'_by_'+instance.book.author.replace(' ','_') + return '%s/%s' % (book_dir, filename.replace(' ', '_')) def get_image_dir(instance, filename): @@ -104,6 +106,7 @@ class Book(models.Model): def __unicode__(self): name = self.title or 'Book' return '%s'%(name) + class Chapters(models.Model): name = models.CharField(max_length=200) @@ -113,6 +116,8 @@ class Chapters(models.Model): def __unicode__(self): name = self.name or 'Chapter' return '%s'%(name) + def get_absolute_url(self): + return sitemap_path+str(self.notebook) class ScreenShots(models.Model): |