使用python下载图片且批量将图片插入word文档

最近有一个小的功能实现，从小某书上下载指定帖子的图片们，然后批量插入到word文档中，便于打印。于是有了以上需求。

一、下载图片

1、首先获取图片们的链接img_urls

首先，获取到的指定帖子的所有信息可以存入一个json文件中，如下样式：

读取这个json文件，获取title和image_list。

def read_json_file(file_path):
    with open(file_path, 'r',encoding='utf-8') as file:
        data = json.load(file)
    return data

file_path='这个json文件的路径'
data = read_json_file(file_path)
print(data)

one_data=data[0]

#获取title，并格式化为可以命名文件夹的样式
import re
def validateTitle(title):
    rstr = r"[\/\\\:\*\?\"\<\>\|]"  # '/ \ : * ? " < > |'
    new_title = re.sub(rstr, "_", title)  # 替换为下划线
    return new_title

doc_name=one_data['title']
doc_name=validateTitle(doc_name)

#获取图片们的链接
img_list=one_data['image_list']
print(img_list)

2、批量下载图片们，上代码

def gif2jpg(path):
    #使用Image模块的open()方法打开gif动态图像时，默认是第一帧
    im = Image.open(path)
    # pngDir = gifFileName[:-4]
    #创建存放每帧图片的文件夹
    # os.mkdir(pngDir)
    try:
     # while True:
      #保存当前帧图片
      # current = im.tell()
      im.save(path)
      #获取下一帧图片
      # im.seek(current+1)
    except EOFError:
      pass

i=1
for img_url in img_urls:
    print(img_url)
    response = requests.get(img_url)
    # 获取的文本实际上是图片的二进制文本
    img = response.content
    # 将他拷贝到本地文件 w 写  b 二进制  wb代表写入二进制文本
    filename=str(i)+'.png'#给文件命名，接下来存放到一个叫做'jpg'的文件夹中
    with open( './jpg/'+filename,'wb' ) as f:
       f.write(img)
    time.sleep(2)
    i=i+1
    gif2jpg('./jpg/'+filename)#有时图片会打不开，用这个函数处理一下就可以了。
    time.sleep(2)

二、插入word文档

import os
from docx import Document
from docx.shared import Inches

def add_images_to_word(folder_path, output_doc):
    # Create a new Word document
    doc = Document()

    # Iterate through all files in the given folder
    for filename in os.listdir(folder_path):
        if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif')):
            file_path = os.path.join(folder_path, filename)
            # Add image to the document
            doc.add_picture(file_path, width=Inches(6))

            # Optionally, add a caption or break
            # doc.add_paragraph(filename)
            #doc.add_paragraph("\n")

    # Save the document
    doc.save(output_doc)
    print(f"Document saved as {output_doc}")

# Define the folder containing images and the output document name
image_folder = "jpg"#存放图片的文件夹
doc_name='my_doc'#就是第一段代码中的tilte
output_document ='f"{doc_name}.docx"#输出到指定名字的word中

# Add images to the Word document
add_images_to_word(image_folder, output_document)

注意首先安装需要的库。

pip38 install python-docx

安装完毕后可以先看一下情况