setup.py 1.25 KB
Newer Older
赵小蒙's avatar
赵小蒙 committed
1
from setuptools import setup, find_packages
赵小蒙's avatar
赵小蒙 committed
2
from magic_pdf.libs.version import __version__
赵小蒙's avatar
赵小蒙 committed
3
4


5
6
7
8
9
10
11
12
def parse_requirements(filename):
    with open(filename) as f:
        lines = f.read().splitlines()

    requires = []

    for line in lines:
        if "http" in line:
赵小蒙's avatar
赵小蒙 committed
13
14
            pkg_name_without_url = line.split('@')[0].strip()
            requires.append(pkg_name_without_url)
15
16
17
18
19
        else:
            requires.append(line)

    return requires

20

赵小蒙's avatar
赵小蒙 committed
21
22
23
if __name__ == '__main__':
    setup(
        name="magic_pdf",  # 项目名
赵小蒙's avatar
赵小蒙 committed
24
        version=__version__,  # 自动从tag中获取版本号
赵小蒙's avatar
赵小蒙 committed
25
26
        packages=find_packages(),  # 包含所有的包
        install_requires=parse_requirements('requirements.txt'),  # 项目依赖的第三方库
赵小蒙's avatar
赵小蒙 committed
27
28
29
30
        extras_require={
            "gpu": ["paddlepaddle-gpu"],
            "cpu": ["paddlepaddle"],
        },
赵小蒙's avatar
赵小蒙 committed
31
        python_requires=">=3.9",  # 项目依赖的 Python 版本
赵小蒙's avatar
update:  
赵小蒙 committed
32
33
34
35
36
        entry_points={
            "console_scripts": [
                "magic-pdf = magic_pdf.cli.magicpdf:cli"
            ],
        },  # 项目提供的可执行命令
赵小蒙's avatar
赵小蒙 committed
37
38
39
        include_package_data=True,  # 是否包含非代码文件,如数据文件、配置文件等
        zip_safe=False,  # 是否使用 zip 文件格式打包,一般设为 False
    )