从数据库中读出图片插入word

一、前期准备

1、数据库配置与连接

确保你有一个包含图片数据的数据库，常见的数据库类型有MySQL、Oracle、SQL Server等，以MySQL为例，假设数据库中有一张名为images_table的表，其中包含两个字段：image_id（用于唯一标识图片，数据类型可以是INT等）和image_data（存储图片二进制数据，数据类型为BLOB）。

要与数据库建立连接，需要使用相应的数据库连接库，在Python中，对于MySQL可以使用pymysql库，安装该库后，通过以下代码建立连接：

import pymysql
connection = pymysql.connect(host='localhost', user='your_username', password='your_password', database='your_database_name')
cursor = connection.cursor()

这里需要将localhost替换为数据库服务器地址（如果是本地数据库则保持此值），your_username替换为数据库用户名，your_password替换为密码，your_database_name替换为数据库名称。

2、Word文档操作准备

要操作Word文档，可使用python comtypes库（适用于Windows系统）或python docx库（跨平台），以python docx为例，安装该库后，可以通过以下代码创建一个新的Word文档对象：

from docx import Document
doc = Document()

二、从数据库读取图片并插入Word

1、查询数据库获取图片数据

使用SQL语句从数据库中查询图片数据，要获取所有图片，可以执行以下SQL查询：

select_sql = "SELECT image_id, image_data FROM images_table"
cursor.execute(select_sql)
results = cursor.fetchall()

这将返回一个包含所有图片记录的列表，每个元素是一个元组，包含image_id和image_data。

2、将图片数据写入文件并插入Word

遍历查询结果，对于每一条记录：

提取image_data字段，它是一个二进制数据，将其写入一个临时文件（例如JPEG格式），假设图片格式为JPEG，文件名为image_{image_id}.jpg：

for result in results:
    image_id = result[0]
    image_data = result[1]
    with open(f"image_{image_id}.jpg", "wb") as f:
        f.write(image_data)

在Word文档中插入图片，使用docx库的add_picture方法：

    doc.add_picture(f"image_{image_id}.jpg", width=Inches(5))  # 可根据需要设置图片宽度，Inches是docx库中的单位

3、保存Word文档

完成所有图片插入后，保存Word文档，将文档保存为output.docx：

doc.save("output.docx")

4、关闭数据库连接

操作完成后，关闭数据库连接：

cursor.close()
connection.close()

三、示例代码整合

以下是完整的示例代码：

import pymysql
from docx import Document
from docx.shared import Inches
数据库连接配置
connection = pymysql.connect(host='localhost', user='your_username', password='your_password', database='your_database_name')
cursor = connection.cursor()
创建Word文档对象
doc = Document()
查询数据库获取图片数据
select_sql = "SELECT image_id, image_data FROM images_table"
cursor.execute(select_sql)
results = cursor.fetchall()
将图片数据写入文件并插入Word
for result in results:
    image_id = result[0]
    image_data = result[1]
    with open(f"image_{image_id}.jpg", "wb") as f:
        f.write(image_data)
    doc.add_picture(f"image_{image_id}.jpg", width=Inches(5))
保存Word文档
doc.save("output.docx")
关闭数据库连接
cursor.close()
connection.close()

四、相关问答FAQs

1、问：如果图片格式不是JPEG怎么办？

答：如果图片格式不是JPEG，需要根据实际格式修改写入文件时的扩展名和add_picture方法中的参数，如果是PNG格式，将文件扩展名改为.png，并在add_picture方法中指定对应的图片格式参数，如format=MSO.PICTURE_TYPE_PNG（MSO.PICTURE_TYPE_PNG是python docx库中表示PNG格式的常量）。

2、问：如何处理大量图片插入导致内存不足的问题？

答：当处理大量图片时，可能会出现内存不足的情况，一种解决方法是分批处理图片，例如每次从数据库中读取一定数量的图片（如100张），处理完后保存Word文档，然后清空文档对象，再继续读取下一批图片进行处理，这样可以有效控制内存使用，避免一次性加载过多图片导致内存溢出，在处理过程中可以适当增加垃圾回收机制，如使用gc.collect()函数手动触发垃圾回收。