23 lines
746 B
Plaintext
23 lines
746 B
Plaintext
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import subprocess
|
|
|
|
def main():
|
|
# If "-E" is in the arguments, modify the following argument
|
|
args = sys.argv[1:]
|
|
if "-E" in args:
|
|
index = args.index("-E")
|
|
if index + 1 < len(args): # Check if there's another argument after "-E"
|
|
environment_value = args[index + 1]
|
|
# Replace \ and - with _
|
|
modified_environment_value = environment_value.replace("\\", "_").replace("-", "_").replace("/","_").replace(".","_")
|
|
args[index + 1] = modified_environment_value
|
|
|
|
# Construct the full puppet command with the modified args
|
|
command = ["/opt/puppetlabs/bin/puppet"] + args
|
|
subprocess.run(command)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|