Skip to content

Instantly share code, notes, and snippets.

@akatreyt
Created November 4, 2014 01:51
Show Gist options
  • Save akatreyt/4a7c4e685fab272db2b5 to your computer and use it in GitHub Desktop.
Save akatreyt/4a7c4e685fab272db2b5 to your computer and use it in GitHub Desktop.
class return_my_connections(APIView):
serializer_class = GetConnectedUsersSerializer
def get(self, request, *args, **kwargs):
user = request.user
fetch_objc = ConnectedUsers.objects.filter(Q(from_user=user) | Q(to_user=user))
data = []
for connectedUser in fetch_objc:
temp_obj = ConnectedUsers.objects.get(pk = connectedUser.id)
me = User.objects.get(pk = connectedUser.from_user.id)
them = User.objects.get(pk = connectedUser.to_user.id)
if me != user:
them = connectedUser.from_user
me = connectedUser.to_user
setattr(temp_obj, 'me', me)
setattr(temp_obj, 'them', them)
data.append(temp_obj)
print data
return JsonResponse(data, status=200)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment