Function _decode_base64_payload

astrbot/core/utils/media_utils.py:272–300

Decode a base64 payload while tolerating omitted padding. Args: payload: Base64 payload without a data URI header. error_message: Message to use when decoding fails. validate: Whether to ask ``base64.b64decode`` to reject non-base64 characters. Returns:

(
    payload: str,
    *,
    error_message: str,
    validate: bool = False,
)

def _decode_base64_payload(
    payload: str,
    *,
    error_message: str,
    validate: bool = False,
) -> bytes:
    """Decode a base64 payload while tolerating omitted padding.

    Args:
        payload: Base64 payload without a data URI header.
        error_message: Message to use when decoding fails.
        validate: Whether to ask ``base64.b64decode`` to reject non-base64
            characters.

    Returns:
        Decoded bytes.

    Raises:
        ValueError: Raised when the payload cannot be decoded.
    """
    payload = "".join(payload.split())
    missing_padding = len(payload) % 4
    if missing_padding:
        payload += "=" * (4 - missing_padding)

    try:
        return base64.b64decode(payload, validate=validate)
    except (binascii.Error, ValueError) as exc:
        raise ValueError(error_message) from exc


def describe_media_ref(media_ref: object | None) -> str: