p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

island: Extract method _create_access_token() from _get_credentials_from_request()

This commit is contained in:
Mike Salvatore 2021-05-04 12:14:25 -04:00
parent a8646fc056
commit c7d47fee9c
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
monkey/monkey_island/cc/resources/auth/auth.py
View File

@ -41,12 +41,7 @@ class Authenticate(flask_restful.Resource):
(username, password) = Authenticate._get_credentials_from_request(request)
if self._credentials_match_registered_user(username, password):
access_token = flask_jwt_extended.create_access_token(
identity=user_store.UserStore.username_table[username].id
)
logger.debug(
f"Created access token for user {username} that begins with {access_token[:4]}"
)
access_token = Authenticate._create_access_token(username)
return make_response({"access_token": access_token, "error": ""}, 200)
else:
return make_response({"error": "Invalid credentials"}, 401)
@ -68,6 +63,17 @@ class Authenticate(flask_restful.Resource):
return False
@staticmethod
def _create_access_token(username):
access_token = flask_jwt_extended.create_access_token(
identity=user_store.UserStore.username_table[username].id
)
logger.debug(
f"Created access token for user {username} that begins with {access_token[:4]}"
)
return access_token
# See https://flask-jwt-extended.readthedocs.io/en/stable/custom_decorators/
def jwt_required(fn):