diff --git a/creyPY/services/auth0/manage.py b/creyPY/services/auth0/manage.py index d809859..28682fb 100644 --- a/creyPY/services/auth0/manage.py +++ b/creyPY/services/auth0/manage.py @@ -8,7 +8,7 @@ @cached(cache) def get_management_token() -> str: - re = requests.post( + response = requests.post( f"https://{AUTH0_DOMAIN}/oauth/token", json={ "client_id": AUTH0_CLIENT_ID, @@ -16,5 +16,6 @@ def get_management_token() -> str: "audience": f"https://{AUTH0_DOMAIN}/api/v2/", # This should be the management audience "grant_type": "client_credentials", }, + timeout=5, # Add a timeout parameter to avoid hanging requests ).json() - return re["access_token"] + return response["access_token"] diff --git a/creyPY/services/auth0/utils.py b/creyPY/services/auth0/utils.py index 3cb209e..917ff4b 100644 --- a/creyPY/services/auth0/utils.py +++ b/creyPY/services/auth0/utils.py @@ -54,6 +54,7 @@ def get_user(sub) -> dict: re = requests.get( f"https://{AUTH0_DOMAIN}/api/v2/users/{sub}", headers={"Authorization": f"Bearer {get_management_token()}"}, + timeout=5, ) if re.status_code != 200: raise HTTPException(re.status_code, re.json()) @@ -65,6 +66,7 @@ def patch_user(input_obj: dict, sub) -> dict: f"https://{AUTH0_DOMAIN}/api/v2/users/{sub}", headers={"Authorization": f"Bearer {get_management_token()}"}, json=input_obj, + timeout=5, ) if re.status_code != 200: raise HTTPException(re.status_code, re.json()) @@ -92,6 +94,7 @@ def request_verification_mail(sub: str) -> None: f"https://{AUTH0_DOMAIN}/api/v2/jobs/verification-email", headers={"Authorization": f"Bearer {get_management_token()}"}, json={"user_id": sub}, + timeout=5, ) if re.status_code != 201: raise HTTPException(re.status_code, re.json()) @@ -109,6 +112,7 @@ def create_user_invite(email: str) -> dict: "verify_email": False, "app_metadata": {"invitedToMyApp": True}, }, + timeout=5, ) if re.status_code != 201: raise HTTPException(re.status_code, re.json()) @@ -124,6 +128,7 @@ def password_change_mail(email: str) -> bool: "email": email, "connection": "Username-Password-Authentication", }, + timeout=5, ) if re.status_code != 200: