import base64
def safe_base64_decode(s): if isinstance(s, str): s += (4 - len(s) % 4) '=' return base64.b64decode(s) else: s += (4 - len(s) % 4) b'=' return base64.b64decode(s)
assert b'abcd' == safe_base64_decode(b'YWJjZA=='), safe_base64_decode('YWJjZA==') assert b'abcd' == safe_base64_decode(b'YWJjZA'), safe_base64_decode('YWJjZA') print('ok')
Sign in to make a reply
alohaHall
!/usr/bin/env python3
---- coding:utf-8 ----
import base64
def safe_base64_decode(s): if isinstance(s, str): s += (4 - len(s) % 4) '=' return base64.b64decode(s) else: s += (4 - len(s) % 4) b'=' return base64.b64decode(s)
测试:
assert b'abcd' == safe_base64_decode(b'YWJjZA=='), safe_base64_decode('YWJjZA==') assert b'abcd' == safe_base64_decode(b'YWJjZA'), safe_base64_decode('YWJjZA') print('ok')