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
|
from __future__ import annotations
import os
import subprocess
from setuptools import Extension, setup
from setuptools.command.build_ext import build_ext
class BuildExt(build_ext):
def run(self) -> None:
cmd = "gcc -fPIC -shared -o b/libb.so b/b.c"
subprocess.check_call(cmd.split())
cmd = (
"gcc -fPIC -shared -o a/liba.so "
"-Wl,{dtags_flag} -Wl,-rpath=$ORIGIN/../b "
"-Ib a/a.c -Lb -lb"
).format(
dtags_flag=(
"--enable-new-dtags" if os.getenv("DTAG") == "runpath" else "--disable-new-dtags"
),
)
subprocess.check_call(cmd.split())
super().run()
setup(
name="testrpath",
version="0.0.1",
packages=["testrpath"],
package_dir={"": "src"},
cmdclass={"build_ext": BuildExt},
ext_modules=[
Extension(
"testrpath/testrpath",
sources=["src/testrpath/testrpath.c"],
include_dirs=["a"],
libraries=["a"],
library_dirs=["a"],
),
],
)
|
