diff --git a/amiibofindr/apps/amiibo/context_processors.py b/amiibofindr/apps/amiibo/context_processors.py index 22df09e..a162342 100644 --- a/amiibofindr/apps/amiibo/context_processors.py +++ b/amiibofindr/apps/amiibo/context_processors.py @@ -26,9 +26,12 @@ def user_amiibo(request): wishlisted = [] trading = [] if request.user.is_authenticated(): - owned = UserAmiibo.objects.filter(own=True).values_list('_amiibo__id', flat=True) - wishlisted = UserAmiibo.objects.filter(want=True).values_list('_amiibo__id', flat=True) - trading = UserAmiibo.objects.filter(trade=True).values_list('_amiibo__id', flat=True) + owned = UserAmiibo.objects.filter(own=True, + user_id=request.user.pk).values_list('_amiibo__id', flat=True) + wishlisted = UserAmiibo.objects.filter(want=True, + user_id=request.user.pk).values_list('_amiibo__id', flat=True) + trading = UserAmiibo.objects.filter(trade=True, + user_id=request.user.pk).values_list('_amiibo__id', flat=True) return { 'USER_AMIIBO_OWNED': owned,