30 lines
778 B
Python
30 lines
778 B
Python
import base64
|
|
import os
|
|
import string
|
|
from moto.moto_api._internal import mock_random as random
|
|
|
|
ACCOUNT_SPECIFIC_ACCESS_KEY_PREFIX = "8NWMTLYQ"
|
|
ACCOUNT_SPECIFIC_ASSUMED_ROLE_ID_PREFIX = "3X42LBCD"
|
|
SESSION_TOKEN_PREFIX = "FQoGZXIvYXdzEBYaD"
|
|
DEFAULT_STS_SESSION_DURATION = 3600
|
|
|
|
|
|
def random_session_token() -> str:
|
|
return (
|
|
SESSION_TOKEN_PREFIX
|
|
+ base64.b64encode(os.urandom(266))[len(SESSION_TOKEN_PREFIX) :].decode()
|
|
)
|
|
|
|
|
|
def random_assumed_role_id() -> str:
|
|
return (
|
|
ACCOUNT_SPECIFIC_ASSUMED_ROLE_ID_PREFIX + _random_uppercase_or_digit_sequence(9)
|
|
)
|
|
|
|
|
|
def _random_uppercase_or_digit_sequence(length: int) -> str:
|
|
return "".join(
|
|
str(random.choice(string.ascii_uppercase + string.digits))
|
|
for _ in range(length)
|
|
)
|