Python 文件拷贝

    添加时间:2013-7-18 点击量:

    比来在备份上的照片的时辰,纯手工操纵感觉有些麻烦,就想写个脚本主动进行。因为备份的时辰有些照片以前备份过了,所以须要有个判重操纵。


    首要功能在copyFiles()函数里实现,如下:



     1 def copyFiles(src, dst):
    
    2 srcFiles = os.listdir(src)
    3 dstFiles = dict(map(lambda x:[x, ], os.listdir(dst)))
    4 filesCopiedNum = 0
    5
    6 对源文件夹中的每个文件若不存在于目标文件夹则复制
    7 for file in srcFiles:
    8 src_path = os.path.join(src, file)
    9 dst_path = os.path.join(dst, file)
    10 若源路径为文件夹,若存在于目标文件夹,则递归调用本函数;不然先创建再递归。
    11 if os.path.isdir(src_path):
    12 if not os.path.isdir(dst_path):
    13 os.makedirs(dst_path)
    14 filesCopiedNum += copyFiles(src_path, dst_path)
    15 若源路径为文件,不反复则复制,不然无操纵。
    16 elif os.path.isfile(src_path):
    17 if not dstFiles.has_key(file):
    18 shutil.copyfile(src_path, dst_path)
    19 filesCopiedNum += 1
    20
    21 return filesCopiedNum


     这里我起首应用os.listdir()函数来遍历源文件夹src和目标文件夹dst,获得两个文件列表,但因为我须要判重操纵,是以须要在dst文件列表中进行查询操纵。因为列表的查询效力不高,而字典是一个哈希表,查询效力较高,是以我将目标文件列表转换成一个只有键没有值的字典:



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


    然后我遍历源文件列表,若该路径是一个文件夹,先断定该文件夹在目标路径中是否存在,若不存在,则先创建一个新路径。然后递归调用本函数。其实不存在的时辰更高效的办法是调用shutil.copytree()函数,但因为此处须要策画拷贝的文件数量,是以就没有调用该函数。


    若该路径是一个文件,则起首断定该文件在目标文件夹中是否存在。若不存在,则拷贝。


    因为写这个脚本主如果为了同步相册到PC,是以只简单地断定一下文件名。若要断定不合名但雷同的文件,则可以持续断定一下md5值,这里就不再赘述。



    完全代码如下:



     1 !/usr/bin/env python
    
    2 -- coding: UTF-8 --
    3
    4 输入两个文件夹a和b路径,将a中的文件拷进b,并策画拷贝的文件数。反复的不作处理惩罚。
    5
    6 import os
    7 import shutil
    8
    9 def copyFiles(src, dst):
    10 srcFiles = os.listdir(src)
    11 dstFiles = dict(map(lambda x:[x, ], os.listdir(dst)))
    12 filesCopiedNum = 0
    13
    14 对源文件夹中的每个文件若不存在于目标文件夹则复制
    15 for file in srcFiles:
    16 src_path = os.path.join(src, file)
    17 dst_path = os.path.join(dst, file)
    18 若源路径为文件夹,若存在于目标文件夹,则递归调用本函数;不然先创建再递归。
    19 if os.path.isdir(src_path):
    20 if not os.path.isdir(dst_path):
    21 os.makedirs(dst_path)
    22 filesCopiedNum += copyFiles(src_path, dst_path)
    23 若源路径为文件,不反复则复制,不然无操纵。
    24 elif os.path.isfile(src_path):
    25 if not dstFiles.has_key(file):
    26 shutil.copyfile(src_path, dst_path)
    27 filesCopiedNum += 1
    28
    29 return filesCopiedNum
    30
    31 def test():
    32 src_dir = os.path.abspath(raw_input(Please enter the source path: ))
    33 if not os.path.isdir(src_dir):
    34 print Error: source folder does not exist!
    35 return 0
    36
    37 dst_dir = os.path.abspath(raw_input(Please enter the destination path: ))
    38 if os.path.isdir(dst_dir):
    39 num = copyFiles(src_dir, dst_dir)
    40 else:
    41 print Destination folder does not exist, a new one will be created.
    42 os.makedirs(dst_dir)
    43 num = copyFiles(src_dir, dst_dir)
    44
    45 print Copy complete:, num, files copied.
    46
    47 if __name__ == __main__:
    48 test()


    读书,不要想着实用,更不要有功利心。读书只为了自身的修养。邂逅一本好书如同邂逅一位知己,邂逅一个完美之人。有时心生敬意,有时怦然心动。仿佛你心底埋藏多年的话,作者替你说了出来,你们在时光深处倾心相遇的一瞬间,情投意合,心旷神怡。
    分享到: