python - 如何正确地将base64字符串转换为图像文件？

Question

您好，我正在尝试将 Base64 字符串转换为图像文件，这很简单。这是我的代码。

with open(os.path.join('avatars', avatar_filename), 'wb') as avatar_file:
    avatar_file.write(base64.decodebytes(avatar_b64))

但问题是该字符串是由客户端发送的，所以我如何知道base64字符串是否对应于图像文件？

Answer 1

最后我决定使用Python自带的imghdr库。因此，我将 Base64 图像保存到文件中，然后检查该文件以了解是否是有效图像，如下所示:

if imghdr.what(avatar_filename):
    print('is a image')
else:
    print('is not a image')

imghdr.what() 函数返回图像的类型，因此如果没有返回类型，我假设这不是图像。