#!bin/python 


#Imports# 
import os 
import os.path as op 
import time 
from sys import argv 



class Replacer(object): 

    def __init__(self, parentDir, dic={}, exts=[]): 
        self.__parentDir=parentDir 
        self.__dic=dic 
        self.__exts=exts 
        self.__replacingTimes=0 
        

    def __getExt(self, s): 
        #idx=s.rfind(".") 
        #ext=s[idx:] 
        #return ext 
    	return op.splitext(s)[1] 
    	 
    def replace(self): 
        start=self.__parentDir 
        for e in os.listdir(start): #foreach entry in os.listdir.. 
            path=start+op.sep+e 
            if os.path.isfile(path): 
                ext=self.__getExt(path) 
                if not ext in self.__exts: 
                    continue #Re-Loop.. 
                #replace.. 
                f=file(path, "r") 
                #Note: Not reading line by line as I've never met a more than 1MB text file! 
                src=f.read() 
                f.close() 
                for key in self.__dic:   
                    self.__replacingTimes += src.count(key) 
                    src=src.replace(key, self.__dic[key]) 
                    
                f=file(path, "w") 
                f.write(src) 
                f.close() 
            elif os.path.isdir(path): 
                #RECURSE.. 
                rep=Replacer(path, self.__dic, self.__exts) 
                rep.replace() 
            else: 
                continue 

    def __init__(self, parentDir, dic={}, exts=[]): 
        self.__parentDir=parentDir 
        self.__dic=dic 
        self.__exts=exts 
        self.__replacingTimes=0 

    def __getExt(self, s): 
        #idx=s.rfind(".") 
        #ext=s[idx:] 
        #return ext 
    	return op.splitext(s)[1]

    def replace(self): 
        start=self.__parentDir 
        for e in os.listdir(start): #foreach entry in os.listdir.. 
            path=start+op.sep+e 
            if os.path.isfile(path): 
                ext=self.__getExt(path) 
                if not ext in self.__exts: 
                    continue #Re-Loop.. 
                #replace.. 
                f=file(path, "r") 
                src=f.read() 
                f.close() 
                for key in self.__dic:   
                    self.__replacingTimes += src.count(key) 
                    src=src.replace(key, self.__dic[key]) 
                    
                f=file(path, "w") 
                f.write(src) 
                f.close() 
            elif os.path.isdir(path): 
                #RECURSE.. 
                rep=Replacer(path, self.__dic, self.__exts) 
                rep.replace() 
            else: 
                continue 

        for e in os.listdir(start): #foreach entry in os.listdir.. 
            path=start+op.sep+e 

          if os.path.isfile(path): 
                ext=self.__getExt(path) 

                if not ext in self.__exts: 
                    continue #Loop.. 

                #replace.. 
                f=file(path, "r") 
          
                src=f.read() 
                f.close() 
                for key in self.__dic:   
                    self.__replacingTimes += src.count(key) 
                    src=src.replace(key, self.__dic[key]) 

            elif os.path.isdir(path): 
                #RECURSE.. 
                rep=Replacer(path, self.__dic, self.__exts) 
                rep.replace() 
            else: 
                continue

def consoleMain(): 
	##python replacer.py root [old] [new] [exts] 

	root, oldones, newones, exts=argv[1:5] 
	oldones=oldones.split(",") #use comma in between. 
	newones=newones.split(",") #use comma in between. 
	exts=exts.split(",") 	   #use comma in between. 
	if len(oldones)==len(newones): 
		dic=dict(zip(oldones, newones)) 
		#print "DIC:", dic 
		#print "EXTS:", exts 
		start=time.time() 
		rep=Replacer(root, dic, exts) 
		rep.replace() 
		end=time.time() 
		print "Time: ", (end-start) 
	else: 	
		print "len(keys)!=len(values)" 
		exit(1) 
	 
	start=time.time() 
	rep=Replacer(root, {old:new}, [exts]) 
	rep.replace() 
	end=time.time() 
	print "Time: ", (end-start)