python-计算文件夹里的.pdf文件的总页数

KjPrime

已于 2022-08-01 10:28:55 修改

阅读量1.7k

点赞数 1

分类专栏： python 文章标签： python PyPDF2

于 2022-03-24 17:25:17 首次发布

本文链接：https://blog.csdn.net/qq_45019494/article/details/123716652

版权

python 专栏收录该内容

28 篇文章 2 订阅

订阅专栏

该博客介绍了一个Python脚本，用于统计指定文件夹及其子文件夹中所有PDF文件的总页数。通过使用PyPDF2库，脚本可以逐个读取PDF并计算页数。此外，它还提供了查找文件夹内PDF文件的功能。这个脚本对于需要预估打印成本或管理大量PDF文档的场景非常实用。

摘要由CSDN通过智能技术生成

为什么做这个？

因为有时候我们需要打印，需要知道自己的总页数然后算钱。
如果有word，毕竟去打印店，转化为.pdf打印好些。

代码展示

先安装PyPDF2依赖

pip install PyPDF2

代码

from PyPDF2 import PdfFileReader as pr
import os


def single_pdf_count(pdf_path: str) -> int:  #计算单个pdf的页数
    count = 0
    try:
        f = open(pdf_path, 'rb')
        pdf = pr(f)
        # print(pdf_path, "正在计算...")
        count += pdf.getNumPages()
        f.close()
    except:
        print(pdf_path, '该文件出现异常，可能是权限问题或者被加密了')
    return count


def folder_pdf_files(folder: str) -> list[str]:  #一个文件夹里面有多少pdf文件
    file_list = []
    for a, b, c in os.walk(folder):
        for filename in c:
            if filename[-3:].lower() == 'pdf':
                file_path = os.path.join(a, filename)
                file_list.append(file_path)
    print(folder, ": 有", len(file_list), "个pdf文件")
    return file_list


def folders_pdf_pages(folders: list[str]) -> int:  #一个文件夹列表中的pdf有多少页
    print("正在计算...")
    pdf_files = []
    for folder in folders:
        pdf_files += folder_pdf_files(folder)
    count = 0
    for pdf_path in pdf_files:
        count += single_pdf_count(pdf_path)
    return count


if __name__ == '__main__':
    folders_list = ["文件夹目录1", "文件夹目录2"]
    pdf_count = folders_pdf_pages(folders_list)
    print("总的页数：", pdf_count)