Spaces:
Running
on
Zero
Running
on
Zero
File size: 712 Bytes
df6c67d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import re
from requests import Response
API_KEY_PATTERN = re.compile(r"api_key=(.[^&]*)")
KEY_VALUE_GROUP = 1
MIN_KEY_LENGTH_TO_REVEAL_PREFIX = 8
def api_key_safe_raise_for_status(response: Response) -> None:
request_is_successful = response.status_code < 400
if request_is_successful:
return None
response.url = API_KEY_PATTERN.sub(deduct_api_key, response.url)
response.raise_for_status()
def deduct_api_key(match: re.Match) -> str:
key_value = match.group(KEY_VALUE_GROUP)
if len(key_value) < MIN_KEY_LENGTH_TO_REVEAL_PREFIX:
return f"api_key=***"
key_prefix = key_value[:2]
key_postfix = key_value[-2:]
return f"api_key={key_prefix}***{key_postfix}"
|