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
|
import glob
import os
import shutil
import tempfile
from setuptools import setup, Extension
from setuptools._distutils import ccompiler
from setuptools._distutils import sysconfig
setupArgs = {}
# Check if the yajl library + headers are present
# We don't use compiler.has_function because it leaves a lot of files behind
# without properly cleaning up
def yajl_present():
compiler = ccompiler.new_compiler(verbose=1)
sysconfig.customize_compiler(compiler) # CC, CFLAGS, LDFLAGS, etc
yajl_version_test_file = tempfile.NamedTemporaryFile(suffix=".c", prefix="yajl_version", delete=False)
try:
yajl_version_test_file.write(b'''
#include <yajl/yajl_version.h>
int main(int args, char **argv)
{
#if YAJL_MAJOR != 2
fail to compile
#else
yajl_version();
#endif
return 0;
}
''')
yajl_version_test_file.close()
try:
objs = compiler.compile([yajl_version_test_file.name])
compiler.link_shared_lib(objs, 'a', libraries=["yajl"])
return True
finally:
os.remove(compiler.library_filename('a', lib_type='shared'))
for obj in objs:
os.remove(obj)
except:
return False
finally:
os.remove(yajl_version_test_file.name)
ORIGINAL_SOURCES = os.path.join("cextern", "yajl")
PATCHED_SOURCES = "yajl-patched-sources"
def patch_yajl_sources() -> None:
"""Make yajl sources ready for direct compilation against them"""
# cp cextern/yajl -R $yajl_sources_copy
# mkdir $yajl_sources_copy/yajl
# cp $yajl_sources_copy/src/api/*.h $yajl_sources_copy/yajl
if os.path.isdir(PATCHED_SOURCES):
if os.path.isdir(ORIGINAL_SOURCES):
print("Folder for yajl sources already exists, removing it")
shutil.rmtree(PATCHED_SOURCES)
else:
# Wheels are build in an isolated environment
# where the original sources aren't available
print("Yajl Sources are already in the correct place. Skip copy")
return
print(f"Copy yajl sources to '{PATCHED_SOURCES}'")
shutil.copytree(
ORIGINAL_SOURCES, PATCHED_SOURCES,
ignore=shutil.ignore_patterns("example", "test"),
)
headers_original = os.path.join(PATCHED_SOURCES, "src", "api")
headers_copy = os.path.join(PATCHED_SOURCES, "yajl")
shutil.copytree(headers_original, headers_copy)
extra_sources = []
extra_include_dirs = []
libs = ['yajl']
embed_yajl = os.environ.get('IJSON_EMBED_YAJL', None) == '1'
if not embed_yajl:
have_yajl = yajl_present()
else:
patch_yajl_sources()
extra_sources = sorted(glob.glob(os.path.join(PATCHED_SOURCES, 'src', '*.c')))
extra_sources.remove(os.path.join(PATCHED_SOURCES, 'src', 'yajl_version.c'))
extra_include_dirs = [PATCHED_SOURCES, os.path.join(PATCHED_SOURCES, 'src')]
libs = []
build_yajl_default = '1' if embed_yajl or have_yajl else '0'
build_yajl = os.environ.get('IJSON_BUILD_YAJL2C', build_yajl_default) == '1'
if build_yajl:
yajl_ext = Extension('ijson.backends._yajl2',
language='c',
sources=sorted(glob.glob('src/ijson/backends/ext/_yajl2/*.c')) + extra_sources,
include_dirs=['src/ijson/backends/ext/_yajl2'] + extra_include_dirs,
libraries=libs,
depends=glob.glob('src/ijson/backends/ext/_yajl2/*.h'))
setupArgs['ext_modules'] = [yajl_ext]
setup(**setupArgs)
|
