Python中读取PDF页数遍历文件名称和父级文件夹名称

笔记：

#给定文件路径
#遍历子文件
#读取PDF页数

# 第一步安装PyPDF2  pip install PyPDF2
# 第二步导入相应的模块
import os
from PyPDF2 import PdfFileReader


# 第三步：定义相对应的函数
def get_num_pages(file_path):
    """
    获取文件总页码
    :param file_path: 文件路径
    :return:
    """
    reader = PdfFileReader(file_path)
    # 不解密可能会报错：PyPDF2.utils.PdfReadError: File has not been decrypted
    if reader.isEncrypted:
        reader.decrypt('')
    page_num = reader.getNumPages()
    return page_num

#遍历文件名
def get_files():
    for filepath,dirnames,filenames in os.walk(r'E:\test2'):
        for filename in filenames:
            print(os.path.join(filepath,filename))

def list_all_files(rootdir):

    _files = []

    # 列出文件夹下所有的目录与文件
    list_file = os.listdir(rootdir)

    for i in range(0, len(list_file)):

        # 构造路径
        path = os.path.join(rootdir, list_file[i])

        # 判断路径是否是一个文件目录或者文件
        # 如果是文件目录，继续递归

        if os.path.isdir(path):
            _files.extend(list_all_files(path))
        if os.path.isfile(path):
            _files.append(path)
    return _files




if __name__ == '__main__':
    src_dir = r'C:\Users\Administrator\Desktop\众锐广告10-13'  # 源文件目录地址
    a=0
    files = list_all_files(src_dir)
    for i in range(0, len(files)):
        if ".pdf" in files[i]:
            a=a + 1
            pages=get_num_pages(files[i])
            print(files[i].split("\\")[-2]+"\t"+files[i].split("\\")[-1]+"\t页数\t"+str(pages))##父级文件夹+文件名+页码数量
print("总共:"+str(a)+"个文件")##输出文件数量

代码来源于本论坛，本人修改自定义输出，用作学习和自己提高工作效率使用，如有侵权，请告知！
2022-10-13