Better implement tags, with separate namespaces

[djsite.git] / quotes / views.py

diff --git a/quotes/views.py b/quotes/views.py
index 981af054515b7817efb40daae2138b7d34de0b1a..9bed533b045f73e0385f32c934e5c780d3cc2b70 100644 (file)
--- a/quotes/views.py
+++ b/quotes/views.py
@@ -1,10 +1,8 @@
 from django.shortcuts import render
-from django.http import HttpResponse
-from django.template import loader
 
 from random import randint
 
-from .models import Author, Quote, Tag
+from .models import Author, Work, Quote, QuoteTag
 
 # Create your views here.
 def onequote(request, quote_id):
@@ -17,7 +15,7 @@ def random(request):
     return onequote(request, randint(1, count))
 
 def tags(request, tag_id):
-    tag = Tag.objects.get(id=tag_id)
+    tag = QuoteTag.objects.get(id=tag_id)
     context = { 'tag' : tag }
     return render(request, 'quotes/tag.html', context)
 
@@ -26,6 +24,11 @@ def author(request, author_id):
     context = { 'author' : author }
     return render(request, 'quotes/author.html', context)
 
+def work(request, work_id):
+    work = Work.objects.get(id=work_id)
+    context = { 'work': work }
+    return render(request, 'quotes/work.html', context)
+
 def all(request):
     quotes = Quote.objects.all()
     context = { 'quotes' : quotes }