diff --git a/account/serializers.py b/account/serializers.py index b783efd..6356be5 100644 --- a/account/serializers.py +++ b/account/serializers.py @@ -373,17 +373,16 @@ class UpdateUserNotificationSerializer(serializers.ModelSerializer): fields = ['push_notification', 'email_notification', 'sms_notification'] def create(self, validated_data): - instance = UserNotification.objects.filter(user=self.context).last() - if instance: - # change notification status - instance.push_notification = validated_data.get('push_notification',instance.push_notification) - instance.email_notification = validated_data.get('email_notification', instance.email_notification) - instance.sms_notification = validated_data.get('sms_notification', instance.sms_notification) - instance.save() - else: - instance = UserNotification.objects.create(user=self.context) + instance, _ = UserNotification.objects.update_or_create( + user=self.context, + defaults={ + 'push_notification': validated_data.get('push_notification'), + 'email_notification': validated_data.get('email_notification'), + 'sms_notification': validated_data.get('sms_notification'), + }) return instance + class UserPhoneOtpSerializer(serializers.ModelSerializer): """User Phone serializers""" class Meta(object):