blob: 9f74b655709a016c424be541963ebee4bd93861a (
plain) (
blame)
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
|
import os
import subprocess
import sys
def main():
cmd = sys.argv[1:]
h_file = None
try:
index = cmd.index('-o')
h_file = cmd[index+1]
cmd[index+1] = os.path.dirname(h_file)
except (ValueError, IndexError):
pass
p = subprocess.run(cmd, capture_output=True, text=True)
if p.returncode:
if p.stdout:
sys.stderr.write('stdout:\n{}\n'.format(p.stdout))
if p.stderr:
sys.stderr.write('stderr:\n{}\n'.format(p.stderr))
sys.exit(p.returncode)
if h_file and h_file.endswith(('.fbs.h', '.fbs64.h')):
cpp_file = '{}.cpp'.format(h_file[:-2])
with open(cpp_file, 'w') as f:
f.write('#include "{}"\n'.format(os.path.basename(h_file)))
sys.exit(0)
if __name__ == '__main__':
main()
|
