Hide amiibo not marked as visible from lists

This commit is contained in:
Felipe Martín 2015-09-11 20:59:38 +02:00
parent 879dd04d46
commit 8b17a9d22f
1 changed files with 4 additions and 4 deletions

View File

@ -21,14 +21,14 @@ class CollectionView(View):
if collection:
collection = get_object_or_404(Collection, pk=collection)
if self.type == self.model.FIGURE:
amiibo_list = collection.figures
amiibo_list = collection.figures.filter(visible=True)
elif self.type == self.model.CARD:
amiibo_list = collection.cards
amiibo_list = collection.cards.filter(visible=True)
else:
amiibo_list = collection.amiibo
amiibo_list = collection.amiibo.filter(visible=True)
else:
collection = None
amiibo_list = self.model.objects.all()
amiibo_list = self.model.objects.filter(visible=True)
return render(request, self.template, {
'selected_collection': collection,