From 42bdb2cf140d39261a0ef5faa9562844473bac0a Mon Sep 17 00:00:00 2001 From: jiangweidong Date: Mon, 25 Sep 2023 15:54:15 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=E4=BC=98=E5=8C=96=E6=89=BE=E5=9B=9E?= =?UTF-8?q?=E5=AF=86=E7=A0=81=E6=97=B6=E5=8C=BA=E5=8F=B7=E5=B8=A6=E5=8A=A0?= =?UTF-8?q?=E5=8F=B7=E6=97=A0=E6=B3=95=E5=8C=B9=E9=85=8D=E7=9A=84=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/users/views/profile/reset.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/apps/users/views/profile/reset.py b/apps/users/views/profile/reset.py index b8e18a647..cde1463ea 100644 --- a/apps/users/views/profile/reset.py +++ b/apps/users/views/profile/reset.py @@ -114,6 +114,11 @@ class UserForgotPasswordView(FormView): target = form.cleaned_data[form_type] code = form.cleaned_data['code'] + query_key = form_type + if form_type == 'sms': + query_key = 'phone' + target = target.lstrip('+') + try: sender_util = SendAndVerifyCodeUtil(target, backend=form_type) sender_util.verify(code) @@ -121,7 +126,6 @@ class UserForgotPasswordView(FormView): form.add_error('code', str(e)) return super().form_invalid(form) - query_key = 'phone' if form_type == 'sms' else form_type user = get_object_or_none(User, **{'username': username, query_key: target}) if not user: form.add_error('code', _('No user matched'))