Compare commits

...

3 Commits

Author SHA1 Message Date
5daddf260e fix: added timeouts to the requests to fix Bandit issue 2024-11-25 13:20:17 +01:00
364e07daa1 fix: fixed random issue (codacy) 2024-11-25 13:14:07 +01:00
5daf6eb8c5 fix: fixed missing import 2024-11-25 12:55:35 +01:00
4 changed files with 16 additions and 9 deletions

View File

@@ -1,4 +1,4 @@
import random import secrets
import string import string
@@ -6,11 +6,11 @@ def create_random_password(length: int = 12) -> str:
all_characters = string.ascii_letters + string.digits + string.punctuation all_characters = string.ascii_letters + string.digits + string.punctuation
password = [ password = [
random.choice(string.ascii_lowercase), secrets.choice(string.ascii_lowercase),
random.choice(string.ascii_uppercase), secrets.choice(string.ascii_uppercase),
random.choice(string.digits), secrets.choice(string.digits),
random.choice(string.punctuation), secrets.choice(string.punctuation),
] ]
password += random.choices(all_characters, k=length - 4) password += [secrets.choice(all_characters) for _ in range(length - 4)]
random.shuffle(password) secrets.SystemRandom().shuffle(password)
return "".join(password) return "".join(password)

View File

@@ -0,0 +1 @@
from .auth0 import * # noqa

View File

@@ -8,7 +8,7 @@ cache = TTLCache(maxsize=100, ttl=600)
@cached(cache) @cached(cache)
def get_management_token() -> str: def get_management_token() -> str:
re = requests.post( response = requests.post(
f"https://{AUTH0_DOMAIN}/oauth/token", f"https://{AUTH0_DOMAIN}/oauth/token",
json={ json={
"client_id": AUTH0_CLIENT_ID, "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 "audience": f"https://{AUTH0_DOMAIN}/api/v2/", # This should be the management audience
"grant_type": "client_credentials", "grant_type": "client_credentials",
}, },
timeout=5, # Add a timeout parameter to avoid hanging requests
).json() ).json()
return re["access_token"] return response["access_token"]

View File

@@ -54,6 +54,7 @@ def get_user(sub) -> dict:
re = requests.get( re = requests.get(
f"https://{AUTH0_DOMAIN}/api/v2/users/{sub}", f"https://{AUTH0_DOMAIN}/api/v2/users/{sub}",
headers={"Authorization": f"Bearer {get_management_token()}"}, headers={"Authorization": f"Bearer {get_management_token()}"},
timeout=5,
) )
if re.status_code != 200: if re.status_code != 200:
raise HTTPException(re.status_code, re.json()) 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}", f"https://{AUTH0_DOMAIN}/api/v2/users/{sub}",
headers={"Authorization": f"Bearer {get_management_token()}"}, headers={"Authorization": f"Bearer {get_management_token()}"},
json=input_obj, json=input_obj,
timeout=5,
) )
if re.status_code != 200: if re.status_code != 200:
raise HTTPException(re.status_code, re.json()) 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", f"https://{AUTH0_DOMAIN}/api/v2/jobs/verification-email",
headers={"Authorization": f"Bearer {get_management_token()}"}, headers={"Authorization": f"Bearer {get_management_token()}"},
json={"user_id": sub}, json={"user_id": sub},
timeout=5,
) )
if re.status_code != 201: if re.status_code != 201:
raise HTTPException(re.status_code, re.json()) raise HTTPException(re.status_code, re.json())
@@ -109,6 +112,7 @@ def create_user_invite(email: str) -> dict:
"verify_email": False, "verify_email": False,
"app_metadata": {"invitedToMyApp": True}, "app_metadata": {"invitedToMyApp": True},
}, },
timeout=5,
) )
if re.status_code != 201: if re.status_code != 201:
raise HTTPException(re.status_code, re.json()) raise HTTPException(re.status_code, re.json())
@@ -124,6 +128,7 @@ def password_change_mail(email: str) -> bool:
"email": email, "email": email,
"connection": "Username-Password-Authentication", "connection": "Username-Password-Authentication",
}, },
timeout=5,
) )
if re.status_code != 200: if re.status_code != 200: