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)
    
  • 相关阅读:
    spark系列-6、对Application,Driver,Job,Task,Stage的理解
    spark系列-5、RDD、DataFrame、Dataset的区别和各自的优势
    spark系列-4、spark序列化方案、GC对spark性能的影响
    spark系列-2、Spark 核心数据结构:弹性分布式数据集 RDD
    nginx学习(九):跨域配置和防盗链配置
    nginx学习(八):nginx配置gzip
    nginx学习(七):nginx提供静态资源服务
    nginx学习(六):日志切割
    nginx学习(五):nginx.conf 核心配置文件详解
    nginx学习(四):nginx处理web请求机制
  • 原文地址:https://www.cnblogs.com/pythonwl/p/15120010.html
Copyright © 2011-2022 走看看