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

233 阅读 0 评论 154 点赞

我是靠谱客的博主秀丽路灯，这篇文章主要介绍请写一个能处理去掉=的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')