mailinabox/tools/editconf.py

#!/usr/bin/python3

import sys, re

# sanity check
if len(sys.argv) < 3:
	print("usage: python3 editconf.py /etc/file.conf [-s] NAME=VAL [NAME=VAL ...]")
	sys.exit(1)

# parse command line arguments
filename = sys.argv[1]
settings = sys.argv[2:]

delimiter = "="
delimiter_re = r"\s*=\s*"
if settings[0] == "-s":
	settings.pop(0)
	delimiter = " "
	delimiter_re = r"\s+"

# create the new config file in memory
found = set()
buf = ""
for line in open(filename):
	for i in range(len(settings)):
		name, val = settings[i].split("=", 1)
		m = re.match("\s*" + re.escape(name) + delimiter_re + "(.*?)\s*$", line)
		if m:
			# If this is already the setting, do nothing.
			if m.group(1) == val:
				buf += line
				found.add(i)
				break
			
			# comment-out the existing line
			buf += "#" + line
			
			# if this option oddly appears more than once, don't add the setting again
			if i in found:
				break
			
			# add the new setting
			buf += name + delimiter + val + "\n"
			
			# note that we've applied this option
			found.add(i)
			
			break
	else:
		# If did not match any setting names, pass this line through.
		buf += line
		
# Put any settings we didn't see at the end of the file.
for i in range(len(settings)):
	if i not in found:
		name, val = settings[i].split("=", 1)
		buf += name + delimiter + val + "\n"

# Write out the new file.
with open(filename, "w") as f:
	f.write(buf)