login, signup, logout and add sandwich buttons appear according to whether a user...

[~kgodey/maayanwich.git] / models.py

diff --git a/models.py b/models.py
index abbf825..3139b47 100644 (file)
--- a/models.py
+++ b/models.py
@@ -1,6 +1,8 @@
 from django.db import models
 from django.contrib.auth.models import User
 import datetime
+from django.core.urlresolvers import reverse
+from django.template.defaultfilters import slugify
 
 
 class Ingredient(models.Model):
@@ -10,6 +12,7 @@ class Ingredient(models.Model):
        def __unicode__(self):
                return self.name
 
+
 class Sandwich(models.Model):
        adjective = models.CharField(max_length=200)
        slug = models.SlugField()
@@ -26,3 +29,10 @@ class Sandwich(models.Model):
                
        def __unicode__(self):
                return self.adjective
+               
+       def get_absolute_url(self): 
+           return reverse('sandwich_by_slug', kwargs={'slug':self.slug})
+       
+       def save(self, *args, **kwargs):
+               self.slug = slugify(self.adjective)
+               super(Sandwich, self).save(*args, **kwargs)