Welcome, guest | Sign In | My Account | Store | Cart

RoadRunner: recursively updates the contents of a directory tree. (Python recipe) by Juan Carlos Leon
ActiveState Code (http://code.activestate.com/recipes/52553/)

This recipe allows you to update desired files in a directory tree. It needs to be run from the shell. Once you run it, it will ask you for: a) the "string" you want to insert in the file b) the extension of the files to update e.g: .html c) a regular expresion ( where you want the string to be inserted ) and d) the path of the directory where you want this update to occur.

Python, 89 lines
Download
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#!/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()
Tags: files