#!/usr/bin/env python
# $Id. RoadRunner.py Thu Apr 5 10:49:48 EST 2001 JuanCarlos.Leon $
import os
import string
import re
import sys
_files_lst=[]
def pluggit(readlines,regx,addstrng):
""" Append a string to desire regex pattern """
lst=readlines
cregex=re.compile(regx)
for eachLine in lst:
match=cregex.search(eachLine)
if match:
rplacement=match.group()+addstrng
f=re.sub(regx,rplacement,eachLine)
lindex=lst.index(eachLine)
lst.remove(eachLine)
lst.insert(lindex,f)
return lst
def compose(strng,regx,listoffiles=[]):
log=open('log.txt','w')
for xfile in listoffiles:
# Open file for read
readlines=open(xfile,'r').readlines()
buffer=pluggit(readlines,regx,strng)
# We are good too open the file for writting
write_file=open(xfile,'w')
# Stuffed all buffer lines back to the original file
for line in buffer:
write_file.write(line)
write_file.close()
log.write('file %s updated\n' % (xfile))
log.close()
def fetch(path=None,ext='.html'):
if not path:
path=os.getcwd()
if os.path.isdir(path):
lst = os.listdir(path)
if lst:
for each in lst:
spath=path+'/'+each
if os.path.isfile(spath):
fileobj=string.find(spath,ext)
if fileobj != -1: _files_lst.append(spath)
fetch(spath)
def RoadRunner():
while 1:
strngToInsert = raw_input ( '\nEnter string to incorporate : ' )
if strngToInsert: break
file_extension = raw_input ( 'Enter extension of the files to be updated [.html] : ')
if not file_extension: file_extension = '.html'
while 1:
regexp = raw_input ( 'Enter regex string [e.g: <[Bb][oO][dD][yY].*>] : ' )
if regexp: break
while 1:
update_path = raw_input ( 'Enter path to update [%s] : ' % (os.getcwd()) )
if not update_path: update_path = os.getcwd()
if os.path.isdir(update_path): break
print "\nYou have entered the following : \n"
print "String to incorporate: %s" % (strngToInsert)
print "Regex string: %s" % (regexp)
print "Path to update: %s\n" % (update_path)
ok = raw_input("Is this ok?[Y/n] : " )
if not ok: ok='Y'
if string.lower(ok)!='y':
sys.exit(1)
else:
fetch(update_path)
compose(strngToInsert,regexp,_files_lst)
if __name__ == '__main__' :
RoadRunner()
