Merge pull request #30 from Kalgoc/fix/piggies-users

Author: DaHunterTime

piggies/views.py

@@ -45,11 +45,11 @@ def list(self, request):
             piggies = Piggies.objects.filter(username=username)
             serializer = PiggiesSerializer(piggies, many=True)
             final_users = []
+            piggies_id = [str(pig["piggy"]) for pig in serializer.data]
 
-            for pig in serializer.data:
-                for user in filtered_users:
-                    if str(pig["piggy"]) == user["user_id"]:
-                        final_users.append(user)
+            for user in filtered_users:
+                if user["user_id"] in piggies_id:
+                    final_users.append(user)
 
             return Response(data=final_users)
         except Piggies.DoesNotExist:
@@ -118,11 +118,11 @@ def users(self, request):
                 return Response(data=filtered_users)
 
             final_users = []
+            piggies_id = [str(pig["piggy"]) for pig in serializer.data]
 
-            for pig in serializer.data:
-                for user in filtered_users:
-                    if str(pig["piggy"]) != user["user_id"]:
-                        final_users.append(user)
+            for user in filtered_users:
+                if user["user_id"] not in piggies_id:
+                    final_users.append(user)
 
             return Response(data=final_users)
         except Exception as e: