FM / josepy

/usr/lib/python3/dist-packages/josepy

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
b64.py	1489	0644	edit dl rm
errors.py	791	0644	edit dl rm
interfaces.py	7790	0644	edit dl rm
json_util.py	18548	0644	edit dl rm
jwa.py	7484	0644	edit dl rm
jwk.py	13985	0644	edit dl rm
jws.py	15489	0644	edit dl rm
magic_typing.py	556	0644	edit dl rm
py.typed	0	0644	edit dl rm
util.py	10049	0644	edit dl rm
__init__.py	1952	0644	edit dl rm

Edit: /usr/lib/python3/dist-packages/josepy/b64.py (1489B)

"""`JOSE Base64`_ is defined as:

- URL-safe Base64
  - padding stripped

.. _`JOSE Base64`:
    https://tools.ietf.org/html/draft-ietf-jose-json-web-signature-37#appendix-C

.. Do NOT try to call this module "base64", as it will "shadow" the
   standard library.

"""
import base64
from typing import Union

def b64encode(data: bytes) -> bytes:
    """JOSE Base64 encode.

:param data: Data to be encoded.
    :type data: bytes

:returns: JOSE Base64 string.
    :rtype: bytes

:raises TypeError: if ``data`` is of incorrect type

"""
    if not isinstance(data, bytes):
        raise TypeError("argument should be bytes")
    return base64.urlsafe_b64encode(data).rstrip(b"=")

def b64decode(data: Union[bytes, str]) -> bytes:
    """JOSE Base64 decode.

:param data: Base64 string to be decoded. If it's unicode, then
                 only ASCII characters are allowed.
    :type data: bytes or unicode

:returns: Decoded data.
    :rtype: bytes

:raises TypeError: if input is of incorrect type
    :raises ValueError: if input is unicode with non-ASCII characters

"""
    if isinstance(data, str):
        try:
            data = data.encode("ascii")
        except UnicodeEncodeError:
            raise ValueError("unicode argument should contain only ASCII characters")
    elif not isinstance(data, bytes):
        raise TypeError("argument should be a str or unicode")

return base64.urlsafe_b64decode(data + b"=" * (4 - (len(data) % 4)))

"""`JOSE Base64`_ is defined as: - URL-safe Base64 - padding stripped .. _`JOSE Base64`: https://tools.ietf.org/html/draft-ietf-jose-json-web-signature-37#appendix-C .. Do NOT try to call this module "base64", as it will "shadow" the standard library. """ import base64 from typing import Union def b64encode(data: bytes) -> bytes: """JOSE Base64 encode. :param data: Data to be encoded. :type data: bytes :returns: JOSE Base64 string. :rtype: bytes :raises TypeError: if ``data`` is of incorrect type """ if not isinstance(data, bytes): raise TypeError("argument should be bytes") return base64.urlsafe_b64encode(data).rstrip(b"=") def b64decode(data: Union[bytes, str]) -> bytes: """JOSE Base64 decode. :param data: Base64 string to be decoded. If it's unicode, then only ASCII characters are allowed. :type data: bytes or unicode :returns: Decoded data. :rtype: bytes :raises TypeError: if input is of incorrect type :raises ValueError: if input is unicode with non-ASCII characters """ if isinstance(data, str): try: data = data.encode("ascii") except UnicodeEncodeError: raise ValueError("unicode argument should contain only ASCII characters") elif not isinstance(data, bytes): raise TypeError("argument should be a str or unicode") return base64.urlsafe_b64decode(data + b"=" * (4 - (len(data) % 4)))