最近有一个小的功能实现,从小某书上下载指定帖子的图片们,然后批量插入到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
安装完毕后可以先看一下情况