python实现文件分组复制到不同目录的例子__current_folder_folder_capacity_path_count_abs_item_

当前位置: 编程技术>python

python实现文件分组复制到不同目录的例子

来源: 互联网发布时间：2014-10-07

本文导语: 场景：某个文件夹下面包含数量巨大的文件，需求需要将这些文件按组（比如5000个一组）存放到不同的目录中去。代码如下:# Filename: CopyFiles.pyimport osimport os.path folder_capacity = 20 def copy_files(src_dir, dest_dir): count = 0 curre...

场景：某个文件夹下面包含数量巨大的文件，需求需要将这些文件按组（比如5000个一组）存放到不同的目录中去。

代码如下:

# Filename: CopyFiles.py
import os
import os.path

folder_capacity = 20

def copy_files(src_dir, dest_dir):
    count = 0
    current_folder = ''

    for item in os.listdir(src_dir):
        abs_item = os.path.join(src_dir, item)
        if os.path.isfile(abs_item):
            count += 1
            if count%folder_capacity == 1:
                current_folder = os.path.join(dest_dir, str(count/folder_capacity))
                os.mkdir(current_folder)
            open(os.path.join(current_folder, item), 'wb').write(open(abs_item, 'rb').read())

copy_files(r'C:\src', r'C:\dest')