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
|
"""Build optional cython modules."""
import logging
import os
from distutils.command.build_ext import build_ext
from os.path import join
from typing import Any
try:
from setuptools import Extension
except ImportError:
from distutils.core import Extension
def getenv_bool(key: str, default: bool = False) -> bool:
value = os.environ.get(key, str(default)).lower()
if value in ("1", "true", "yes"):
return True
if value in ("0", "false", "no"):
return False
raise ValueError(f"Invalid value for boolean envvar {key}: {value}")
ulid_module = Extension(
"ulid_transform._ulid_impl",
[
join("src", "ulid_transform", "_ulid_impl.pyx"),
join("src", "ulid_transform", "ulid_wrapper.cpp"),
],
language="c++",
extra_compile_args=["-std=c++11", "-O3", "-g0"],
extra_link_args=["-std=c++11"],
)
class BuildExt(build_ext):
def build_extensions(self) -> None:
try:
super().build_extensions()
except Exception: # nosec
logging.exception("Failed to build extensions")
if getenv_bool("REQUIRE_CYTHON"):
raise
def build(setup_kwargs: Any) -> None:
if getenv_bool("SKIP_CYTHON"):
return
try:
from Cython.Build import cythonize
setup_kwargs.update(
{
"ext_modules": cythonize(
[
ulid_module,
],
compiler_directives={"language_level": "3"}, # Python 3
verbose=True,
),
"cmdclass": {"build_ext": BuildExt},
}
)
setup_kwargs["exclude_package_data"] = {
pkg: ["_ulid_impl.cpp"] for pkg in setup_kwargs["packages"]
}
except Exception:
logging.exception("Failed to configure cython")
if getenv_bool("REQUIRE_CYTHON"):
raise
|
