優化簡單的Python文件匹配和替換腳本

Question

只是想知道如果任何人有我如何才能優化我的簡單而緩慢的文件替換腳本一個建議：

def switchFiles(args): 
for root1, dirs1, files1 in os.walk(args.folder): 
    for f1 in files1: 
     for root2, dirs2, files2 in os.walk(args.database): 
      for f2 in files2: 
       if fnmatch.fnmatch(f1, f2): 
        command = 'cp '+os.path.join(root1, f1)+' '+os.path.join(root2, f2) 
        print(command) 
        os.system(command) 

謝謝！

Answer 1

這是一個清理代碼：

def switchFiles(args): 
    pairs = [] 
    for root1, dirs1, files1 in os.walk(args.folder): 
     for f1 in files1: 
      for root2, dirs2, files2 in os.walk(args.database): 
       for f2 in files2: 
        if f1 == f2: 
         pairs.append(os.path.join(root1, f1), os.path.join(root2, f2)) 
    for src, dst in pairs: 
     shutil.copyfile(src, dst) 

如果args.folder和args.database是分開的（不是子目錄），和所有的文件在其目錄名稱是獨一無二的，那麼你可以這樣做：

def switchFiles(args): 
    f, d = {}, {} 
    for root1, dirs1, files1 in os.walk(args.folder): 
     for f1 in files1: 
      f[f1] = os.path.join(root1, f1) 
    for root2, dirs2, files2 in os.walk(args.database): 
     for f2 in files2: 
      d[f2] = os.path.join(root2, f2) 
    ns = set(f.keys()) & set(d.keys()) 
    for n in ns: 
     shutil.copyfile(f[n], d[n]) 

Answer 2

我覺得如果args.folder只有很少的文件，這個會更快。

def switchFiles(args): 
    srclst = {} 
    for root, dirs, files in os.walk(args.folder): 
     rootp = (root,) 
     for filename in files: 
      srclst[filename] = rootp 
    for root, dirs, files in os.walk(args.database): 
     for filename in files: 
      srcrootp = srclst.get(filename) 
      if not srcrootp: 
       continue 
      srcpath = os.path.join(srcrootp[0], filename) 
      dstpath = os.path.join(root, filename) 
      print "replace", srcpath, dstpath 
      shutil.copy(srcpath, dstpath)