百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术资源 > 正文

三步教你用Elasticsearch+PyMuPDF实现PDF大文件秒搜!

off999 2025-06-30 19:19 17 浏览 0 评论

面对100页以上的大型PDF文件时,阅读和搜索往往效率低下。传统关系型数据库在处理此类数据时容易遇到性能瓶颈,而 Elasticsearch 凭借其强大的全文检索和分布式架构,成为理想解决方案。通过 PyMuPDF 提取PDF文本内容,结合Elasticsearch的倒排索引技术,能够快速定位和检索信息,即使面对海量数据也能保持高效查询。其分布式特性支持多节点扩展,进一步提升性能和可靠性,完美解决传统数据库的性能瓶颈问题。

以下是详细的开发方案:

1. 系统设计

  • 功能需求

(1)支持上传 PDF 文件,并从中提取文本。
(2)支持分词操作以及全文检索。
(3)支持将匹配内容以高亮形式显示。
(4)支持分页展示搜索结果

  • 技术栈
    • PyMuPDF:用于解析PDF文件。
    • jieba:用于中文分词。
    • Elasticsearch:用于存储和检索文本数据。
    • Flask:用于构建Web界面和API。

2. 实现步骤

2.1 安装依赖

Python

pip install pymupdf flask jieba elasticsearch

2.2 启动Elasticsearch

(1)下载并安装Elasticsearch

