2013-08-21 02:27:32 +00:00
|
|
|
#!/usr/bin/python3
|
|
|
|
|
|
|
|
import sys, re
|
|
|
|
|
|
|
|
# sanity check
|
|
|
|
if len(sys.argv) < 3:
|
2013-08-23 15:59:28 +00:00
|
|
|
print("usage: python3 editconf.py /etc/file.conf [-s] NAME=VAL [NAME=VAL ...]")
|
2013-08-21 02:27:32 +00:00
|
|
|
sys.exit(1)
|
|
|
|
|
|
|
|
# parse command line arguments
|
|
|
|
filename = sys.argv[1]
|
|
|
|
settings = sys.argv[2:]
|
|
|
|
|
2013-08-23 15:59:28 +00:00
|
|
|
delimiter = "="
|
|
|
|
delimiter_re = r"\s*=\s*"
|
|
|
|
if settings[0] == "-s":
|
|
|
|
settings.pop(0)
|
|
|
|
delimiter = " "
|
|
|
|
delimiter_re = r"\s+"
|
|
|
|
|
2013-08-21 02:27:32 +00:00
|
|
|
# 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)
|
2013-08-23 15:59:28 +00:00
|
|
|
m = re.match("\s*" + re.escape(name) + delimiter_re + "(.*?)\s*$", line)
|
2013-08-21 13:37:33 +00:00
|
|
|
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
|
2013-08-21 02:27:32 +00:00
|
|
|
buf += "#" + line
|
2013-08-21 13:37:33 +00:00
|
|
|
|
|
|
|
# if this option oddly appears more than once, don't add the settingg again
|
|
|
|
if i in found:
|
|
|
|
break
|
|
|
|
|
|
|
|
# add the new setting
|
2013-08-23 15:59:28 +00:00
|
|
|
buf += name + delimiter + val + "\n"
|
2013-08-21 13:37:33 +00:00
|
|
|
|
|
|
|
# note that we've applied this option
|
2013-08-21 02:27:32 +00:00
|
|
|
found.add(i)
|
2013-08-21 13:37:33 +00:00
|
|
|
|
2013-08-21 02:27:32 +00:00
|
|
|
break
|
|
|
|
else:
|
2013-08-21 13:37:33 +00:00
|
|
|
# If did not match any setting names, pass this line through.
|
2013-08-21 02:27:32 +00:00
|
|
|
buf += line
|
|
|
|
|
2013-08-21 13:37:33 +00:00
|
|
|
# Put any settings we didn't see at the end of the file.
|
2013-08-21 02:27:32 +00:00
|
|
|
for i in range(len(settings)):
|
|
|
|
if i not in found:
|
2013-08-23 15:59:28 +00:00
|
|
|
name, val = settings[i].split("=", 1)
|
|
|
|
buf += name + delimiter + val + "\n"
|
2013-08-21 02:27:32 +00:00
|
|
|
|
|
|
|
# Write out the new file.
|
|
|
|
with open(filename, "w") as f:
|
|
|
|
f.write(buf)
|
|
|
|
|