OLD | NEW |
| (Empty) |
1 #!/usr/bin/python | |
2 # Copyright 2013 The Chromium Authors. All rights reserved. | |
3 # Use of this source code is governed by a BSD-style license that can be | |
4 # found in the LICENSE file. | |
5 | |
6 # This script can either source a file and dump the enironment changes done by | |
7 # it, or just simply dump the current environment as JSON into a file. | |
8 | |
9 import json | |
10 import optparse | |
11 import os | |
12 import pipes | |
13 import subprocess | |
14 import sys | |
15 | |
16 | |
17 def main(): | |
18 parser = optparse.OptionParser() | |
19 parser.add_option('-f', '--output-json', | |
20 help='File to dump the environment as JSON into.') | |
21 parser.add_option( | |
22 '-d', '--dump-mode', action='store_true', | |
23 help='Dump the environment to sys.stdout and exit immediately.') | |
24 | |
25 parser.disable_interspersed_args() | |
26 options, args = parser.parse_args() | |
27 if options.dump_mode: | |
28 if args or options.output_json: | |
29 parser.error('Cannot specify args or --output-json with --dump-mode.') | |
30 json.dump(dict(os.environ), sys.stdout) | |
31 else: | |
32 if not options.output_json: | |
33 parser.error('Requires --output-json option.') | |
34 | |
35 envsetup_cmd = ' '.join(map(pipes.quote, args)) | |
36 full_cmd = [ | |
37 'bash', '-c', | |
38 '. %s > /dev/null; %s -d' % (envsetup_cmd, os.path.abspath(__file__)) | |
39 ] | |
40 try: | |
41 output = subprocess.check_output(full_cmd) | |
42 except Exception as e: | |
43 sys.exit('Error running %s and dumping environment.' % envsetup_cmd) | |
44 | |
45 env_diff = {} | |
46 new_env = json.loads(output) | |
47 for k, val in new_env.items(): | |
48 if k == '_' or (k in os.environ and os.environ[k] == val): | |
49 continue | |
50 env_diff[k] = val | |
51 with open(options.output_json, 'w') as f: | |
52 json.dump(env_diff, f) | |
53 | |
54 | |
55 if __name__ == '__main__': | |
56 sys.exit(main()) | |
OLD | NEW |