mirror of
https://gitee.com/fantix/kloop.git
synced 2024-04-25 09:48:45 +00:00
143 lines
4.8 KiB
Python
143 lines
4.8 KiB
Python
# Copyright (c) 2022 Fantix King https://fantix.pro
|
|
# kLoop is licensed under Mulan PSL v2.
|
|
# You can use this software according to the terms and conditions of the Mulan PSL v2.
|
|
# You may obtain a copy of Mulan PSL v2 at:
|
|
# http://license.coscl.org.cn/MulanPSL2
|
|
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
# See the Mulan PSL v2 for more details.
|
|
|
|
|
|
import os
|
|
import subprocess
|
|
import sysconfig
|
|
|
|
from setuptools import setup
|
|
|
|
from Cython.Build import cythonize
|
|
from Cython.Distutils import build_ext
|
|
from Cython.Distutils import Extension
|
|
from distutils import dir_util
|
|
from distutils import log
|
|
from distutils.command.clean import clean
|
|
|
|
|
|
class build_ext_with_resolver(build_ext):
|
|
def finalize_options(self):
|
|
self.set_undefined_options(
|
|
"build",
|
|
("debug", "debug"),
|
|
("force", "force"),
|
|
)
|
|
if self.debug is None:
|
|
self.debug = os.getenv("KLOOP_DEBUG", "0") == "1"
|
|
if self.force is None:
|
|
self.force = os.getenv("KLOOP_FORCE", "0") == "1"
|
|
|
|
for ext in self.distribution.ext_modules:
|
|
if ext.cython_directives:
|
|
ext.cython_directives["language_level"] = "3"
|
|
else:
|
|
ext.cython_directives = {"language_level": "3"}
|
|
if ext.cython_compile_time_env:
|
|
ext.cython_compile_time_env["DEBUG"] = self.debug
|
|
else:
|
|
ext.cython_compile_time_env = {"DEBUG": self.debug}
|
|
if self.debug:
|
|
if "-O0" not in ext.extra_compile_args:
|
|
ext.extra_compile_args.append("-O0")
|
|
if ext.name == "kloop.loop":
|
|
resolver = (
|
|
f"resolver/target/{'debug' if self.debug else 'release'}"
|
|
f"/libkloop_resolver.a"
|
|
)
|
|
if resolver not in ext.extra_link_args:
|
|
ext.extra_link_args.append(resolver)
|
|
if resolver not in ext.depends:
|
|
ext.depends.append(resolver)
|
|
|
|
super().finalize_options()
|
|
|
|
def run(self):
|
|
if self.force:
|
|
cmd = ["cargo", "clean", "-p", "resolver"]
|
|
if not self.debug:
|
|
cmd.append("-r")
|
|
self.announce(f"Running: {cmd}", log.INFO)
|
|
subprocess.check_call(cmd, cwd="resolver")
|
|
|
|
cmd = ["cargo", "build"]
|
|
if not self.debug:
|
|
cmd.append("-r")
|
|
self.announce(f"Running: {cmd}", log.INFO)
|
|
subprocess.check_call(cmd, cwd="resolver")
|
|
super().run()
|
|
|
|
|
|
class clean_with_resolver(clean):
|
|
def run(self):
|
|
super().run()
|
|
for d in self.distribution.package_dir.values():
|
|
self._clean_dir(d)
|
|
cmd = ["cargo", "clean"]
|
|
if not self.all:
|
|
cmd.extend(["-p", "resolver"])
|
|
self.announce(f"Running: {cmd}", log.INFO)
|
|
if not self.dry_run:
|
|
subprocess.check_call(cmd, cwd="resolver")
|
|
|
|
def _clean_dir(self, path):
|
|
for f in os.listdir(path):
|
|
name, ext = os.path.splitext(f)
|
|
real_f = os.path.join(path, f)
|
|
is_dir = os.path.isdir(real_f) and not os.path.islink(real_f)
|
|
if name == "__pycache__" or ext in {".egg-info", ".so", ".c"}:
|
|
if is_dir:
|
|
dir_util.remove_tree(real_f, dry_run=self.dry_run)
|
|
else:
|
|
self.announce(f"removing {real_f!r}", log.INFO)
|
|
if not self.dry_run:
|
|
os.remove(real_f)
|
|
elif is_dir:
|
|
self._clean_dir(real_f)
|
|
|
|
|
|
setup(
|
|
cmdclass={
|
|
"build_ext": build_ext_with_resolver,
|
|
"clean": clean_with_resolver,
|
|
},
|
|
ext_modules=[
|
|
Extension(
|
|
"kloop.loop",
|
|
["src/kloop/loop.pyx"],
|
|
),
|
|
Extension(
|
|
"kloop.tls",
|
|
["src/kloop/tls.pyx"],
|
|
libraries=[
|
|
lib.strip().removeprefix("-l")
|
|
for lib in sysconfig.get_config_var("OPENSSL_LIBS").split()
|
|
],
|
|
include_dirs=[
|
|
d.strip().removeprefix("-I")
|
|
for d in sysconfig.get_config_var("OPENSSL_INCLUDES").split()
|
|
],
|
|
library_dirs=[
|
|
d.strip().removeprefix("-L")
|
|
for d in sysconfig.get_config_var("OPENSSL_LDFLAGS").split()
|
|
if d.strip().startswith("-L")
|
|
],
|
|
extra_link_args=[
|
|
d.strip()
|
|
for d in sysconfig.get_config_var("OPENSSL_LDFLAGS").split()
|
|
if not d.strip().startswith("-L")
|
|
],
|
|
runtime_library_dirs=(lambda x: [x] if x else [])(
|
|
sysconfig.get_config_var("OPENSSL_RPATH")
|
|
),
|
|
),
|
|
],
|
|
)
|