[Bugfix] 修复首次登录条款问题及引导页面MFA配置问题

This commit is contained in:
BaiJiangjie
2018-04-26 19:49:09 +08:00
parent d5debc375e
commit 0962a16b22
4 changed files with 181 additions and 118 deletions

View File

@@ -278,6 +278,16 @@ class UserFirstLoginView(LoginRequiredMixin, SessionWizardView):
def get_form(self, step=None, data=None, files=None):
form = super().get_form(step, data, files)
form.instance = self.request.user
if isinstance(form, forms.UserMFAForm):
choices = form.fields["otp_level"].choices
if self.request.user.otp_force_enabled:
choices = [(k, v) for k, v in choices if k == 2]
else:
choices = [(k, v) for k, v in choices if k in [0, 1]]
form.fields["otp_level"].choices = choices
form.fields["otp_level"].initial = self.request.user.otp_level
return form