blob: 9b14653067f67b062e978fb24b776ccdfd9acf7e [file] [edit]
#!/usr/bin/env python3
############################################################################
# Copyright (C) SchedMD LLC.
############################################################################
import json
import os
import sys
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()