python base64 编码与自定义转换表

import base64import randomimport stringdef test_base64_encode():    result = None    # 要编码的数据，是bytes类型    # str --> byte    print("hello".encode('utf-8'))    # byte    b'hello'    # base64编码    temp01 = "hello"    # 对数据进行Base64编码，编码后也是byte类型    encoded_data = base64.b64encode(temp01.encode())    print(encoded_data)    # byte --> str    print("byte --> str " + encoded_data.decode('utf-8'))def test_base64_decode():    str1 = "aGVsbG8="    # 对Base64编码的字符串进行解码，返回值是byte    decoded_data = base64.b64decode(str1)    print(decoded_data)    print(decoded_data.decode())"""自定义/随机字符集"""def test_custom_base64():    normal_base_64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/="    str_list = list(normal_base_64)    random.shuffle(str_list)    custom_base_64 = "".join(str_list)    print("随机字符集: " + custom_base_64)    str1 = "test"    print("原版: " + base64.b64encode(str1.encode()).decode())    base64_temp = EncodeCustomBase64(str1.encode(), custom_base_64)    print("自定义后: " + base64_temp)    decode_temp = DecodeCustomBase64(base64_temp, custom_base_64)    print("自定义解码: " + decode_temp)# custom encoding functiondef EncodeCustomBase64(stringToBeEncoded,custom_b64):    standard_b64 = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/='    encoded = base64.b64encode(stringToBeEncoded)    final_encoded = ""    for ch in encoded.decode():        if (str(ch) in custom_b64):            final_encoded += custom_b64[standard_b64.find(str(ch))]        elif (ch == '='):            final_encoded += '='    return final_encoded# custom decoding functiondef DecodeCustomBase64(stringToBeDecoded:str,custom_b64):    standard_b64 = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/='    temp_decoded = ""    for ch in stringToBeDecoded:        if (str(ch) in custom_b64):            temp_decoded += standard_b64[custom_b64.find(str(ch))]        elif (ch == '='):            temp_decoded += '='    decoded = base64.b64decode(temp_decoded)    return decoded.decode()if __name__ == '__main__':    # test_base64_encode()    # test_base64_decode()    test_custom_base64()