41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
from flask import request, session
|
|
from requests_oauthlib import OAuth2Session
|
|
|
|
from config import *
|
|
from helpers import HydraError
|
|
|
|
|
|
class HydraOauth:
|
|
SESSION_KEY = "oauth_state"
|
|
|
|
@staticmethod
|
|
def authorize():
|
|
try:
|
|
hydra = OAuth2Session(HYDRA_CLIENT_ID)
|
|
authorization_url, state = hydra.authorization_url(
|
|
HYDRA_AUTHORIZATION_BASE_URL
|
|
)
|
|
|
|
# State is used to prevent CSRF, keep this for later.
|
|
session[HydraOauth.SESSION_KEY] = state
|
|
|
|
return authorization_url
|
|
except Exception as err:
|
|
raise HydraError(str(err), 500)
|
|
|
|
@staticmethod
|
|
def get_token(state, code):
|
|
try:
|
|
hydra = OAuth2Session(
|
|
client_id=HYDRA_CLIENT_ID,
|
|
state=state,
|
|
token_endpoint_auth_method="client_secret_basic",
|
|
)
|
|
token = hydra.fetch_token(
|
|
TOKEN_URL, code=code, state=state, client_secret=HYDRA_CLIENT_SECRET
|
|
)
|
|
|
|
session["hydra_token"] = token
|
|
return token
|
|
except Exception as err:
|
|
raise HydraError(str(err), 500)
|