6b40f95033
Co-authored-by: Eve832 <eve.meelan@grafana.com>
Co-authored-by: Francisco Montes de Oca <nevermind89x@gmail.com>
Co-authored-by: Ildar Iskhakov <ildar.iskhakov@grafana.com>
Co-authored-by: Innokentii Konstantinov <innokenty.konstantinov@grafana.com>
Co-authored-by: Julia <ferril.darkdiver@gmail.com>
Co-authored-by: maskin25 <kengurek@gmail.com>
Co-authored-by: Matias Bordese <mbordese@gmail.com>
Co-authored-by: Matvey Kukuy <motakuk@gmail.com>
Co-authored-by: Michael Derynck <michael.derynck@grafana.com>
Co-authored-by: Richard Hartmann <richih@richih.org>
Co-authored-by: Robby Milo <robbymilo@fastmail.com>
Co-authored-by: Timur Olzhabayev <timur.olzhabayev@grafana.com>
Co-authored-by: Vadim Stepanov <vadimkerr@gmail.com>
Co-authored-by: Yulia Shanyrova <yulia.shanyrova@grafana.com>
45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
import binascii
|
|
from os import urandom as generate_bytes
|
|
from typing import Tuple
|
|
|
|
from cryptography.hazmat.backends import default_backend
|
|
from cryptography.hazmat.primitives import hashes
|
|
from cryptography.hazmat.primitives.hashes import SHA512
|
|
|
|
from apps.auth_token import constants
|
|
|
|
sha = SHA512
|
|
|
|
|
|
def generate_token_string() -> str:
|
|
num_bytes = int(constants.AUTH_TOKEN_CHARACTER_LENGTH / 2)
|
|
return binascii.hexlify(generate_bytes(num_bytes)).decode()
|
|
|
|
|
|
def generate_short_token_string() -> str:
|
|
num_bytes = int(constants.AUTH_SHORT_TOKEN_CHARACTER_LENGTH / 2)
|
|
return binascii.hexlify(generate_bytes(num_bytes)).decode()
|
|
|
|
|
|
def hash_token_string(token_string: str) -> str:
|
|
digest = hashes.Hash(sha(), backend=default_backend())
|
|
digest.update(binascii.unhexlify(token_string))
|
|
return binascii.hexlify(digest.finalize()).decode()
|
|
|
|
|
|
def generate_plugin_token_string_and_salt(stack_id: int, org_id: int) -> Tuple[str, str]:
|
|
random_salt = generate_bytes(int(constants.AUTH_TOKEN_CHARACTER_LENGTH / 2))
|
|
return generate_plugin_token_string(random_salt, stack_id, org_id), binascii.hexlify(random_salt).decode()
|
|
|
|
|
|
def generate_plugin_token_string(salt: bytes, stack_id: int, org_id: int) -> str:
|
|
digest = hashes.Hash(sha(), backend=default_backend())
|
|
digest.update(salt)
|
|
digest.update(bytes(stack_id))
|
|
digest.update(bytes(org_id))
|
|
return binascii.hexlify(digest.finalize()).decode()
|
|
|
|
|
|
def generate_schedule_token_string() -> str:
|
|
num_bytes = int(constants.SCHEDULE_EXPORT_TOKEN_CHARACTER_LENGTH / 2)
|
|
return binascii.hexlify(generate_bytes(num_bytes)).decode()
|