1 files changed, 9 insertions, 5 deletions
diff --git a/app/books/models.py b/app/books/models.py
index 1464c78..f5a4a04 100644
--- a/app/books/models.py
+++ b/app/books/models.py
@@ -1,12 +1,13 @@
 from django.db import models
+from django.core.urlresolvers import reverse
 from django.utils.encoding import force_text
 from django.conf import settings
 from django.template.defaultfilters import slugify
-from taggit.managers import TaggableManager
 
 # http://freewisdom.org/projects/python-markdown/
 import markdown
 
+
 class Book(models.Model):
     title = models.CharField(max_length=200)
     author_name = models.CharField(max_length=200)
@@ -39,14 +40,18 @@ class Book(models.Model):
         return self.title
 
     def get_absolute_url(self):
-        return "/books/%s" % (self.slug)
+        return reverse("books:detail", kwargs={"slug": self.slug})
 
     def get_image_url(self):
         return '%sbook-covers/%s.jpg' % (settings.IMAGES_URL, self.slug)
 
-    def get_rating (self):
+    def get_rating(self):
         return int(self.rating)
 
+    @property
+    def ratings_range(cls):
+        return range(1, 6)
+
     def admin_thumbnail(self):
         return force_text('<a href=""><img src="%s" width="100" style="width:100px"></a>' % (self.get_image_url()))
     admin_thumbnail.allow_tags = True
@@ -58,12 +63,11 @@ class Book(models.Model):
         self.slug = slugify(self.title[:50])
         super(Book, self).save()
 
-#class Book
 
 class BookHighlight(models.Model):
     book = models.ForeignKey(Book)
     page = models.PositiveSmallIntegerField(null=True)
-    location  = models.CharField(max_length=200, blank=True, null=True)
+    location = models.CharField(max_length=200, blank=True, null=True)
     date_added = models.DateTimeField()
     body_markdown = models.TextField()