From 713e058aa10cfa16f3ebeb06e82ee22ecefbd929 Mon Sep 17 00:00:00 2001 From: yeezy-na-izi Date: Wed, 23 Nov 2022 12:57:42 +0300 Subject: [PATCH] Handle error if user role does not exist --- users/admin.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/users/admin.py b/users/admin.py index b7fb254c..229cac73 100644 --- a/users/admin.py +++ b/users/admin.py @@ -91,14 +91,17 @@ def save_model(self, request, obj, form, change): if change: old_user = CustomUser.objects.get(id=obj.id) if obj.user_type != old_user.user_type: - if old_user.user_type == 1: - old_user.member.delete() - elif old_user.user_type == 2: - old_user.mentor.delete() - elif old_user.user_type == 3: - old_user.expert.delete() - elif old_user.user_type == 4: - old_user.investor.delete() + try: + if old_user.user_type == 1: + old_user.member.delete() + elif old_user.user_type == 2: + old_user.mentor.delete() + elif old_user.user_type == 3: + old_user.expert.delete() + elif old_user.user_type == 4: + old_user.investor.delete() + except Exception: + pass if obj.user_type == 1: Member.objects.create(user=old_user)