diff options
Diffstat (limited to 'tbc/models.py')
| -rw-r--r-- | tbc/models.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tbc/models.py b/tbc/models.py index b063274..f9e721e 100644 --- a/tbc/models.py +++ b/tbc/models.py @@ -44,6 +44,10 @@ PROPOSAL_STATUS = (("pending","Pending"), ("book completed","Book Completed"), ("rejected","Rejected")) +BOOK_PREFERENCE = (("book1","1st Book"), + ("book2","2nd Book"), + ("book3","3rd Book")) + def get_notebook_dir(instance, filename): return '%s/%s/%s' % (instance.book.contributor, instance.book.title.replace(' ', '_'), filename.replace(' ', '_')) @@ -53,6 +57,10 @@ def get_image_dir(instance, filename): return '%s/%s/screenshots/%s' % (instance.book.contributor, instance.book.title.replace(' ', '_'), filename.replace(' ', '_')) +def get_sample_dir(instance, filename): + user_name = instance.user.user.first_name+instance.user.user.last_name + return 'sample_notebooks/%s/%s' % (user_name, filename.replace(' ', '_')) + class Profile(models.Model): user = models.ForeignKey(User) about = models.CharField(max_length=256) @@ -132,3 +140,9 @@ class Proposal(models.Model): def __unicode__(self): user = self.user.user.username or 'User' return '%s'%(user) + + +class SampleNotebook(models.Model): + proposal = models.ForeignKey(Proposal) + notebook = models.FileField(upload_to=get_sample_dir) + book_preference = models.CharField(max_length=30, choices=BOOK_PREFERENCE) |