最近有一个小的功能实现,从小某书上下载指定帖子的图片们,然后批量插入到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 datafile_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_titledoc_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:passi=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+1gif2jpg('./jpg/'+filename)#有时图片会打不开,用这个函数处理一下就可以了。time.sleep(2)
二、插入word文档
import os
from docx import Document
from docx.shared import Inchesdef add_images_to_word(folder_path, output_doc):# Create a new Word documentdoc = Document()# Iterate through all files in the given folderfor 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 documentdoc.add_picture(file_path, width=Inches(6))# Optionally, add a caption or break# doc.add_paragraph(filename)#doc.add_paragraph("\n")# Save the documentdoc.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
安装完毕后可以先看一下情况