diff --git a/SubscribesApp/views.py b/SubscribesApp/views.py index 3f4b207..59934e3 100644 --- a/SubscribesApp/views.py +++ b/SubscribesApp/views.py @@ -16,17 +16,21 @@ class SubscribersView(APIView): email_notification = data['email_notification'] auto_subscribe = data['auto_subscribe'] - user = User.objects.get(email=email) - subscribe_for_user = SubscribeForUser.objects.filter(user_id=user.id) + user = User.objects.filter(email=email) - if email_notification: - subscribe_for_user.update(receive_finish_subscribe_msg=True) - else: - subscribe_for_user.update(receive_finish_subscribe_msg=False) + if user: + subscribe_for_user = SubscribeForUser.objects.filter(user_id=user[0].id) - if auto_subscribe: - subscribe_for_user.update(auto_continue=True) - else: - subscribe_for_user.update(auto_continue=False) + if email_notification: + subscribe_for_user.update(receive_finish_subscribe_msg=True) + else: + subscribe_for_user.update(receive_finish_subscribe_msg=False) - return Response("Subscriptions updated successfully", status=status.HTTP_200_OK) + if auto_subscribe: + subscribe_for_user.update(auto_continue=True) + else: + subscribe_for_user.update(auto_continue=False) + + return Response("Subscriptions updated successfully", status=status.HTTP_200_OK) + + return Response("User not found", status=status.HTTP_404_NOT_FOUND)