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
|
from __future__ import annotations
import os
import platform
from typing import cast
from typing import TYPE_CHECKING
from setuptools import setup
if TYPE_CHECKING:
from setuptools import Extension
try:
from Cython.Build import build_ext as _cy_build_ext
from Cython.Distutils.extension import Extension as _cy_Extension
HAS_CYTHON = True
except ImportError:
_cy_build_ext = _cy_Extension = None
HAS_CYTHON = False
IS_CPYTHON = platform.python_implementation() == "CPython"
DISABLE_EXTENSION = bool(os.environ.get("DISABLE_SQLALCHEMY_CEXT"))
REQUIRE_EXTENSION = bool(os.environ.get("REQUIRE_SQLALCHEMY_CEXT"))
if DISABLE_EXTENSION and REQUIRE_EXTENSION:
raise RuntimeError(
"Cannot set both 'DISABLE_SQLALCHEMY_CEXT' and "
"'REQUIRE_SQLALCHEMY_CEXT' environment variables"
)
if HAS_CYTHON and IS_CPYTHON and not DISABLE_EXTENSION:
assert _cy_Extension is not None
assert _cy_build_ext is not None
# when adding a cython module, also update the imports in _has_cy
cython_files = [
"collections.pyx",
"immutabledict.pyx",
"processors.pyx",
"resultproxy.pyx",
"util.pyx",
]
cython_directives = {"language_level": "3"}
module_prefix = "sqlalchemy.cyextension."
source_prefix = "lib/sqlalchemy/cyextension/"
ext_modules = cast(
"list[Extension]",
[
_cy_Extension(
f"{module_prefix}{os.path.splitext(file)[0]}",
sources=[f"{source_prefix}{file}"],
cython_directives=cython_directives,
optional=not REQUIRE_EXTENSION,
)
for file in cython_files
],
)
cmdclass = {"build_ext": _cy_build_ext}
elif REQUIRE_EXTENSION:
reasons = []
if not HAS_CYTHON:
reasons.append("Cython is missing")
if not IS_CPYTHON:
reasons.append("Not CPython, build is supported only on it")
raise RuntimeError(
"Cython extension build is required because REQUIRE_SQLALCHEMY_CEXT "
f"is set but it was deselected because: {'; '.join(reasons)}; "
"will not degrade to pure python install"
)
else:
ext_modules = []
cmdclass = {}
setup(cmdclass=cmdclass, ext_modules=ext_modules)
|
