7x31
/
cmdb
mirror of https://github.com/veops/cmdb.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(api): oauth2.0 authentication

This commit is contained in:
pycook 2023-12-19 13:07:21 +08:00
parent aa98a304c1
commit 090007487d
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
cmdb-api/api/lib/perm/authentication/oauth2/routing.py
View File

@ -12,8 +12,10 @@ from flask import redirect
from flask import request from flask import request
from flask import session from flask import session
from flask import url_for from flask import url_for
from flask_login import login_user, logout_user from flask_login import login_user
from flask_login import logout_user
from six.moves.urllib.parse import urlencode from six.moves.urllib.parse import urlencode
from six.moves.urllib.parse import urlparse
from api.lib.common_setting.common_data import AuthenticateDataCRUD from api.lib.common_setting.common_data import AuthenticateDataCRUD
from api.lib.perm.acl.audit import AuditCRUD from api.lib.perm.acl.audit import AuditCRUD
@ -34,9 +36,12 @@ def login(auth_type):
auth_type = auth_type.upper() auth_type = auth_type.upper()
redirect_uri = "{}://{}{}".format(urlparse(request.referrer).scheme,
urlparse(request.referrer).netloc,
url_for('oauth2.callback', auth_type=auth_type.lower()))
qs = urlencode({ qs = urlencode({
'client_id': config['client_id'], 'client_id': config['client_id'],
'redirect_uri': url_for('oauth2.callback', auth_type=auth_type.lower(), _external=True), 'redirect_uri': redirect_uri,
'response_type': current_app.config[f'{auth_type}_RESPONSE_TYPE'], 'response_type': current_app.config[f'{auth_type}_RESPONSE_TYPE'],
'scope': ' '.join(config['scopes'] or []), 'scope': ' '.join(config['scopes'] or []),
'state': session[f'{auth_type.lower()}_state'], 'state': session[f'{auth_type.lower()}_state'],