1#!/usr/bin/env python3
2
3import os
4import sys
5import argparse
6import sysconfig
7import distutils.sysconfig
8
9
10def relpath_nodots(path, base):
11    rel = os.path.normpath(os.path.relpath(path, base))
12    assert not os.path.isabs(rel)
13    parts = rel.split(os.path.sep)
14    if parts and parts[0] == '..':
15        raise ValueError(f"{path} is not under {base}")
16    return rel
17
18def main():
19    parser = argparse.ArgumentParser(description="extract cmake variables from python")
20    parser.add_argument("variable_name")
21    args = parser.parse_args()
22    if args.variable_name == "LLDB_PYTHON_RELATIVE_PATH":
23        print(distutils.sysconfig.get_python_lib(True, False, ''))
24    elif args.variable_name == "LLDB_PYTHON_EXE_RELATIVE_PATH":
25        tried = list()
26        exe = sys.executable
27        while True:
28            try:
29                print(relpath_nodots(exe, sys.prefix))
30                break
31            except ValueError:
32                tried.append(exe)
33                if os.path.islink(exe):
34                    exe = os.path.join(os.path.dirname(exe), os.readlink(exe))
35                    continue
36                else:
37                    print("Could not find a relative path to sys.executable under sys.prefix", file=sys.stderr)
38                    for e in tried:
39                        print("tried:", e, file=sys.stderr)
40                    print("sys.prefix:", sys.prefix, file=sys.stderr)
41                    sys.exit(1)
42    elif args.variable_name == "LLDB_PYTHON_EXT_SUFFIX":
43        print(sysconfig.get_config_var('EXT_SUFFIX'))
44    else:
45        parser.error(f"unknown variable {args.variable_name}")
46
47if __name__ == '__main__':
48    main()