Demo/scripts/update.py

#! /usr/local/bin/python

# Update a bunch of files according to a script.
# The input file contains lines of the form <filename>:<lineno>:<text>,
# meaning that the given line of the given file is to be replaced
# by the given text.  This is useful for performing global substitutions
# on grep output: 

import os
import sys
import regex

pat = '^\([^: \t\n]+\):\([1-9][0-9]*\):'
prog = regex.compile(pat)

class FileObj:
	def __init__(self, filename):
		self.filename = filename
		self.changed = 0
		try:
			self.lines = open(filename, 'r').readlines()
		except IOError, msg:
			print '*** Can\'t open "%s":' % filename, msg
			self.lines = None
			return
		print 'diffing', self.filename

	def finish(self):
		if not self.changed:
			print 'no changes to', self.filename
			return
		try:
			os.rename(self.filename, self.filename + '~')
			fp = open(self.filename, 'w')
		except (os.error, IOError), msg:
			print '*** Can\'t rewrite "%s":' % self.filename, msg
			return
		print 'writing', self.filename
		for line in self.lines:
			fp.write(line)
		fp.close()
		self.changed = 0

	def process(self, lineno, rest):
		if self.lines is None:
			print '(not processed): %s:%s:%s' % (
				  self.filename, lineno, rest),
			return
		i = eval(lineno) - 1
		if not 0 <= i < len(self.lines):
			print '*** Line number out of range: %s:%s:%s' % (
				  self.filename, lineno, rest),
			return
		if self.lines[i] == rest:
			print '(no change): %s:%s:%s' % (
				  self.filename, lineno, rest),
			return
		if not self.changed:
			self.changed = 1
		print '%sc%s' % (lineno, lineno)
		print '<', self.lines[i],
		print '---'
		self.lines[i] = rest
		print '>', self.lines[i],

def main():
	if sys.argv[1:]:
		try:
			fp = open(sys.argv[1], 'r')
		except IOError, msg:
			print 'Can\'t open "%s":' % sys.argv[1], msg
			sys.exit(1)
	else:
		fp = sys.stdin
	curfile = None
	while 1:
		line = fp.readline()
		if not line:
			if curfile: curfile.finish()
			break
		n = prog.match(line)
		if n < 0:
			print 'Funny line:', line,
			continue
		filename, lineno = prog.group(1, 2)
		if not curfile or filename <> curfile.filename:
			if curfile: curfile.finish()
			curfile = FileObj(filename)
		curfile.process(lineno, line[n:])

main()