mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-07-29 12:38:27 -04:00
57e9fc9ba1
This changeset copies everything in notifications_utils to turn into a local project module. It includes dependency changes to account for this adjustment and the notifications_utils tests. It also fixes all of the import statements. Signed-off-by: Carlo Costino <carlo.costino@gsa.gov>
23 lines
588 B
Python
23 lines
588 B
Python
from base64 import urlsafe_b64decode, urlsafe_b64encode
|
|
from uuid import UUID
|
|
|
|
|
|
def base64_to_bytes(key):
|
|
return urlsafe_b64decode(key + "==")
|
|
|
|
|
|
def bytes_to_base64(bytes):
|
|
# remove trailing = to save precious bytes
|
|
return urlsafe_b64encode(bytes).decode("ascii").rstrip("=")
|
|
|
|
|
|
def base64_to_uuid(value):
|
|
# uuids are 16 bytes, and will always have two ==s of padding
|
|
return UUID(bytes=urlsafe_b64decode(value.encode("ascii") + b"=="))
|
|
|
|
|
|
def uuid_to_base64(value):
|
|
if not isinstance(value, UUID):
|
|
value = UUID(value)
|
|
return bytes_to_base64(value.bytes)
|