2021-01-05

Python批量复制图片

需求：批量复制各级目录的所有图片到新的文件夹

1、需要复制的文件夹，有多级目录

2、文件夹中不只是图片还有其他文件夹

3、可以指定图片的索引复制

4、也可以指定需要复制的每个文件夹的图片数量

 

控制台输入：
输入需要复制的文件夹
输入目标目录的文件夹
输入需要复制图片的开始索引
输入需要复制的图片数量

import os
import shutil
from past.builtins import raw_input


def copy_image(files, destfolder, startIndex, endIndex):
    n = len(files)
    if n > endIndex:
        n = endIndex

    for i in range(startIndex - 1 , n):
        shutil.copy(files[i], destfolder)


def copy_folder(sourcefolder, destfolder, startIndex, endIndex):
    subfolders = os.listdir(sourcefolder)
    subfolders.sort()

    images = []
    for f in subfolders:
        file_path = os.path.join('%s/%s' % (sourcefolder, f))
        if os.path.isdir(file_path):
            copy_folder(file_path, destfolder, startIndex, endIndex)
        else:
            if os.path.splitext(file_path)[-1] == ".jpg":
                images.append(file_path)
            if len(images) > endIndex - 1:
                break
    if len(images) > 0:
        copy_image(images, destfolder, startIndex, endIndex)


# 批量复制图片
if __name__ == '__main__':
    # startFile为原始路径，startFile是移动的目标目录
    # 原始图片文件父目录
    sourcefolder = raw_input("请输入需要复制的文件夹：")
    # startFile = r"F:\\test"
    # 移动到目标文件夹
    destfolder = raw_input("请输入目标的文件夹：")
    # destinationFile = r"D:\\图片"
    startIndex = input("请输入开始图片的位置：")
    startIndex = int(startIndex)
    endIndex = input("请输入导入图片的数量：")
    endIndex = int(endIndex)
    copy_folder(sourcefolder, destfolder, startIndex, endIndex)