/
githubmirror
/
PromethAI-Backend
Обзор
Документация
Войти
/
githubmirror
/
PromethAI-Backend
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
auth/auth_utils.py
61 строка
2 KB
vasa
Added working version of Auth with example, refactor of the API to support adding additional assistants
01 июл 2023, 17:40
01 июл 2023, 17:40
c5ef56d
Код
Авторство
О чём код?
from cognito.JWTBearer import JWKS, JWTBearer, JWTAuthorizationCredentials import requests region = "eu-west-1" user_pool_id = "" #needed cognito_url = f"https://cognito-idp.{region}.amazonaws.com/{user_pool_id}/.well-known/jwks.json" # Fetch the JWKS using the updated URL jwks = JWKS.parse_obj(requests.get(cognito_url).json()) print(jwks) auth = JWTBearer(jwks) import requests # Set the Cognito authentication endpoint URL auth = JWTBearer(jwks) # Set the user credentials username = "" #needed password = "" #needed # Create the authentication payload payload = { "username": username, "password": password } # Set the Cognito authentication endpoint URL # Set the Cognito token endpoint URL token_endpoint = f"https://your-cognito-domain.auth.{region}.amazoncognito.com/oauth2/token" # Set the client credentials client_id = "" #needed client_secret = "" import boto3 def authenticate_and_get_token(username: str, password: str, user_pool_id: str, app_client_id: str) -> None: client = boto3.client('cognito-idp') resp = client.admin_initiate_auth( UserPoolId=user_pool_id, ClientId=app_client_id, AuthFlow='ADMIN_NO_SRP_AUTH', AuthParameters={ "USERNAME": username, "PASSWORD": password } ) print("Log in success") print("Access token:", resp['AuthenticationResult']['AccessToken']) print("ID token:", resp['AuthenticationResult']['IdToken']) authenticate_and_get_token(username, password, user_pool_id, client_id)