From 09d0337193112ae0c90c12d94bdd7d8f2d8505e5 Mon Sep 17 00:00:00 2001 From: Alrott SlimRG <39348033+SlimRG@users.noreply.github.com> Date: Wed, 24 Apr 2024 02:04:24 +0300 Subject: [PATCH] Syntax cleaning --- moonraker/components/authorization.py | 20 ++++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/moonraker/components/authorization.py b/moonraker/components/authorization.py index e0e43b6a8..82edc3f0f 100644 --- a/moonraker/components/authorization.py +++ b/moonraker/components/authorization.py @@ -470,7 +470,10 @@ async def _handle_password_reset(self, self.users[username]['password'] = new_hashed_pass self._sync_user(username) if (self.enable_totp): - self.totp_secrets[username] = {'secret': pyotp.random_base32(), 'is_activated': False} + self.totp_secrets[username] = { + 'secret': pyotp.random_base32(), + 'is_activated': False + } self.totp_secret_db.sync(self.totp_secrets) return { 'username': username, @@ -524,7 +527,10 @@ async def _login_jwt_user( action = "user_logged_in" create = False if (self.enable_totp): - self.totp_secrets[username] = {'secret': pyotp.random_base32(), 'is_activated': False} + self.totp_secrets[username] = { + 'secret': pyotp.random_base32(), + 'is_activated': False + } self.totp_secret_db.sync(self.totp_secrets) else: if username not in self.users: @@ -543,7 +549,10 @@ async def _login_jwt_user( if hashed_pass != user_info['password']: raise self.server.error("Invalid Password") if (self.enable_totp): - user_data_totp = self.totp_secrets.get(username, {'secret': '', 'is_activated': True}) + user_data_totp = self.totp_secrets.get(username, { + 'secret': '', + 'is_activated': True + }) secret = user_data_totp['secret'] is_activated = user_data_totp['is_activated'] if secret == '': @@ -551,7 +560,10 @@ async def _login_jwt_user( if pyotp.TOTP(secret).verify(totp_code) is False: raise self.server.error("Invalid TOTP code") if is_activated is False: - self.totp_secrets[username] = {'secret': secret, 'is_activated': True} + self.totp_secrets[username] = { + 'secret': secret, + 'is_activated': True + } self.totp_secret_db.sync(self.totp_secrets) jwt_secret_hex: Optional[str] = user_info.get('jwt_secret', None) if jwt_secret_hex is None: