zoukankan      html  css  js  c++  java
  • openpyxl 读取前端传输的文件,并处理

        def create(self, request, *args, **kwargs):
            container_type_list = settings.CONTAINER_TYPE_LIST
            gas_file = request.data.get('gas_file')
            if gas_file:
                """gas_file存在,创建非TA计划申请表"""
                try:
                    df = openpyxl.load_workbook(gas_file, data_only=True)
                    sheet = df[df.sheetnames[0]]
                    rows = get_max_row(sheet)
                    asset_type = 'asset_type' in sheet.cell(1, 1).value
                    asset_type_id = 'asset_type_id' in sheet.cell(1, 2).value
                    container_type = 'container_type' in sheet.cell(1, 3).value
                    container_type_comments = 'container_type_comments' in sheet.cell(1, 4).value
                    is_fields_right = all([asset_type,asset_type_id,container_type,container_type_comments])
                    if not is_fields_right: raise Exception("字段不匹配")
                except Exception as e:
                    logger.error(e)
                    resp = unified_response(code=status.InputErr, message='文件格式不正确')
                    return Response(resp)
                try:
                    with transaction.atomic():
                        for row in range(1, rows + 1):
                            if row > 1:
                                asset_type = sheet.cell(row, 1).value
                                asset_type_id = sheet.cell(row, 2).value
                                container_type = str(sheet.cell(row, 3).value).strip()
                                container_type_comments = sheet.cell(row, 4).value
                                if container_type not in container_type_list:
                                    raise ValueError('上传表格失败,container_type:%s 不符合要求'%container_type)
                                Gas.objects.create(
                                    asset_type=asset_type,
                                    asset_type_id=asset_type_id,
                                    container_type=container_type,
                                    container_type_comments=container_type_comments
                                )
                except ValueError as e:
                    logger.error(e)
                    resp = unified_response(code=status.InputErr, message=str(e))
                    return Response(resp)
                except Exception as e:
                    logger.error(e)
                    resp = unified_response(code=status.InputErr, message='上传失败,文件内容有误,请修改后重新上传')
                    return Response(resp)
                resp = unified_response(message='文件上传成功!')
                return Response(resp)
            container_type = str(request.data.get('container_type')).strip()
            if container_type not in container_type_list:
                resp = unified_response(code=status.InputErr, message='操作失败,container_type:%s 不符合要求'%container_type)
                return Response(resp)
            return super().create(request, *args, **kwargs)
    
  • 相关阅读:
    BZOJ4416 SHOI2013阶乘字符串(状压dp)
    雅礼集训 Day2 T3 联盟 解题报告
    雅礼集训 Day1 T2 折射
    雅礼集训 Day1 T1 养花
    P1494 [国家集训队]小Z的袜子/莫队学习笔记(误
    洛谷 P2155 [SDOI2008]沙拉公主的困惑 解题报告
    动态MST
    洛谷 P2606 [ZJOI2010]排列计数 解题报告
    牛客 2018NOIP 模你赛2 T2 分糖果 解题报告
    洛谷 P3396 哈希冲突 解题报告
  • 原文地址:https://www.cnblogs.com/pythonwl/p/15120010.html
Copyright © 2011-2022 走看看