|
功能解析：使用python对指定文件夹的所有pdf进行截图并且保存到目标文件夹。步骤：一，遍历存储目标文件夹的存储pdf的文件夹；二，在指定的路径创建原先存储pdf的文件夹，名称相同；三，遍历存储pdf的文件，截图并存储到目标文件夹。
import os
import datetime
import fitz
NOW_DAY = datetime.date.today().strftime('%m-%d')
#存储pdf的文件
FEILE_ADDR = r'路径'
#输出目标文件夹
TARGET_ADDR = r'路径'
#存储pdf的文件夹名称
RP_DIRS_ARR = []
#截取pdf的图片并且保存
def pdf_image(pdf_name,save_addr,rp_name):
img_paths = []
pdf = fitz.Document(pdf_name)
#for i,pg in enumerate(range(0, pdf.pageCount)):
arr=[0,int(pdf.page_count/2)-1,pdf.page_count-1]
for i,pg in enumerate(arr):
print(pg)
page = pdf[pg] # 获得每一页的对象
trans = fitz.Matrix(3.0, 3.0).preRotate(0)
pm = page.getPixmap(matrix=trans, alpha=False) # 获得每一页的流对象
img_path = save_addr+ '\\' +rp_name+ '_' + str(pg + 1) + '.jpg'
print(img_path)
pm.writePNG(img_path) # 保存图片
img_paths.append(img_path)
pdf.close()
return img_paths
#遍历保存pdf文件夹名称
for root,dirs,files in os.walk(FEILE_ADDR+NOW_DAY):
RP_DIRS_ARR = dirs
print(RP_DIRS_ARR)
break
#在指定的路径创建文件夹
if not os.path.exists(TARGET_ADDR+NOW_DAY):
os.mkdir(TARGET_ADDR+NOW_DAY)
for dirs in RP_DIRS_ARR:
if not os.path.exists(TARGET_ADDR+NOW_DAY+'\\'+dirs):
os.mkdir(TARGET_ADDR+NOW_DAY+'\\'+dirs)
#获取pdf文件名称并截取图片
for rp_dirs in RP_DIRS_ARR:
for root,pdirs,files in os.walk(FEILE_ADDR+NOW_DAY+'\\'+rp_dirs):
pdf_arr = []
for file_name in files:
if(file_name[-3:]=='pdf'):
pdf_arr.append(file_name)
for pdf_name in pdf_arr:
pdf_image(FEILE_ADDR+NOW_DAY+'\\'+rp_dirs+'\\'+pdf_name, TARGET_ADDR+NOW_DAY+'\\'+rp_dirs,rp_dirs)
break
|