python實現檔案分組複製到不同目錄的例子

情境：某個檔案夾下麵包含數量巨大的檔案，需求需要將這些檔案按組（比如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')

