From 2ed0927b18358e53626e7da9d3166c4ebe9817a5 Mon Sep 17 00:00:00 2001 From: xinwen Date: Wed, 29 Jul 2020 17:46:43 +0800 Subject: [PATCH] =?UTF-8?q?fix(login):=20=E7=94=A8=E6=88=B7=E7=99=BB?= =?UTF-8?q?=E5=BD=95=E5=A0=A1=E5=9E=92=E6=9C=BA=E7=9A=84=E6=97=B6=E5=80=99?= =?UTF-8?q?=E5=81=B6=E5=B0=94=E4=BC=9A=E5=87=BA=E7=8E=B0=E2=80=9C=E5=AF=86?= =?UTF-8?q?=E7=A0=81=E8=A7=A3=E5=AF=86=E5=A4=B1=E8=B4=A5=E2=80=9D=EF=BC=8C?= =?UTF-8?q?=E5=AF=BC=E8=87=B4=E6=97=A0=E6=B3=95=E6=AD=A3=E5=B8=B8=E7=99=BB?= =?UTF-8?q?=E5=BD=95=E3=80=82=20#4408?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/authentication/views/login.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/apps/authentication/views/login.py b/apps/authentication/views/login.py index 141d0f6e7..a0b8804e1 100644 --- a/apps/authentication/views/login.py +++ b/apps/authentication/views/login.py @@ -109,12 +109,18 @@ class UserLoginView(mixins.AuthMixin, FormView): def get_context_data(self, **kwargs): # 生成加解密密钥对，public_key传递给前端，private_key存入session中供解密使用 - rsa_private_key, rsa_public_key = utils.gen_key_pair() - self.request.session['rsa_private_key'] = rsa_private_key + rsa_private_key = self.request.session.get('rsa_private_key') + rsa_public_key = self.request.session.get('rsa_public_key') + if not all((rsa_private_key, rsa_public_key)): + rsa_private_key, rsa_public_key = utils.gen_key_pair() + rsa_public_key = rsa_public_key.replace('\n', '\\n') + self.request.session['rsa_private_key'] = rsa_private_key + self.request.session['rsa_public_key'] = rsa_public_key + context = { 'demo_mode': os.environ.get("DEMO_MODE"), 'AUTH_OPENID': settings.AUTH_OPENID, - 'rsa_public_key': rsa_public_key.replace('\n', '\\n') + 'rsa_public_key': rsa_public_key } kwargs.update(context) return super().get_context_data(**kwargs)