You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
4.3 KiB
Python
94 lines
4.3 KiB
Python
4 weeks ago
|
from pathlib import Path
|
||
|
|
from setuptools import setup, find_packages
|
||
|
|
from magic_pdf.libs.version import __version__
|
||
|
|
|
||
|
|
|
||
|
|
def parse_requirements(filename):
|
||
|
|
with open(filename) as f:
|
||
|
|
lines = f.read().splitlines()
|
||
|
|
|
||
|
|
requires = []
|
||
|
|
|
||
|
|
for line in lines:
|
||
|
|
if "http" in line:
|
||
|
|
pkg_name_without_url = line.split('@')[0].strip()
|
||
|
|
requires.append(pkg_name_without_url)
|
||
|
|
else:
|
||
|
|
requires.append(line)
|
||
|
|
|
||
|
|
return requires
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
with Path(Path(__file__).parent,
|
||
|
|
'README.md').open(encoding='utf-8') as file:
|
||
|
|
long_description = file.read()
|
||
|
|
setup(
|
||
|
|
name="magic_pdf", # 项目名
|
||
|
|
version=__version__, # 自动从tag中获取版本号
|
||
|
|
license="AGPL-3.0",
|
||
|
|
packages=find_packages() + ["magic_pdf.resources"] + ["magic_pdf.model.sub_modules.ocr.paddleocr2pytorch.pytorchocr.utils.resources"], # 包含所有的包
|
||
|
|
package_data={
|
||
|
|
"magic_pdf.resources": ["**"], # 包含magic_pdf.resources目录下的所有文件
|
||
|
|
"magic_pdf.model.sub_modules.ocr.paddleocr2pytorch.pytorchocr.utils.resources": ["**"], # pytorchocr.resources目录下的所有文件
|
||
|
|
},
|
||
|
|
install_requires=parse_requirements('requirements.txt'), # 项目依赖的第三方库
|
||
|
|
extras_require={
|
||
|
|
"lite": [
|
||
|
|
"paddleocr==2.7.3",
|
||
|
|
"paddlepaddle==3.0.0b1;platform_system=='Linux'",
|
||
|
|
"paddlepaddle==2.6.1;platform_system=='Windows' or platform_system=='Darwin'",
|
||
|
|
],
|
||
|
|
"full": [
|
||
|
|
"matplotlib>=3.10,<4",
|
||
|
|
"ultralytics>=8.3.48,<9", # yolov8,公式检测
|
||
|
|
"doclayout_yolo==0.0.2b1", # doclayout_yolo
|
||
|
|
"dill>=0.3.8,<1", # doclayout_yolo
|
||
|
|
"rapid_table>=1.0.5,<2.0.0", # rapid_table
|
||
|
|
"PyYAML>=6.0.2,<7", # yaml
|
||
|
|
"ftfy>=6.3.1,<7", # unimernet_hf
|
||
|
|
"openai>=1.70.0,<2", # openai SDK
|
||
|
|
"shapely>=2.0.7,<3", # imgaug-paddleocr2pytorch
|
||
|
|
"pyclipper>=1.3.0,<2", # paddleocr2pytorch
|
||
|
|
"omegaconf>=2.3.0,<3", # paddleocr2pytorch
|
||
|
|
],
|
||
|
|
"full_old_linux": [
|
||
|
|
"matplotlib>=3.10,<=3.10.1",
|
||
|
|
"ultralytics>=8.3.48,<=8.3.104", # yolov8,公式检测
|
||
|
|
"doclayout_yolo==0.0.2b1", # doclayout_yolo
|
||
|
|
"dill==0.3.8", # doclayout_yolo
|
||
|
|
"PyYAML==6.0.2", # yaml
|
||
|
|
"ftfy==6.3.1", # unimernet_hf
|
||
|
|
"openai==1.71.0", # openai SDK
|
||
|
|
"shapely==2.1.0", # imgaug-paddleocr2pytorch
|
||
|
|
"pyclipper==1.3.0.post6", # paddleocr2pytorch
|
||
|
|
"omegaconf==2.3.0", # paddleocr2pytorch
|
||
|
|
"albumentations==1.4.20", # 1.4.21引入的simsimd不支持2019年及更早的linux系统
|
||
|
|
"rapid_table==1.0.3", # rapid_table新版本依赖的onnxruntime不支持2019年及更早的linux系统
|
||
|
|
],
|
||
|
|
},
|
||
|
|
description="A practical tool for converting PDF to Markdown", # 简短描述
|
||
|
|
long_description=long_description, # 详细描述
|
||
|
|
long_description_content_type="text/markdown", # 如果README是Markdown格式
|
||
|
|
project_urls={
|
||
|
|
"Home": "https://mineru.net/",
|
||
|
|
"Repository": "https://github.com/opendatalab/MinerU",
|
||
|
|
},
|
||
|
|
keywords=["magic-pdf, mineru, MinerU, convert, pdf, markdown"],
|
||
|
|
classifiers=[
|
||
|
|
"Programming Language :: Python :: 3.10",
|
||
|
|
"Programming Language :: Python :: 3.11",
|
||
|
|
"Programming Language :: Python :: 3.12",
|
||
|
|
"Programming Language :: Python :: 3.13",
|
||
|
|
],
|
||
|
|
python_requires=">=3.10,<3.14", # 项目依赖的 Python 版本
|
||
|
|
entry_points={
|
||
|
|
"console_scripts": [
|
||
|
|
"magic-pdf = magic_pdf.tools.cli:cli",
|
||
|
|
"magic-pdf-dev = magic_pdf.tools.cli_dev:cli"
|
||
|
|
],
|
||
|
|
}, # 项目提供的可执行命令
|
||
|
|
include_package_data=True, # 是否包含非代码文件，如数据文件、配置文件等
|
||
|
|
zip_safe=False, # 是否使用 zip 文件格式打包，一般设为 False
|
||
|
|
)
|