识别pdf标题并重命名pdf

该文章介绍了一个改进的Python脚本,用于识别PDF文件中最大字体的句子,特别关注包含arXiv的标题部分,并解决期刊名可能为最大字体的问题。脚本通过fitz库操作PDF,找到并替换冒号为特殊符号以创建新的文件名。
摘要由CSDN通过智能技术生成

最新改进版:
https://blog.csdn.net/weixin_44751432/article/details/135247984

已导出为可执行文件
下载链接在评论区
此为改进版
解决了期刊名可能是最大字体的问题:
思路 识别字符串的空格>3则为标题

import os
import PyPDF2

import fitz

def find_largest_font_sentence(pdf_path):
    largest_font_size = 0
    largest_font_sentence = ''
    maxsize=0
    # 打开PDF文件
    document = fitz.open(pdf_path)

    for page_number in range(document.page_count):
        page = document.load_page(page_number)
        blocks = page.get_text("dict")["blocks"]

        for block in blocks:
            if "lines" in block:  # 检查是否存在 lines 字段
                for line in block["lines"]:
                    for span in line["spans"]:
                        if span["size"] > largest_font_size:
                            largest_font_size = span["size"]
                            largest_font_sentence = span["text"]
        maxsize=largest_font_size
        for block in blocks:
            if "lines" in block:  # 检查是否存在 lines 字段
                for line in block["lines"]:
                    for span in line["spans"]:
                        if span["size"] ==maxsize:
                            if largest_font_sentence!=span["text"]:
                                largest_font_sentence = largest_font_sentence+' '+span["text"]

        if largest_font_sentence.count(' ')<4:
            largest_font_sentence=''
            largest_font_size=0
            for block in blocks:
                if "lines" in block:  # 检查是否存在 lines 字段
                    for line in block["lines"]:
                        for span in line["spans"]:
                            if span["size"] > largest_font_size and span["text"].count(' ')>3:
                                largest_font_size = span["size"]
                                largest_font_sentence = span["text"]
            maxsize = largest_font_size
            for block in blocks:
                if "lines" in block:  # 检查是否存在 lines 字段
                    for line in block["lines"]:
                        for span in line["spans"]:
                            if span["size"] == maxsize:
                                if largest_font_sentence != span["text"]:
                                    largest_font_sentence = largest_font_sentence + ' ' + span["text"]

    return largest_font_sentence

# 用于测试的PDF文件路径



def rename_pdfs(directory):
    for filename in os.listdir(directory):
        if filename.endswith(".pdf"):
            filepath = os.path.join(directory, filename)
            filepath=filepath.strip()
            largest_font_sentence = find_largest_font_sentence(filepath)
            title=largest_font_sentence
            if  ":" in title:
                title = title.replace(":", ":")
            new_filename = title.strip() + ".pdf"
            new_filepath = os.path.join(directory, new_filename)
            os.rename(filepath, new_filepath)
            print(f"Renamed {filename} to {new_filename}")


# 指定包含英文论文PDF的目录路径
pdf_directory = "PDF文件目录"


# 调用函数进行重命名
rename_pdfs(pdf_directory)

再次更新
import os
import time
import PyPDF2

import fitz

def find_largest_font_sentence(pdf_path):

largest_font_size = 0

largest_font_sentence = ''

maxsize=0

# 打开PDF文件

document = fitz.open(pdf_path)



for page_number in range(1):

    page = document.load_page(page_number)

    blocks = page.get_text("dict")["blocks"]



    for block in blocks:

        if "lines" in block:  # 检查是否存在 lines 字段

            for line in block["lines"]:

                for span in line["spans"]:

                    if span["size"] > largest_font_size:

                        largest_font_size = span["size"]

                        largest_font_sentence = span["text"]

    maxsize=largest_font_size

    for block in blocks:

        if "lines" in block:  # 检查是否存在 lines 字段

            for line in block["lines"]:

                for span in line["spans"]:

                    if span["size"] ==maxsize:

                        if largest_font_sentence!=span["text"]:

                            largest_font_sentence = largest_font_sentence+' '+span["text"]



    if largest_font_sentence.count(' ')<4 or "arXiv" in largest_font_sentence:

        largest_font_sentence=''

        largest_font_size=0

        for block in blocks:

            if "lines" in block:  # 检查是否存在 lines 字段

                for line in block["lines"]:

                    for span in line["spans"]:

                        if span["size"] > largest_font_size and span["text"].count(' ')>3 and "arXiv" not in span["text"]:

                            largest_font_size = span["size"]

                            largest_font_sentence = span["text"]

        maxsize = largest_font_size

        for block in blocks:

            if "lines" in block:  # 检查是否存在 lines 字段

                for line in block["lines"]:

                    for span in line["spans"]:

                        if span["size"] == maxsize:

                            if largest_font_sentence != span["text"]:

                                largest_font_sentence = largest_font_sentence + ' ' + span["text"]



return largest_font_sentence

用于测试的PDF文件路径

def rename_pdfs(directory):

for filename in os.listdir(directory):

    if filename.endswith(".pdf"):

        filepath = os.path.join(directory, filename)

        filepath=filepath.strip()

        largest_font_sentence = find_largest_font_sentence(filepath)

        title=largest_font_sentence

        if  ":" in title:

            title = title.replace(":", ":")

        new_filename = title.strip() + ".pdf"

        new_filepath = os.path.join(directory, new_filename)

        os.rename(filepath, new_filepath)

        print(f"Renamed {filename} to {new_filename}")

指定包含英文论文PDF的目录路径

#比如pdf文件在 D:\学习\论文

#那么引号内需要输入的是 D:\学习\论文
print(“请输入论文的存放目录”)
a=input(“输入路径”)
a=a.replace(“\”,“/”)
pdf_directory = a

调用函数进行重命名

rename_pdfs(pdf_directory)
time.sleep(5)

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值