blob: e2a6e4d60148e653ae7f5765f63829b79dfa4aeb [file] [log] [blame] [edit]
#!/usr/bin/env python3
############################################################################
# Copyright (C) SchedMD LLC.
############################################################################
import os
import sys
import json
def main():
"""
Print the os.environ in a JSON parsable format
"""
args = sys.argv[1:]
env = dict(os.environ)
if args:
filtered = {k: env[k] for k in args if k in env}
# If any requested vars are missing, match printenv behavior (non-zero exit)
missing = [k for k in args if k not in env]
if missing:
print(
f"Missing environment variables: {', '.join(missing)}",
file=sys.stderr,
)
sys.exit(1)
env = filtered
print(json.dumps(env))
if __name__ == "__main__":
main()