1. 程式人生 > >Python——目錄對比原始碼

Python——目錄對比原始碼

#/usr/bin/env python
#coding=utf8
#用來對比目錄中檔案的區別
#2015.11.12
#V0.0.0.1


import os
import sys
import filecmp
import re
import shutil


holderlist=[]


#遞迴獲取更新項函式
def compareme(dir1,dir2):
        dircomp=filecmp.dircmp(dir1,dir2)


#獲取左目錄中的檔案或者目錄
        only_in_one=dircomp.left_only


#不匹配的檔案
        diff_in_one=dircomp.diff_files


#定義源目錄絕對路徑
        dirpath=os.path.abspath(dir1)




#將更新檔名或者目錄追加到holderlist
        [holderlist.append(os.path.abspath(os.path.join(dir1,x))) for x in only_in_one]
        [holderlist.append(os.path.abspath(os.path.join(dir1,x))) for x in diff_in_one]


#判斷是否存在相同子目錄,以便遞迴
        if len(dircomp.common_dirs) > 0:
#遞迴子目錄
                for item in dircomp.common_dirs:
                        compareme(os.path.abspath(os.path.join(dir1,item)),os.path.abspath(os.path.join(dir2,item)))
        return holderlist




def main():
        if len(sys.argv) > 2:


#獲取傳入引數
                dir1=sys.argv[1]
                dir2=sys.argv[2]
        else:
                print "Usage: ",sys.argv[0],"datadir backupdir"
                sys.exit()


        source_files=compareme(dir1,dir2)


#獲取dir1的絕對路徑
        dir1=os.path.abspath(dir1)


#如果不是以/結尾則在後面加上個/
        if not dir2.endswith('/'): 
                dir2=dir2+'/'


#獲取dir2的絕對路徑
        dir2=os.path.abspath(dir2)


        destination_files=[]


#遍歷輸出返回的差異檔案或者目錄的清單
        for item in source_files:
                createdir_bool=False
#將原目錄差異清單對應替換成備份目錄
                destination_dir=re.sub(dir1,dir2,item)
                destination_files.append(destination_dir)


#再次呼叫compareme函式標記
                if os.path.isdir(item):
                        if not os.path.exists(destination_dir):
                                os.makedirs(destination_dir)
                                createdir_bool=True


#通過createdir_bool引數來判斷是否還有子目錄
                if createdir_bool:
                        destination_files=[]
                        source_files=[]
                        source_files=compareme(dir1,dir2)


                for item in source_files:
                        destination_dir=re.sub(dir1,dir2,item)
                        destination_files.append(destination_dir)


        print "update item:"
        print source_files


#將源目錄與備份目錄檔案清單拆分成元組
        copy_pair=zip(source_files,destination_files)


        for item in copy_pair:


#判斷是否為檔案,是則進行復制操作
                if os.path.isfile(item[0]):
                        shutil.copyfile(item[0],item[1])




if __name__ == '__main__':
        main()