快捷導(dǎo)航

python復(fù)制文件代碼實現(xiàn)

更新時間：2013年12月23日 11:39:27 作者：

最近在備份手機上的照片的時候，純手工操作覺得有些麻煩，就想寫個腳本自動進(jìn)行。因為備份的時候有些照片以前備份過了，所以需要有個判重操作,看下面的實現(xiàn)代碼

主要功能在copyFiles()函數(shù)里實現(xiàn)，如下：

def copyFiles(src, dst):
    srcFiles = os.listdir(src)
    dstFiles = dict(map(lambda x:[x, ''], os.listdir(dst)))
    filesCopiedNum = 0

    # 對源文件夾中的每個文件若不存在于目的文件夾則復(fù)制
    for file in srcFiles:
        src_path = os.path.join(src, file)
        dst_path = os.path.join(dst, file)
        # 若源路徑為文件夾，若存在于目標(biāo)文件夾，則遞歸調(diào)用本函數(shù)；否則先創(chuàng)建再遞歸。
        if os.path.isdir(src_path):
            if not os.path.isdir(dst_path):
                os.makedirs(dst_path)  
            filesCopiedNum += copyFiles(src_path, dst_path)
        # 若源路徑為文件，不重復(fù)則復(fù)制，否則無操作。
        elif os.path.isfile(src_path):                
            if not dstFiles.has_key(file):
                shutil.copyfile(src_path, dst_path)
                filesCopiedNum += 1

    return filesCopiedNum

這里我首先使用os.listdir()函數(shù)來遍歷源文件夾src和目標(biāo)文件夾dst，得到兩個文件列表，但由于我需要判重操作，因此需要在dst文件列表中進(jìn)行查詢操作。由于列表的查詢效率不高，而字典是一個哈希表，查詢效率較高，因此我將目標(biāo)文件列表轉(zhuǎn)換成一個只有鍵沒有值的字典：

復(fù)制代碼代碼如下:

dstFiles = dict(map(lambda x:[x, ''], os.listdir(dst)))

然后我遍歷源文件列表，若該路徑是一個文件夾，先判斷該文件夾在目標(biāo)路徑中是否存在，若不存在，則先創(chuàng)建一個新路徑。然后遞歸調(diào)用本函數(shù)。其實不存在的時候更高效的方法是調(diào)用shutil.copytree()函數(shù)，但由于此處需要計算拷貝的文件數(shù)量，因此就沒有調(diào)用該函數(shù)。

若該路徑是一個文件，則首先判斷該文件在目標(biāo)文件夾中是否存在。若不存在，則拷貝。

由于寫這個腳本主要是為了同步手機相冊到PC，因此只簡單地判斷一下文件名。若要判斷不同名但相同的文件，則可以繼續(xù)判斷一下md5值，這里就不再贅述。

完整代碼如下：

復(fù)制代碼代碼如下:

#!/usr/bin/env python
# -*- coding: UTF-8 -*-

# 輸入兩個文件夾a和b路徑，將a中的文件拷進(jìn)b，并計算拷貝的文件數(shù)。重復(fù)的不作處理。

import os
import shutil

def copyFiles(src, dst):
    srcFiles = os.listdir(src)
    dstFiles = dict(map(lambda x:[x, ''], os.listdir(dst)))
    filesCopiedNum = 0

    # 對源文件夾中的每個文件若不存在于目的文件夾則復(fù)制
    for file in srcFiles:
        src_path = os.path.join(src, file)
        dst_path = os.path.join(dst, file)
        # 若源路徑為文件夾，若存在于目標(biāo)文件夾，則遞歸調(diào)用本函數(shù)；否則先創(chuàng)建再遞歸。
        if os.path.isdir(src_path):
            if not os.path.isdir(dst_path):
                os.makedirs(dst_path)
            filesCopiedNum += copyFiles(src_path, dst_path)
        # 若源路徑為文件，不重復(fù)則復(fù)制，否則無操作。
        elif os.path.isfile(src_path):
            if not dstFiles.has_key(file):
                shutil.copyfile(src_path, dst_path)
                filesCopiedNum += 1

    return filesCopiedNum

def test():
    src_dir = os.path.abspath(raw_input('Please enter the source path: '))
    if not os.path.isdir(src_dir):
        print 'Error: source folder does not exist!'
        return 0

    dst_dir = os.path.abspath(raw_input('Please enter the destination path: '))
    if os.path.isdir(dst_dir):
        num = copyFiles(src_dir, dst_dir)
    else:
        print 'Destination folder does not exist, a new one will be created.'
        os.makedirs(dst_dir)
        num = copyFiles(src_dir, dst_dir)

print 'Copy complete:', num, 'files copied.'

if __name__ == '__main__':
test()

您可能感興趣的文章: