文件拷贝python脚本

import os
import sys

import shutil
import argparse

from glob import glob as glob


def parse_opt(known=False):
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--src_dir', type=str, default='./', help='farther folder of folders.')
    parser.add_argument(
        '--dst_path', type=str, default='./view_split', help='father folder of dst files in different folders.')

    opt = parser.parse_known_args()[0] if known else parser.parse_args()
    return opt


def copy_first_10_files(source_dir, dest_dir):
    for root, dirs, files in os.walk(source_dir):
        for dir in dirs:
            source_subdir = os.path.join(root, dir)
            dest_subdir = source_subdir.replace(source_dir, dest_dir+"/", 1)

            os.makedirs(dest_subdir, exist_ok=True)

            if not os.path.exists(dest_subdir):
                os.makedirs(dest_subdir)

            file_list = os.listdir(source_subdir)
            file_list.sort()  # Sort files for consistent order

            for i, file in enumerate(file_list):
                if i < 1:
                    source_file = os.path.join(source_subdir, file)
                    dest_file = os.path.join(dest_subdir, file)
                    shutil.copy2(source_file, dest_file)
                    print("Copied: {} to {}".format(source_file, dest_file))


def main(opt):
    cur_dir = os.path.dirname(__file__)
    source_directory = opt.src_dir
    destination_directory = opt.dst_path

    copy_first_10_files(source_directory, destination_directory)


if __name__ == "__main__":

    opt = parse_opt()

    main(opt)