SEAL-Python/setup.py at main · Huelse/SEAL-Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
import os
import platform
from glob import glob
from pathlib import Path
from shutil import copy2
import sys
import sysconfig
from setuptools import setup
from distutils.sysconfig import get_python_inc
import pybind11
from pybind11.setup_helpers import Pybind11Extension, build_ext

__version__ = "4.1.2.1"
BASE_DIR = Path(__file__).resolve().parent

include_dirs = [get_python_inc(), pybind11.get_include(), 'SEAL/native/src', 'SEAL/build/native/src']

if platform.system() == "Windows":
    extra_objects = sorted(
        glob('SEAL/build/lib/*.lib')
        + glob('SEAL/build/lib/Release/*.lib')
        + glob('SEAL/build/**/libseal*.lib', recursive=True)
    )
else:
    extra_objects = sorted(glob('SEAL/build/lib/*.a'))

cpp_args = ['/std:c++17', '/utf-8', '/bigobj'] if platform.system() == "Windows" else ['-std=c++17']
link_args = []

if platform.system() == "Darwin":
    deployment_target = os.environ.get("MACOSX_DEPLOYMENT_TARGET") or sysconfig.get_config_var(
        "MACOSX_DEPLOYMENT_TARGET"
    )
    if deployment_target:
        deployment_flag = f"-mmacosx-version-min={deployment_target}"
        os.environ.setdefault("MACOSX_DEPLOYMENT_TARGET", deployment_target)
        cpp_args.append(deployment_flag)
        link_args.append(deployment_flag)

if len(extra_objects) < 1 or not os.path.exists(extra_objects[0]):
    print('Not found the seal lib file, check the "SEAL/build/lib"')
    sys.exit(1)


class build_ext_with_typing(build_ext):
    """Copy PEP 561 typing files next to the compiled extension."""
    typing_files = ("seal.pyi", "py.typed")

    def run(self):
        super().run()
        self._copy_typing_files()

    def _copy_typing_files(self):
        output_dirs = {Path(self.build_lib)}
        if self.inplace:
            output_dirs.add(BASE_DIR)

        for target_dir in output_dirs:
            target_dir.mkdir(parents=True, exist_ok=True)
            for filename in self.typing_files:
                source = BASE_DIR / filename
                if source.exists():
                    destination = target_dir / filename
                    if source.resolve() == destination.resolve():
                        continue
                    copy2(source, destination)


ext_modules = [
    Pybind11Extension(
        "seal",
        sorted(glob('src/*.cpp')),
        include_dirs=include_dirs,
        extra_compile_args=cpp_args,
        extra_link_args=link_args,
        extra_objects=extra_objects,
        define_macros=[('VERSION_INFO', __version__)],
    ),
]

setup(
    name="seal-python",
    version=__version__,
    author="Huelse",
    author_email="topmaxz@protonmail.com",
    url="https://github.com/Huelse/SEAL-Python",
    description="Python wrapper for the Microsoft SEAL",
    long_description=(BASE_DIR / "README.md").read_text(encoding="utf-8"),
    long_description_content_type="text/markdown",
    ext_modules=ext_modules,
    cmdclass={"build_ext": build_ext_with_typing},
    zip_safe=False,
    license='MIT',
    python_requires=">=3.8",
    classifiers=[
        "Development Status :: 4 - Beta",
        "Intended Audience :: Developers",
        "Intended Audience :: Science/Research",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3 :: Only",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
        "Programming Language :: Python :: 3.12",
        "Programming Language :: Python :: 3.13",
        "Programming Language :: Python :: 3.14",
        "Programming Language :: C++",
        "Topic :: Security :: Cryptography",
    ],
    project_urls={
        "Repository": "https://github.com/Huelse/SEAL-Python",
        "Issues": "https://github.com/Huelse/SEAL-Python/issues",
    },
)