xref: /aosp_15_r20/development/tools/repo_diff/service/repodiff/tools/env_variable_load.py (revision 90c8c64db3049935a07c6143d7fd006e26f8ecca)
1import json
2import os
3import sys
4
5
6if __name__ == "__main__":
7  try:
8    serialized_env_vars = sys.argv[1]
9  except IndexError:
10    print "Usage: python env_variable_load.py <serialized json>"
11    print "Obtain JSON dict from https://valentine.corp.google.com/#/show/1521659275969805"
12    sys.exit(1)
13
14  with open("%s/.bashrc" % os.environ["HOME"], "rb") as f:
15    existing_contents = f.read()
16
17  with open("%s/.bashrc" % os.environ["HOME"], "a") as f:
18    f.write("\n")
19
20    for key, value in sorted(json.loads(serialized_env_vars).items()):
21      if key in existing_contents:
22        continue
23      f.write("export %s=\"%s\"\n" % (key, value))
24