setup.py 1.2 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
32
33
34
35
        python_requires=">=3.9",  # 项目依赖的 Python 版本
        # entry_points={"console_scripts": ["my_command=my_project.main:run"]}, # 项目提供的可执行命令
        include_package_data=True,  # 是否包含非代码文件,如数据文件、配置文件等
        zip_safe=False,  # 是否使用 zip 文件格式打包,一般设为 False
    )