URLsafe Base64 encoding and decoding without trailing '='

base64util.py

from typing import Union
import base64


def base64encode(data: Union[bytes, str]) -> bytes:
    data = data.encode() if type(data) is str else data
    return base64.urlsafe_b64encode(data).rstrip(b'=')


def base64decode(b64: Union[bytes, str]) -> bytes:
    b64 = b64.encode() if type(b64) is str else b64
    return base64.urlsafe_b64decode(b64 + b'=' * (-len(b64) % 4))