aboutsummaryrefslogtreecommitdiff
path: root/gn_auth/auth/authentication
diff options
context:
space:
mode:
Diffstat (limited to 'gn_auth/auth/authentication')
-rw-r--r--gn_auth/auth/authentication/oauth2/grants/authorisation_code_grant.py8
-rw-r--r--gn_auth/auth/authentication/oauth2/grants/jwt_bearer_grant.py15
2 files changed, 9 insertions, 14 deletions
diff --git a/gn_auth/auth/authentication/oauth2/grants/authorisation_code_grant.py b/gn_auth/auth/authentication/oauth2/grants/authorisation_code_grant.py
index ae604df..a40292e 100644
--- a/gn_auth/auth/authentication/oauth2/grants/authorisation_code_grant.py
+++ b/gn_auth/auth/authentication/oauth2/grants/authorisation_code_grant.py
@@ -26,16 +26,16 @@ class AuthorisationCodeGrant(grants.AuthorizationCodeGrant):
GRANT_TYPE = "authorization_code"
RESPONSE_TYPES = {'code'}
- def create_authorization_response(self, redirect: str, grant_user):
+ def create_authorization_response(self, redirect_uri: str, grant_user):
"""Add some data to the URI"""
- response = super().create_authorization_response(redirect, grant_user)
+ response = super().create_authorization_response(
+ redirect_uri, grant_user)
headers = dict(response[-1])
headers = {
**headers,
"Location": f"{headers['Location']}&user_id={grant_user.user_id}"
}
- return (response[0], response[1], [
- (header, value) for header, value in headers.items()])
+ return (response[0], response[1], list(headers.items()))
def save_authorization_code(self, code, request):
"""Persist the authorisation code to database."""
diff --git a/gn_auth/auth/authentication/oauth2/grants/jwt_bearer_grant.py b/gn_auth/auth/authentication/oauth2/grants/jwt_bearer_grant.py
index 85c8d2e..895acb7 100644
--- a/gn_auth/auth/authentication/oauth2/grants/jwt_bearer_grant.py
+++ b/gn_auth/auth/authentication/oauth2/grants/jwt_bearer_grant.py
@@ -1,19 +1,12 @@
"""JWT as Authorisation Grant"""
-import uuid
-from urllib.parse import urlparse
-from datetime import datetime, timedelta
-
-from flask import request, current_app as app
-
-from authlib.jose import jwt
+from flask import current_app as app
from authlib.oauth2.rfc7523.jwt_bearer import JWTBearerGrant as _JWTBearerGrant
from authlib.oauth2.rfc7523.token import (
JWTBearerTokenGenerator as _JWTBearerTokenGenerator)
+from gn_auth.auth.db.sqlite3 import with_db_connection
from gn_auth.auth.authentication.users import user_by_id
-from gn_auth.auth.db.sqlite3 import connection, with_db_connection
-from gn_auth.auth.authentication.oauth2.models.oauth2client import client
class JWTBearerTokenGenerator(_JWTBearerTokenGenerator):
@@ -23,7 +16,9 @@ class JWTBearerTokenGenerator(_JWTBearerTokenGenerator):
DEFAULT_EXPIRES_IN = 300
- def get_token_data(self, grant_type, client, expires_in=300, user=None, scope=None):
+ def get_token_data(#pylint: disable=[too-many-arguments]
+ self, grant_type, client, expires_in=300, user=None, scope=None
+ ):
"""Post process data to prevent JSON serialization problems."""
tokendata = super().get_token_data(
grant_type, client, expires_in, user, scope)