importsys, getoptimportosimportos.pathimporthashlibimportshutil
searchFolders= ["\\src\\", "\\res\\"]defusage():print "this is useage"
defgetmd5(filename, parentName, folderName):
m=hashlib.md5()
mfile= open(filename, 'rb')
m.update(mfile.read())
mfile.close()
md5value=m.hexdigest()#get last part of file name
subStr = filename[len(parentName) -len(folderName):len(filename)]return md5value+"\t"+subStrdefgetCurDir():
path=sys.path[0]ifos.path.isdir(path):returnpathelifos.path.isfile(path):returnos.path.dirname(path)defsaveFile(str, ver):
f= open(getCurDir() + "/" + ver + ".md5", 'w')
f.write(str)
f.close()defgetDescOfUrl(url):
saveStr= ""descList=[]
index=0for parent,dirnames,filenames inos.walk(url):for filename infilenames:for i, val inenumerate(searchFolders):
targParent= url +valif targParent inparent:
fullPath=os.path.join(parent, filename)
md5=getmd5(fullPath, targParent, val)#print "file is: " + md5
saveStr = saveStr + md5 + "\n"descList.append(md5)returnsaveStr, descListdefonIArg(url, ver):
saveStr, descList=getDescOfUrl(url)
saveFile(saveStr, ver)defgetDicByFile(url):
file=open(url)
line=f.readline()
retDic={}whileline:
md5, filename= line.split("\t")
retDic[filename]=md5returnretDicdefgetDicByUrl(url):
saveStr, splitArr=getDescOfUrl(url)
retDic={}for value insplitArr:
md5, filename= value.split("\t")
retDic[filename]=md5returnretDicdefgetDicOfUrl(url):
fileList=os.listdir(url)
version= ""
for value infileList:if os.path.splitext(value)[1] == '.md5':
version=os.path.splitext(value)[0]break
if version != "":return getDicByFile(url + "/" + version + ".md5")else:returngetDicByUrl(url)
opts,args= getopt.getopt(sys.argv[1:], "hi:l:r:v:o:")
inputFile= ""leftFile= ""rightFile= ""outFile= ""version= ""showHelp=Falsefor op, value inopts:if op == "-i":
inputFile=valueelif op == "-v":
version=valueelif op == "-l":
leftFile=valueelif op == "-r":
rightFile=valueelif op == "-o":
outFile=valueelif op == "-h":
showHelp=TrueifshowHelp:
usage()
sys.exit()elif inputFile != "":if version == "":print "you must input the version value using -v"sys.exit()
onIArg(inputFile, version)elif leftFile != "":if rightFile == "":print "you must input the right file url using -r"sys.exit()elif outFile == "":print "you must input the out direct using -o"sys.exit()else:print "on diff .."diffDic={}
leftDic=getDicOfUrl(leftFile)
rightDic=getDicOfUrl(rightFile)for (filename, md5) inrightDic.items():if notleftDic.has_key(filename):
diffDic[filename]=Trueelif leftDic.get(filename) !=md5:
diffDic[filename]=Truefor (filename, val) indiffDic.items():#if the folder exit
inUrl = rightFile +filename
outUrl= outFile +filename
folderIndex= outUrl.rfind("\\")
outFolder=outUrl[0:folderIndex]if notos.path.isdir(outFolder):
os.makedirs(outFolder)print("copy from" + rightFile +filename)print("to" +outUrl)
shutil.copy(inUrl, outUrl)