threadr.lostcave.ddnss.de/variable_grabbler.py

98 lines
3.6 KiB
Python
Raw Normal View History

2020-02-23 21:40:16 +01:00
#!/usr/bin/python2
print """
2020-04-08 05:22:50 +02:00
Variable Grabbler - version 4.0_pre6
------------------------------------"""
2020-04-08 02:06:32 +02:00
################################################################
# Changes in this version:
#===============================================================
2020-04-08 03:02:54 +02:00
# *"\?" in a variable will now be replaced with a simple ?
# before processing, remember to double escape that because
2020-04-08 05:22:50 +02:00
# json doesn't like \? (=> \\?)
2020-04-08 03:02:54 +02:00
# *commands are now not run over and over again if not needed
2020-04-08 05:22:50 +02:00
# *files are now not rewritten for each variable individually
# *some technical stuff
2020-04-08 02:06:32 +02:00
################################################################
# Exit codes:
#===============================================================
# 0 => normal exit
2020-04-08 05:22:50 +02:00
# 1 => command line argument error
2020-04-08 02:06:32 +02:00
# 2 => error while processing
#
2020-02-23 21:40:16 +01:00
# definitions
2020-04-08 02:06:32 +02:00
import os, json, sys, subprocess
def print_err(text):
print >> sys.stderr, text
2020-02-23 21:40:16 +01:00
if not len(sys.argv)==3:
2020-04-08 02:06:32 +02:00
print_err("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
2020-04-08 03:52:18 +02:00
print "Reading file: "+sys.argv[1]+"..."
2020-04-08 03:56:45 +02:00
file_in = open(sys.argv[1], "r")
file_content = file_in.read()
file_in.close()
2020-04-08 03:52:18 +02:00
print "Replacing variables:"
for variable in config_values:
print "> "+variable.upper()+" => ",
if config_values[variable][0] == "file":
print "Found array. Adding file instead: " + config_values[variable][1]
file_replacement = open(config_values[variable][1], "r")
2020-04-08 05:22:50 +02:00
file_content = file_content.decode('utf-8').replace("%" + variable.upper() + "%", file_replacement.read()).encode('utf-8')
file_replacement.close()
elif config_values[variable][0] == "exec":
print "Found array. Executing command instead: " + config_values[variable][1]
if variable.upper() in file_content:
process = subprocess.Popen(config_values[variable][1], shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
stdout, stderr = process.communicate()
if process.returncode == 0:
print "Process exited normally. Replacing variable with the output from STDOUT..."
print_err("Subprocess exited normally.\nThe following error messages were produced:")
print_err(stderr)
2020-04-08 05:22:50 +02:00
file_content = file_content.decode('utf-8').replace("%" + variable.upper() + "%", stdout).encode('utf-8')
else:
print "Process exited abnormally. Exiting. No changes will be made."
print_err("Subprocess exited abnormally. Exiting.\nThe following output was produced:")
print_err(stdout)
print_err("The following error messages were produced:")
print_err(stderr)
print "Exiting now."
sys.exit(2)
2020-04-08 02:06:32 +02:00
else:
print variable.upper(),
print "was not found in the file. Skipping command execution."
else:
2020-04-08 03:28:59 +02:00
replacement = str(config_values[variable])
# look for variable sections that contain question marks
argument_positions = []
i = 0
while i<len(replacement):
if replacement[i] == '?' and (i==0 or not replacement[i-1]=='\\'):
argument_positions = argument_positions + [ i ]
2020-04-08 03:13:41 +02:00
i = i+1
2020-04-08 05:22:50 +02:00
#print "DEBUG: " + str(argument_positions)
2020-04-08 03:41:12 +02:00
replacement = replacement.replace("\\?", "?")
2020-04-08 03:37:17 +02:00
print repr(replacement)
2020-04-08 03:52:18 +02:00
# actually replace variables
2020-04-08 05:22:50 +02:00
if argument_positions == []:
file_content = file_content.decode('utf-8').replace("%" + variable.upper() + "%", replacement).encode('utf-8')
2020-04-08 03:52:18 +02:00
print "Saving new file..."
file_out = open(sys.argv[1], "w")
file_out.write(file_content)
file_out.close()
print "Done."