(2)从 Elasticsearch官网(
https://www.elastic.co/cn/elasticsearch) 下载并解压

(3)运行 bin/elasticsearch 启动服务

(4)确保Elasticsearch运行在 http://localhost:9200

2.3 解析PDF并存入Elasticsearch

使用PyMuPDF提取PDF文本,并将文本数据存入Elasticsearch

Python

import fitz  # PyMuPDF
from elasticsearch import Elasticsearch
import jieba

# 连接Elasticsearch
es = Elasticsearch("http://localhost:9200")

def extract_and_store_pdf(filepath):
    # 提取文件名
    filename = filepath.split('/')[-1]

    # 解析PDF
    doc = fitz.open(filepath)
    for page_num in range(len(doc)):
        page = doc.load_page(page_num)
        text = page.get_text()

        # 对文本进行分词
        tokens = jieba.lcut(text)

        # 存入Elasticsearch
        doc_data = {
            "filename": filename,
            "page": page_num + 1,
            "text": text,
            "tokens": tokens
        }
        es.index(index="pdf_documents", body=doc_data)

# 示例:解析并存储PDF
extract_and_store_pdf('example.pdf')

2.4 实现分词搜索

使用Elasticsearch的全文检索功能实现分词搜索:

Python

def search_keyword(keyword):
    # 对关键词进行分词
    keyword_tokens = jieba.lcut(keyword)

    # 构建Elasticsearch查询
    query = {
        "query": {
            "match": {
                "tokens": " ".join(keyword_tokens)  # 将分词结果组合成查询字符串
            }
        },
        "highlight": {
            "fields": {
                "text": {}  # 高亮显示匹配内容
            }
        }
    }

    # 执行查询
    response = es.search(index="pdf_documents", body=query)
    return response['hits']['hits']

# 示例:搜索关键词
results = search_keyword('Python编程')
for hit in results:
    print(f"文件: {hit['_source']['filename']}, 页码: {hit['_source']['page']}")
    print(f"内容: {hit['highlight']['text'][0]}")

2.5 构建Web界面

使用Flask构建一个简单的Web界面,支持文件上传和搜索:

Python

from flask import Flask, render_template, request, redirect, url_for
import os

app = Flask(__name__)
UPLOAD_FOLDER = 'uploads'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

# 首页:上传文件
@app.route('/', methods=['GET', 'POST'])
def index():
    if request.method == 'POST':
        file = request.files['file']
        if file:
            filepath = os.path.join(app.config['UPLOAD_FOLDER'], file.filename)
            file.save(filepath)
            extract_and_store_pdf(filepath)
            return redirect(url_for('index'))
    return render_template('index.html')

# 搜索页面
@app.route('/search', methods=['GET'])
def search():
    keyword = request.args.get('keyword')
    if keyword:
        results = search_keyword(keyword)
        return render_template('search.html', results=results, keyword=keyword)
    return render_template('search.html')

if __name__ == '__main__':
    if not os.path.exists(UPLOAD_FOLDER):
        os.makedirs(UPLOAD_FOLDER)
    app.run(debug=True)

2.6 创建HTML模板

在 templates 目录下创建以下HTML文件:

index.html(文件上传页面):

html



search.html(搜索页面):

html



3. 运行系统

(1) 启动Elasticsearch:

bash

bin/elasticsearch

(2) 启动Flask应用:

bash

python app.py

(3) 访问 http://127.0.0.1:5000,上传PDF文件。

(4) 访问
http://127.0.0.1:5000/search,输入关键词搜索。

4. 功能扩展

  • 分页显示:在Elasticsearch查询中添加 from 和 size 参数,实现分页。
  • 停用词过滤:在分词时过滤常见停用词(如“的”、“是”)。
  • 词性标注:结合词性标注(如名词、动词)提升搜索精度。
  • 性能优化:使用Elasticsearch的分布式特性,支持大规模数据检索。
  • 语义化搜索:通过理解用户意图和上下文,返回相关结果,而非简单关键词匹配。

通过以上步骤,你可以构建一个高性能的PDF文件搜索系统!

创作不易,您的建议,是我们进步的动力!欢迎留言及加关注,帮助我们做得更好!

相关推荐

大文件传不动?WinRAR/7-Zip 入门到高手,这 5 个技巧让你效率翻倍

“这200张照片怎么传给女儿?微信发不了,邮箱附件又超限……”62岁的张阿姨对着电脑犯愁时,儿子只用了3分钟就把照片压缩成一个文件,还教她:“以后用压缩软件,比打包行李还方便!”职场人更懂这...

电脑解压缩软件推荐——7-Zip:免费、高效、简洁的文件管理神器

在日常工作中,我们经常需要处理压缩文件。无论是下载软件包、接收文件,还是存储大量数据,压缩和解压缩文件都成为了我们日常操作的一部分。而说到压缩解压软件,7-Zip绝对是一个不可忽视的名字。今天,我就来...

设置了加密密码zip文件要如何打开?这几个方法可以试试~

Zip是一种常见的压缩格式文件,文件还可以设置密码保护。那设置了密码的Zip文件要如何打开呢?不清楚的小伙伴一起来看看吧。当我们知道密码想要打开带密码的Zip文件,我们需要用到适用于Zip格式的解压缩...

大文件想要传输成功,怎么把ZIP文件分卷压缩

不知道各位小伙伴有没有这样的烦恼,发送很大很大的压缩包会受到限制,为此,想要在压缩过程中将文件拆分为几个压缩包并且同时为所有压缩包设置加密应该如何设置?方法一:使用7-Zip免费且强大的文件管理工具7...

高效处理 RAR 分卷压缩包:合并解压操作全攻略

在文件传输和存储过程中,当遇到大文件时,我们常常会使用分卷压缩的方式将其拆分成多个较小的压缩包,方便存储和传输。RAR作为一种常见的压缩格式,分卷压缩包的使用频率也很高。但很多人在拿到RAR分卷...

2个方法教你如何删除ZIP压缩包密码

zip压缩包设置了加密密码,每次解压文件都需要输入密码才能够顺利解压出文件,当压缩包文件不再需要加密的时候,大家肯定想删除压缩包密码,或是忘记了压缩包密码,想要通过删除操作将压缩包密码删除,就能够顺利...

速转!漏洞预警丨压缩软件Winrar目录穿越漏洞

WinRAR是一款功能强大的压缩包管理器,它是档案工具RAR在Windows环境下的图形界面。该软件可用于备份数据,缩减电子邮件附件的大小,解压缩从Internet上下载的RAR、ZIP及其它类...

文件解压方法和工具分享_文件解压工具下载

压缩文件减少文件大小,降低文件失效的概率,总得来说好处很多。所以很多文件我们下载下来都是压缩软件,很多小伙伴不知道怎么解压,或者不知道什么工具更好,所以今天做了文件解压方法和工具的分享给大家。一、解压...

[python]《Python编程快速上手:让繁琐工作自动化》学习笔记3

1.组织文件笔记(第9章)(代码下载)1.1文件与文件路径通过importshutil调用shutil模块操作目录,shutil模块能够在Python程序中实现文件复制、移动、改名和删除;同时...

Python内置tarfile模块:读写 tar 归档文件详解

一、学习目标1.1学习目标掌握Python内置模块tarfile的核心功能,包括:理解tar归档文件的原理与常见压缩格式(gzip/bz2/lzma)掌握tar文件的读写操作(创建、解压、查看、过滤...

使用python展开tar包_python拓展

类Unix的系统,打包文件经常使用的就是tar包,结合zip工具,可以方便的打包并解压。在python的标准库里面有tarfile库,可以方便实现生成了展开tar包。使用这个库最大的好处,可能就在于不...

银狐钓鱼再升级:白文件脚本化实现GO语言后门持久驻留

近期,火绒威胁情报中心监测到一批相对更为活跃的“银狐”系列变种木马。火绒安全工程师第一时间获取样本并进行分析。分析发现,该样本通过阿里云存储桶下发恶意文件,采用AppDomainManager进行白利...

ZIP文件怎么打开?2个简单方法教你轻松搞定!

在日常工作和生活中,我们经常会遇到各种压缩文件,其中最常见的格式之一就是ZIP。ZIP文件通过压缩数据来减少文件大小,方便我们进行存储和传输。然而,对于初学者来说,如何打开ZIP文件可能会成为一个小小...

Ubuntu—解压多个zip压缩文件.zip .z01 .z02

方法将所有zip文件放在同一目录中:zip_file.z01,zip_file.z02,zip_file.z03,...,zip_file.zip。在Zip3.0版本及以上,使用下列命令:将所有zi...

如何使用7-Zip对文件进行加密压缩

7-Zip是一款开源的文件归档工具,支持多种压缩格式,并提供了对压缩文件进行加密的功能。使用7-Zip可以轻松创建和解压.7z、.zip等格式的压缩文件,并且可以通过设置密码来保护压缩包中的...

取消回复欢迎 发表评论: