import shutil
import sys
import os

"""
69-477326 :: 版本: 1 :: 文件内部处理频繁切换，防止溢出测试

使用方法：
1.先用WriteFile 2.0.exe 工具生成一个 test.txt
2.和这个脚本放在一个目录中
3.执行脚本：
    python  copy_files.py   args
      args:
        init  # 初始化目录，会创建60个目录,有的会先删除
        clean # 清楚创建的目录
        js    # 生成 test_1.txt，test_3.txt ...,并依次放入上述目录中
        os    # 生成 test_2.txt，test_4.txt ...,并依次放入上述目录中
"""

# 定义公共的部分
path = os.getcwd()
file_path = os.path.join(path, 'test.txt')
dir_list = list(range(1, 61))

# 复制奇数
def jishu():
    file_list = list(range(121))[1::2]
    fd = dict(zip(file_list, dir_list))
    copy_file(fd)

# 复制偶数
def oushu():
    file_list = list(range(121))[2::2]
    fd = dict(zip(file_list, dir_list))
    copy_file(fd)

# 复制文件到目录中
def copy_file(fd):
    for file, dir in fd.items():
        newname = 'test_' + str(file) + '.txt'
        todir = os.path.join(path, str(dir), newname)
        shutil.copyfile(file_path, todir)

# 初始化目录
def init_dir():
    for dir in dir_list:
        dir_path = os.path.join(path, str(dir))
        if os.path.exists(dir_path):
            shutil.rmtree(dir_path)
        os.mkdir(dir_path)

def clean():
    for dir in dir_list:
        dir_path = os.path.join(path, str(dir))
        if os.path.exists(dir_path):
            shutil.rmtree(dir_path)


if __name__ == '__main__':
    # 定义传参
    if len(sys.argv) != 2:
        assert False, 'not get args'

    if sys.argv[1] not in ["js", "os", "init","clean"]:
        assert False, 'args is error,please input js or os '

    if sys.argv[1] == 'init':
        init_dir()
    elif sys.argv[1] == 'js':
        jishu()
    elif sys.argv[1] == 'os':
        oushu()
    elif sys.argv[1] == 'clean':
        clean()