aboutsummaryrefslogtreecommitdiffstats
path: root/build/plugins/_unpickler.py
blob: e01e7b311811e5b30aa016a50c579a44a0e5630c (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
31
32
33
34
35
36
37
38
39
40
41
42
import sys

sys.dont_write_bytecode = True

import argparse
import base64
try:
    import cPickle as pickle
except Exception:
    import pickle

import _common as common


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--data', help='pickled object of TCustomCommand class', required=True)
    parser.add_argument('--src-root', help='$S real path', required=True)
    parser.add_argument('--build-root', help='$B real path', required=True)
    parser.add_argument('--tools', help='binaries needed by command', required=True, nargs='+')
    args, unknown_args = parser.parse_known_args()

    encoded_cmd = args.data
    src_root = args.src_root
    build_root = args.build_root
    tools = args.tools

    assert (int(tools[0]) == len(tools[1:])), "tools quantity != tools number!"

    cmd_object = pickle.loads(base64.b64decode(encoded_cmd))

    cmd_object.set_source_root(src_root)
    cmd_object.set_build_root(build_root)

    if len(tools[1:]) == 0:
        cmd_object.run(unknown_args, common.get_interpreter_path())
    else:
        cmd_object.run(unknown_args, *tools[1:])


if __name__ == '__main__':
    main()