python转换并提取pdf文件中的图片

复制代码
#安装fitz包
pip install pymupdf 

脚本如下所示:

复制代码
import fitz
import re
import os
import time
import sys

arguments = sys.argv

for arg in arguments:
    print(arg)


def file_name_list(base_dir):
    for i, j, k in os.walk(base_dir):
        name = [i.replace('.pdf', '') for i in k]
        return name


def pdfExtractPic(filePath, pic_path):
    ret = {}
    try:
        iNum = 0
        with fitz.open(filePath) as doc:
            for page in range(len(doc)):
                for image in doc.get_page_images(page):
                    xref = image[0]
                    pix = fitz.Pixmap(doc, xref)
                    iNum += 1
                    fileName = ("%s\%d.png"%(pic_path, iNum))
                    if pix.n < 5:
                        pix.save(fileName)
                    else:
                        pix1 = fitz.Pixmap(fitz.csRGB, pix)
                        pix1.save(fileName)
                        pix1 = None

    except Exception as e:
        print(e)

    return ret

 
def pdf2image(path, pic_path, image_name):
    try:
        with fitz.open(path) as pdf:
            for pg in range(0, pdf.page_count):
                page = pdf[pg]
        # 设置缩放和旋转系数,zoom_x, zoom_y取相同值,表示等比例缩放
                mat = fitz.Matrix(2, 2)
                pm = page.get_pixmap(matrix=mat, alpha=False)
        # if width or height > 2000 pixels, don't enlarge the image
                if pm.width > 2000 or pm.height > 2000:
                    pm = page.get_pixmap(matrix=fitz.Matrix(1, 1), alpha=False)
                page_num = pg + 1  # 页码从1开始
                new_name = '%s_img_%d.png' % (image_name, page_num)
                pm.save(os.path.join(pic_path, new_name)) 
    except Exception as error:
        print(error)                
 
 
if __name__ == '__main__':
    if len(arguments) < 2:
        print("请输入pdf所在目录的路径.")
        sys.exit(1)
        
    base_dir = arguments[1]

    name_list = file_name_list(base_dir)
    start = time.time()
    for name in name_list:
        image_name = name
        file_name = r'\%s.pdf' % name  # PDF 文件路径
        file_path = base_dir + file_name
        pic_name = r'\pic\%s' % name  # 存放图片的文件夹
        pic_path = base_dir + pic_name
        try:
            os.makedirs(pic_path)   # 根据路径,创建对应路径下的文件夹
        except Exception as error:
            print(error)        
        pdf2image(file_path, pic_path, image_name)
        #print(pic_path)
        pic_path += "\\sub\\"
        try:
            os.makedirs(pic_path)   # 根据路径,创建对应路径下的文件夹
        except Exception as error:
            print(error)   
        pdfExtractPic(file_path, pic_path)
    end = time.time()
    print('task is over: %.2f' % (end-start))

如果执行的时候报错:

复制代码
RuntimeError: Directory 'static' does not exist

说明fitz安装错了版本,参考:PyMuPDF: AttributeError:模块'fitz'没有属性'open'_attributeerror: module 'fitz' has no attribute 'op-CSDN博客

相关推荐
weixin_472339463 小时前
高效处理大体积Excel文件的Java技术方案解析
java·开发语言·excel
枯萎穿心攻击4 小时前
响应式编程入门教程第二节:构建 ObservableProperty<T> — 封装 ReactiveProperty 的高级用法
开发语言·unity·c#·游戏引擎
Eiceblue5 小时前
【免费.NET方案】CSV到PDF与DataTable的快速转换
开发语言·pdf·c#·.net
m0_555762906 小时前
Matlab 频谱分析 (Spectral Analysis)
开发语言·matlab
浪裡遊7 小时前
React Hooks全面解析:从基础到高级的实用指南
开发语言·前端·javascript·react.js·node.js·ecmascript·php
烛阴7 小时前
简单入门Python装饰器
前端·python
lzb_kkk7 小时前
【C++】C++四种类型转换操作符详解
开发语言·c++·windows·1024程序员节
好开心啊没烦恼8 小时前
Python 数据分析:numpy,说人话,说说数组维度。听故事学知识点怎么这么容易?
开发语言·人工智能·python·数据挖掘·数据分析·numpy
面朝大海,春不暖,花不开8 小时前
使用 Python 实现 ETL 流程:从文本文件提取到数据处理的全面指南
python·etl·原型模式
简佐义的博客8 小时前
破解非模式物种GO/KEGG注释难题
开发语言·数据库·后端·oracle·golang