fmartingr
/
amiibofindr
mirror of https://github.com/fmartingr/amiibofindr.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Added 404 errors instead of 500

This commit is contained in:
Felipe Martín 2015-06-27 15:25:26 +02:00
parent c38df904ba
commit c0b398a125
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
amiibofindr/apps/amiibo/views.py
View File

@ -1,7 +1,7 @@
# coding: utf-8
# django
from django.shortcuts import render
from django.shortcuts import render, get_object_or_404
from django.views.generic.base import View
# amiibo
@ -13,7 +13,7 @@ class CollectionView(View):
def get(self, request, collection='all'):
if collection != 'all':
collection = Collection.objects.get(slug=collection)
collection = get_object_or_404(Collection, slug=collection)
amiibo_list = collection.amiibos
else:
collection = None
@ -29,8 +29,9 @@ class AmiiboView(View):
template = 'amiibo/amiibo.html'
def get(self, request, collection=None, amiibo=None):
amiibo_obj = Amiibo.objects.get(slug=amiibo,
collection__slug=collection)
amiibo_obj = get_object_or_404(Amiibo,
slug=amiibo,
collection__slug=collection)
return render(request, self.template, {
'selected_collection': amiibo_obj.collection,