您可以使用正则表达式来删除无效部分:
import re
invalid_tail = re.compile(r'[^a-zA-Z0-9+/=\n\r].*$')
def remove_tail(base64_value):
return invalid_tail.sub('', base64_value)
[^a-zA-Z0-9+/=\n\r]
匹配任何不是有效 Base64 字符的字符,以及尾随填充=
和换行符和回车符(允许在编码值中换行)。
演示:
>>> example = 'The quick brown fox jumps over the lazy dog!'.encode('base64')
>>> remove_tail(example + '*This is a tail').decode('base64')
'The quick brown fox jumps over the lazy dog!'
或者,使用样本的可解码部分:
>>> data = "3AQEFGh8NFgoqDwYsAQALDVFPVltYVkNGXldCRFUbNRk=----------:jhawrewre:--\r\n"
>>> remove_tail(data).decode('base64')
'\xdc\x04\x04\x14h|4X(\xa8<\x18\xb0\x04\x00,5E=YmaY\r\x19y]\t\x11Tl\xd4d'
该解决方案轻松击败了itertools.takewhile()
速度选项:
>>> import timeit
>>> text = "gw3AQEFGh8NFgoqDwYsAQALDVFPVltYVkNGXldCRFUbNRk=----------:jhawrewre:--\r\n"
>>> timeit.timeit('test(text)', 'from __main__ import with_takewhile as test, text')
11.785380125045776
>>> timeit.timeit('test(text)', 'from __main__ import with_re as test, text')
1.480334997177124
对于这个简单的示例,使用正则表达式几乎快 10 倍;对于更长的文本,结果会更快。