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
|
from re import sub
from SCons.Script import *
import pathlib
import subprocess
def path_translate(env, source):
protopath = [pathlib.Path(p) for p in env['PROTOPATH']]
for p in protopath:
f = p / f'{source}.proto'
if not f.exists():
continue
return str(p / source)
return source
def Protonium(env, target_dir, sources):
return [env.ProtoniumBuild(f'{target_dir}/{source}', path_translate(env, source), PROTOC_OUT_DIR = target_dir) for source in sources]
#return [env.ProtoniumBuild(f'{target_dir}/{source}', path_translate(env, source)) for source in sources]
def protonium_builder_actions(source, target, env, for_signature):
#target_dir = target[0].Dir('.')
target_dir = env['PROTOC_OUT_DIR']
protoc_path = ' '.join(f'-I{path}' for path in env['PROTOPATH'])
return f'protoc {protoc_path} $SOURCES --protonium_out={target_dir} --python_out={target_dir} --protonium_python_out={target_dir}'
protonium_builder = Builder(
generator = protonium_builder_actions,
#action = 'protoc $SOURCES --protonium_out=$TARGET',
suffix = '_pb.h',
src_suffix = '.proto',
)
def exists():
return subprocess.call('protonium', shell = True, stdout = subprocess.DEVNULL, stderr = subprocess.DEVNULL) == 0
def generate(env):
if not exists():
return
protopath = [arg[2:] for arg in subprocess.check_output('pkg-config protobuf --cflags-only-I', shell = True, encoding = 'utf-8').split() if arg.startswith('-I')]
protopath += [arg[2:] for arg in subprocess.check_output('protonium --cflags', shell = True, encoding = 'utf-8').split() if arg.startswith('-I')]
env.Append(
BUILDERS = {'ProtoniumBuild': protonium_builder},
PROTOPATH = ['.', *protopath],
)
env.ParseConfig('protonium --cflags')
env.AddMethod(Protonium)
|
