63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
# A rather simple script to replace a block of text, delimited by
|
||
|
|
# markers, with new contents from stdin. Importantly the markers are
|
||
|
|
# left in the file so future runs can update the file without
|
||
|
|
# requiring a separate template. The markers are currently for
|
||
|
|
# reStructuredText only, but more can be added.
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import textwrap
|
||
|
|
|
||
|
|
|
||
|
|
def replace(filename, blockname, content):
|
||
|
|
start = f".. block_start {blockname}"
|
||
|
|
stop = f".. block_end {blockname}"
|
||
|
|
|
||
|
|
tempfile = f"{filename}.tmp"
|
||
|
|
|
||
|
|
with open(filename, 'r') as i, open(tempfile, 'w') as o:
|
||
|
|
lines = i.readlines()
|
||
|
|
# Read lines up to the marker
|
||
|
|
while lines != []:
|
||
|
|
l = lines.pop(0)
|
||
|
|
o.write(l)
|
||
|
|
if l.strip() == start:
|
||
|
|
break
|
||
|
|
|
||
|
|
o.write(content)
|
||
|
|
|
||
|
|
# Skip lines until we get the end marker
|
||
|
|
while lines != []:
|
||
|
|
l = lines.pop(0)
|
||
|
|
if l.strip() == stop:
|
||
|
|
o.write(l)
|
||
|
|
break
|
||
|
|
|
||
|
|
# Now flush the rest of the file
|
||
|
|
for l in lines:
|
||
|
|
o.write(l)
|
||
|
|
|
||
|
|
# Move the temp file over the old one for an atomic replacement
|
||
|
|
os.rename(tempfile, filename)
|
||
|
|
|
||
|
|
|
||
|
|
def main(args):
|
||
|
|
parser = argparse.ArgumentParser(
|
||
|
|
prog='blockreplace'
|
||
|
|
)
|
||
|
|
parser.add_argument('filename')
|
||
|
|
parser.add_argument('blockname')
|
||
|
|
parser.add_argument('--indent', dest="indent", default="")
|
||
|
|
args = parser.parse_args()
|
||
|
|
content = sys.stdin.read()
|
||
|
|
content = textwrap.indent(content, args.indent)
|
||
|
|
|
||
|
|
replace(args.filename, args.blockname, content)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main(sys.argv)
|