Fixed URLs of one-sandwich views. Fixed adding sandwiches.

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

diff --git a/models.py b/models.py
index abbf825..ed6410f 100644 (file)
--- a/models.py
+++ b/models.py
@@ -1,6 +1,10 @@
 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 +14,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 +31,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)