使用python按规则将文件复制到指定文件夹

深藏阁楼爱情的钟 2022-01-23 01:23 319阅读 0赞

import os
    import shutil
    import xlrd
    
    
    def open_excel(f_name="./test.xlsx"):
        """
        打开excel文件
        :param f_name: 文件名
        :return: data
        """
        try:
            data = xlrd.open_workbook(f_name)
            return data
        except IOError as e:
            print(str(e))
    
    
    def get_stage(f_name="./test.xls", by_name="Sheet1",
                quest_index=1, f_dir='客观题', index=4):
        """
        根据索引获取EXCEL表格中的数据
        :param f_name: 文件名
        :param by_name: Sheet1名称
        :param quest_index: 问题索引
        :param f_dir: 复制文件目录
        :param quest_index: 问题ID索引
        :param index: 字段索引
        :return:
        """
        data = open_excel(f_name=f_name)   # 打开excel文件
        table = data.sheet_by_name(by_name)      # 根据sheet名字来获取excel中的sheet
        n_rows = table.nrows - 2           # 行数, 减去前两行
        for row_num in range(n_rows):      # 遍历表格
            row = table.row_values(row_num + 2)
            stage = row[index-1]
            quest_f = row[quest_index-1][3:]+".xls"
            qf_name = os.path.join(f_dir, quest_f)
            if stage == '小学':
                copy_file(f_dir=f_dir+'-'+stage, file=qf_name)
            elif stage == '初中':
                copy_file(f_dir=f_dir+'-'+stage, file=qf_name)
            else:
                print("error")
                break
    
    
    def copy_file(file='', f_dir='test'):
        """
        将文件复制到指定目录
        :param f_dir: 指定目录
        :param file: 指定文件
        :return:
        """
        shutil.copy(src=file, dst=f_dir)
    
    
    ROOT = "E:/python/project/matrix/data/导出数据5.29"
    f_name = '2.总表-主观题-available.xls'
    f_dir = '4.学生答题记录-主观题'
    sheet = 'Worksheet'
    get_stage(f_name=os.path.join(ROOT, f_name),
            by_name=sheet, f_dir=os.path.join(ROOT, f_dir))