Update flask_authgen_jwt.py

Added a better way to update username and password keys in dictionary
dmtzs · Feb 28, 2023 · 764cf5e · 764cf5e
1 parent 5c069b6
commit 764cf5e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = Flask-authgen-jwt
-version = 4.1.1
+version = 4.1.2
 author = Diego Martinez
 license = MIT
 author_email = [email protected]

diff --git a/src/flask_authgen_jwt.py b/src/flask_authgen_jwt.py
@@ -121,10 +121,8 @@ def __create_jwt_payload(self, bauth_credentials: dict[str, str]) -> dict[str, U
             else:
                 bauth_credentials = request.get_json()
         if self.personal_credentials is not None:
-            bauth_credentials[self.personal_credentials[0]] = bauth_credentials["username"]
-            bauth_credentials[self.personal_credentials[1]] = bauth_credentials["password"]
-            del bauth_credentials["username"]
-            del bauth_credentials["password"]
+            bauth_credentials[self.personal_credentials[0]] = bauth_credentials.pop("username")
+            bauth_credentials[self.personal_credentials[1]] = bauth_credentials.pop("password")
         payload = bauth_credentials
         payload.update(self.jwt_fields_attr)