Discuss / Python / 可以解码bytes和str类型，还可以解码包含--__的
Back

可以解码bytes和str类型，还可以解码包含--__的

Topic source

用户5393841808

#1 Created at ... [Delete] [Delete and Lock User] 
import base64

def safe_base64_decode(s):
    if isinstance(s, bytes):
        s = s.decode('utf-8')
    return base64.urlsafe_b64decode(s + '=' * ((4 - len(s) % 4) % 4))

if __name__ == '__main__':
    assert b'abcd' == safe_base64_decode(b'YWJjZA=='), safe_base64_decode('YWJjZA==')
    assert b'abcd' == safe_base64_decode(b'YWJjZA'), safe_base64_decode('YWJjZA')
    assert b'abcd\x0f\xbe' == safe_base64_decode(b'YWJjZA--'), safe_base64_decode('YWJjZA--')
  • 1

Reply