* format files Signed-off-by: Jens Langhammer <jens@goauthentik.io> * fix pyright Signed-off-by: Jens Langhammer <jens@goauthentik.io> * revert #8367 Signed-off-by: Jens Langhammer <jens@goauthentik.io> * sigh Signed-off-by: Jens Langhammer <jens@goauthentik.io> --------- Signed-off-by: Jens Langhammer <jens@goauthentik.io>
27 lines
698 B
Python
27 lines
698 B
Python
"""ID/Secret Generators"""
|
|
|
|
import string
|
|
from random import SystemRandom
|
|
|
|
|
|
def generate_code_fixed_length(length=9) -> str:
|
|
"""Generate a numeric code"""
|
|
rand = SystemRandom()
|
|
num = rand.randrange(1, 10**length)
|
|
return str(num).zfill(length)
|
|
|
|
|
|
def generate_id(length=40) -> str:
|
|
"""Generate a random client ID"""
|
|
rand = SystemRandom()
|
|
return "".join(rand.choice(string.ascii_letters + string.digits) for x in range(length))
|
|
|
|
|
|
def generate_key(length=128) -> str:
|
|
"""Generate a suitable client secret"""
|
|
rand = SystemRandom()
|
|
return "".join(
|
|
rand.choice(string.ascii_letters + string.digits + string.punctuation)
|
|
for x in range(length)
|
|
)
|