请写一个能处理去掉=的base64解码函数——python学习笔记

1. 题目：

请写一个能处理去掉=的base64解码函数：
题目是廖雪峰老师的python教程中base64的练习。

2. 代码如下：

import base64


def safe_base64_decode(s):

    # 判断是否是4的整数u，不够的在末尾添加等号
    if len(s) % 4 != 0:
        s = s + bytes('=', encoding='utf-8') * (4 - len(s) % 4)

    # 解决字符串和bytes类型
    if not isinstance(s, bytes):
        s = bytes(s, encoding='utf-8')

    # 解码 
    base64_str = base64.b64decode(s)

    return base64_str


# 测试:
assert b'abcd' == safe_base64_decode(b'YWJjZA=='), safe_base64_decode('YWJjZA==')
assert b'abcd' == safe_base64_decode(b'YWJjZA'), safe_base64_decode('YWJjZA')

希望能够帮助到大家，有什么问题可以 直接评论即可，如果不够详细的话也可以说，我会及时回复的。