threadr.lostcave.ddnss.de/variable_grabbler.py

43 lines
1.4 KiB
Python
Raw Normal View History

2020-02-23 21:40:16 +01:00
#!/usr/bin/python2
print """
2020-03-07 19:46:34 +01:00
Variable Grabbler - version 3.3
2020-02-23 21:40:16 +01:00
-------------------------------"""
# definitions
import os, json, sys
2020-02-23 21:40:16 +01:00
if not len(sys.argv)==3:
print >> sys.stderr, "Error: Exactly two arguments required: \"python variable_grabbler.py <file to be rewritten> <config file>\""
sys.exit(1)
2020-02-23 21:40:16 +01:00
2020-02-24 00:04:25 +01:00
print "Reading config file...",
config_file = open(sys.argv[2],"r")
config_values = json.loads(config_file.read().decode("utf-8"))
config_file.close()
2020-02-24 00:04:25 +01:00
print "Done."
2020-02-23 21:40:16 +01:00
print "Working on file: "+sys.argv[1]
for variable in config_values:
2020-03-03 15:07:38 +01:00
print "> "+variable+" => ",
file_in = open(sys.argv[1], "r")
file_content = file_in.read()
file_in.close()
if config_values[variable][0] == "file":
print "Found array. Adding file instead: " + config_values[variable][1]
file_replacement = open(config_values[variable][1], "r")
replacement = file_replacement.read()
file_replacement.close()
elif config_values[variable][0] == "exec":
print "Found array. Executing command instead: " + config_values[variable][1]
#Add code to replace variable with command output here
print >> sys.stderr, "Error: This option is not yet supported."
replacement = ""
else:
replacement = config_values[variable]
print repr(replacement)
file_out = open(sys.argv[1], "w")
2020-03-07 19:46:34 +01:00
file_out.write(file_content.decode('utf-8').replace("%" + variable.upper() + "%", replacement).encode('utf-8'))
file_out.close()
print "Done."