递归处理复制变量目录按原路径复制到新目录的脚本

 

脚本如下:

 

# coding: utf-8

"""
    该脚本主要做把源目录下所有文件，照搬原路径基础上复制文件
"""

import os
# import shutil
import sys
import time


def copy_and_save_paths(src_dir, dst_dir):
    """
    :param src_dir: 源目录路径
    :param dst_dir: 目标目录路径
    :return: 成功 True, 失败 False  Bool
    """
    # 检查源目录存在
    if not os.path.exists(src_dir):
        print('【ERROR】 源目录不存在')
        return False

    new_dst_dir = '%s/%s' % (dst_dir, src_dir)  # 拼接出目标路径: 需保存的路径和源路径结合
    # 确保目标目录存在
    if not os.path.exists(new_dst_dir):
        os.makedirs(new_dst_dir)

    cd_cp_cmd = 'cd %s && cp -rf ./* %s' % (src_dir, new_dst_dir)  # cd进去源目录路径, 拷贝到新路径
    # cp_cmd = 'cp -rf ./* %s' % new_dst_dir  # 拷贝到新路径
    print('run cmd: %s' % cd_cp_cmd)
    cmd_res = os.system(cd_cp_cmd)
    if str(cmd_res) != '0':
        print('【ERROR】cd_cp_cmd fail !!!!!!!!!')
        return False
    else:
        print('cd_cp_cmd success.')
    return True


def run_all(file_path, dst_dir):
    """
        file_path: 目录列表文件
        des_dir: 需复制到的目标目录路径
        获取源路径的目录列表文件执行复制操作
    """
    with open(file_path, 'r') as ff:
        file_c = ff.read()
    n = 0  # 进度数
    r_n = 0  # 成功数量
    e_n = 0  # 错误数量
    all_n = 0
    nn_n = 0
    if file_c:
        path_list = file_c.split('\n')
        n = 0    # 进度数
        r_n = 0  # 成功数量
        e_n = 0  # 错误数量
        all_n = len(path_list)  # 总数
        for i in path_list:
            print('*' * 50)
            if i.strip() != '':
                print('run path: %s' % i.strip())
                rr = copy_and_save_paths(i.strip(), dst_dir)
                if not rr:
                    print('run fail path: %s' % i.strip())
                    e_n += 1
                else:
                    r_n += 1
            else:
                nn_n += 1
            n += 1
            # print(n)
            print('run count: %s' % str(round(float(n)/float(all_n) * 100, 2)) + '%')  # 处理进度
    print('run all finish')
    print('all: %s, none: %s, run: %s, success: %s, error: %s' % (str(all_n), str(nn_n), str(n), str(r_n), str(e_n)))


if __name__ == "__main__":
    print('args: ', sys.argv)
    if len(sys.argv) < 3:
        print('参数缺失')
    else:
        print('start run..')
        # 脚本接收参数 第一个是存有目录路径列表的文件； 第二个是目标路径
        run_all(sys.argv[1], sys.argv[2])


# 后台执行示例: nohup python -u /data/tt.py /data/ff.txt /data/test2 >> /data/log.log 2>&1 &