import base64
import os


def base64_encode(data, output=None):
    """
    :param data: Bytes or String or FileStream.
    :param output: Output file stream, when the data parameter is the file stream valid.
    :return: What type of data entering returns the same data type.
    """
    if isinstance(data, bytes):
        return base64.b64encode(data)
    if isinstance(data, str):
        return base64.b64encode(bytes(data, encoding='utf-8')).decode()
    else:
        return base64.encode(data, output)


def base64_decode(data, output=None):
    """
    :param data: Bytes or String or FileStream.
    :param output: Output file stream, when the data parameter is the file stream valid.
    :return: What type of data entering returns the same data type.
    """
    if isinstance(data, bytes):
        return base64.b64decode(data)
    if isinstance(data, str):
        return base64.b64decode(bytes(data, encoding='utf-8')).decode()
    else:
        return base64.decode(data, output)


if __name__ == '__main__':
    # e.g. Code bytes type.
    print(base64_encode(bytes('Some text content.', encoding='utf-8')))
    # e.g. Code string type.
    print(base64_encode('Some text content.'))
    # e.g. Read the file and encode and write to another file.
    # 1.Create a test example file,
    open('./example_of_base64_encode', 'w').write('Some text content.')
    # 2.Then test,
    print(base64_encode(open('./example_of_base64_encode', 'rb'), open('./example_of_base64_encode.base64', 'wb')))
    # 3.Final cleanup.
    os.remove('./example_of_base64_encode')
    os.remove('./example_of_base64_encode.base64')