Python转换Base64格式PNG为图片文件生成无效文件问题排查

我写了一段Python代码，从MySQL读取Base64编码的图片数据，解码后保存为PNG文件，但所有生成的文件都无法打开，图片查看器提示“Unknown format error”。不过这些Base64内容在浏览器里是能正常显示的，部分可能存在填充问题，但按道理转成本地文件应该没问题。以下是我的代码：

import mysql.connector
import base64
import binascii


# base64 decoding function

def decode_base64(data, image_type):
    if data is None:
        return None
    missing_padding = 4 - len(data) % 4
    if missing_padding:
        data += b'=' * missing_padding
    try:
        return base64.b64decode(data)
    except binascii.Error as e:
        print(f"Invalid image for id {id} image type: {image_type}, Invalid Padding")
    return base64.b64decode(data)

# Connect to the MySQL server
cnx = mysql.connector.connect(user='myuser', password='my_password',
                              host='127.0.0.1',
                              database='cms')



# Create a cursor
cursor = cnx.cursor()

# Execute the SELECT statement
cursor.execute("SELECT id, icon, icon_large FROM games")

# Iterate through the results
for (id, icon, icon_large) in cursor:
    image_type = 'icon'
    try:
        # Decode the base64-encoded images
        icon_data = decode_base64(icon, 'icon')
        icon_large_data = decode_base64(icon_large, 'icon_large')

        # Write the images to files if the data is not None
        if icon_data is not None:
            with open("icon_{}.png".format(id), "wb") as f:
                f.write(icon_data)
    except binascii.Error as e:
        print(f"Invalid image for id {id} image type: {image_type}")
    image_type = 'icon_large'
    try:
        if icon_large_data is not None:
            with open("icon_large_{}.png".format(id), "wb") as f:
                f.write(icon_large_data)
    except binascii.Error as e:
        print(f"Invalid image for id {id} image type: {image_type}")

# Close the cursor and connection
cursor.close()
cnx.close()

排查与修复方案 #

我帮你梳理了几个核心问题，这些应该是导致图片无法打开的主要原因：

1. Base64数据可能包含浏览器兼容的前缀 #

浏览器能正常显示的Base64图片通常会带有data:image/png;base64,这类前缀，但你的代码直接解码原始数据——如果数据库里的内容带这个前缀，解码出来的二进制数据会包含前缀的文本内容，自然不是有效的图片文件。

2. 解码函数的错误处理逻辑有漏洞 #

你的decode_base64函数在捕获binascii.Error后，又再次调用base64.b64decode(data)，这会让错误再次抛出，而且没有正确返回异常情况下的结果，可能导致后续写入错误的二进制数据。

3. 强制指定PNG后缀可能不匹配实际图片格式 #

你统一把文件保存为.png，但数据库里的图片可能是JPG、GIF等其他格式，后缀不匹配会导致图片查看器识别失败。

修改后的完整代码 #

import mysql.connector
import base64
import binascii

def decode_base64(data, image_type, item_id):
    if data is None:
        return None, None  # 返回解码后的数据和实际图片格式
    
    # 处理字符串类型的Base64数据（如果数据库存储的是字符串而非bytes）
    if isinstance(data, str):
        data = data.encode('utf-8')
    
    # 移除浏览器常用的data URI前缀（比如data:image/png;base64,）
    img_format = 'png'  # 默认格式
    if data.startswith(b'data:image/'):
        try:
            header_part, data_part = data.split(b';base64,', 1)
            img_format = header_part.split(b'/')[1].decode('utf-8')
            data = data_part
        except ValueError:
            print(f"Malformed data URI for id {item_id} image type: {image_type}")
    
    # 处理Base64填充问题，只有当需要填充时才添加=
    missing_padding = len(data) % 4
    if missing_padding != 0:
        data += b'=' * (4 - missing_padding)
    
    try:
        # 使用validate=True严格验证Base64格式
        decoded_data = base64.b64decode(data, validate=True)
        return decoded_data, img_format
    except binascii.Error as e:
        print(f"Invalid Base64 for id {item_id} image type: {image_type}, Error: {str(e)}")
        return None, None

# 连接数据库
cnx = mysql.connector.connect(user='myuser', password='my_password',
                              host='127.0.0.1',
                              database='cms')

cursor = cnx.cursor()
cursor.execute("SELECT id, icon, icon_large FROM games")

for (item_id, icon, icon_large) in cursor:
    # 处理小图标
    icon_data, icon_format = decode_base64(icon, 'icon', item_id)
    if icon_data is not None:
        with open(f"icon_{item_id}.{icon_format}", "wb") as f:
            f.write(icon_data)
        print(f"Successfully saved: icon_{item_id}.{icon_format}")
    
    # 处理大图标
    large_data, large_format = decode_base64(icon_large, 'icon_large', item_id)
    if large_data is not None:
        with open(f"icon_large_{item_id}.{large_format}", "wb") as f:
            f.write(large_data)
        print(f"Successfully saved: icon_large_{item_id}.{large_format}")

# 关闭连接资源
cursor.close()
cnx.close()

关键修改说明 #

• 新增了移除data:image/...;base64,前缀的逻辑，这是最可能的问题根源
• 解码函数返回实际图片格式，避免强制用PNG后缀保存
• 优化了填充处理逻辑，只在需要时添加填充符
• 使用validate=True让Base64解码更严格，提前发现无效数据
• 把id参数改名为item_id避免和内置函数冲突，同时传递到解码函数里方便错误定位

内容的提问来源于stack exchange，提问作者Erlandas K