xz-vue/src/views/modules/qc/qc_upload.vue

<template>
  <div class="customer-css">
    <el-dialog :title="titleCon" :close-on-click-modal="false" :visible.sync="visible" width="390px" style="height: 520px;" class="customer-dialog">
      <el-form :inline="true" label-position="top" label-width="80px">
        <el-button type="primary" @click="downloadFile()">下载文件模板</el-button>
        <el-row>
          <el-col :span="24">
            <el-upload class="customer-upload" drag action="javascript:void(0);" ref="uploadFile" :limit="1" accept=".xlsx,.xls"
                       :before-upload="beforeUploadHandle" :on-change="onChange" :auto-upload="false" style="text-align: left;">
              <i class="el-icon-upload"></i>
              <div class="el-upload__text">将文件拖到此处，或<em>点击上传</em></div>
            </el-upload>
          </el-col>
        </el-row>
      </el-form>
      <span slot="footer" class="dialog-footer">
        <el-button type="primary" :loading="uploadLoading" @click="saveUploadFile()">保存</el-button>
        <el-button type="primary" @click="closeDialog">关闭</el-button>
      </span>
    </el-dialog>
  </div>
</template>

<script>
import {
  uploadExcel, // 导入项目文件
  uploadTemplateExcel, // 导入模板文件
  uploadPartAttributeExcel, // 导入物料属性文件
  uploadFamilyAttributeExcel, // 导入类别属性文件
  queryFileId, // 查询文件ID
} from "@/api/qc/qc.js"
import { downLoadObjectFile } from '@/api/eam/eam_object_list.js'
import axios from 'axios'
import Vue from 'vue'
export default {
  data() {
    return {
      titleCon: '文件导入',
      visible: false,
      fileList: [],
      uploadLoading: false,
      pageData: {
        flag: '',
        createBy: '',
        site: '',
        buNo: '',
      },
    }
  },
  methods: {
    // 初始化组件的参数
    init (currentRow) {
      // 获得类别
      this.pageData = JSON.parse(JSON.stringify(currentRow))
      // 打开页面
      this.visible = true
    },
    // 上传之前
    beforeUploadHandle (file) {
      let extName = file[0].name.substring(file[0].name.lastIndexOf('.')).toLowerCase()
      if (!(extName === '.xlsx' || extName === '.xls')) {
        this.$message.error('数据导入失败，请选择正确的xlsx模板文件')
        return false
      }
    },
    /*选择上传文件时*/
    onChange (file) {
      this.fileList.push(file)
    },
    /*关闭modal*/
    closeDialog () {
      this.fileList = []
      this.$refs.uploadFile.clearFiles()
      // 刷新报工的页面
      this.$emit('refreshPageTables')
      // 关闭当前的页面
      this.visible = false
    },
    /*保修当前的数据*/
    saveUploadFile () {
      this.uploadLoading = true
      /*判断文件是否上传*/
      if (null == this.fileList || 0 == this.fileList.length) {
        this.$message.error("请先上传文件!")
        this.uploadLoading = false
        return false
      }
      const formData = new FormData()
      formData.append("file", this.fileList[0].raw)
      formData.append("createBy", this.pageData.createBy)
      formData.append("site", this.pageData.site)
      formData.append("buNo", this.pageData.buNo)

      // 根据 flag 发不同的接口
      let promise
      if (this.pageData.flag === 'item') {
        promise = uploadExcel(formData)
      } else if (this.pageData.flag === 'template') {
        promise = uploadTemplateExcel(formData)
      } else if (this.pageData.flag === 'partAttribute') {
        promise = uploadPartAttributeExcel(formData)
      } else if (this.pageData.flag === 'familyAttribute') {
        promise = uploadFamilyAttributeExcel(formData)
      }
      promise
        .then(({ data }) => {
          if (data.code === 0) {
            this.$message.success(data.msg)
            this.$refs.uploadFile.clearFiles()
            this.closeDialog()
          } else {
            this.$message.warning(data.msg)
          }
        })
        .catch(err => {
          this.$message.error("上传失败，请稍后重试")
          console.error(err)
        })
        .finally(() => {
          this.uploadLoading = false  // 请求结束时再关 loading
        })
    },
    // 下载
    async downloadFile () {
      let file = {
        id: 0,
        fileName: ''
      }
      let tempData = {
        orderRef1: '',
        orderRef2: ''
      }
      if (this.pageData.flag === 'item') { // 检验项目
        tempData.orderRef1 = 'qc';
        tempData.orderRef2 = 'itemFormat';
        await queryFileId(tempData).then(({data}) => {
          if (data && data.code === 0) {
            file.id = data.data.id
            file.fileName = data.data.fileName
          } else {
            this.$alert(data.msg, '错误', {
              confirmButtonText: '确定'
            })
          }
        })
      }else if (this.pageData.flag === 'template'){ // 检验模板
        tempData.orderRef1 = 'qc';
        tempData.orderRef2 = 'templateFormat';
        await queryFileId(tempData).then(({data}) => {
          if (data && data.code === 0) {
            file.id = data.data.id
            file.fileName = data.data.fileName
          } else {
            this.$alert(data.msg, '错误', {
              confirmButtonText: '确定'
            })
          }
        })
      }else if (this.pageData.flag === 'partAttribute'){ // 物料属性设置
        tempData.orderRef1 = 'qc';
        tempData.orderRef2 = 'partAttributeFormat';
        await queryFileId(tempData).then(({data}) => {
          if (data && data.code === 0) {
            file.id = data.data.id
            file.fileName = data.data.fileName
          } else {
            this.$alert(data.msg, '错误', {
              confirmButtonText: '确定'
            })
          }
        })
      }else if (this.pageData.flag === 'familyAttribute'){ // 类别属性设置
        tempData.orderRef1 = 'qc';
        tempData.orderRef2 = 'familyAttributeFormat';
        await queryFileId(tempData).then(({data}) => {
          if (data && data.code === 0) {
            file.id = data.data.id
            file.fileName = data.data.fileName
          } else {
            this.$alert(data.msg, '错误', {
              confirmButtonText: '确定'
            })
          }
        })
      }
      await downLoadObjectFile(file).then(({data}) => {
          // 不限制文件下载类型
          const blob = new Blob([data], {type: "application/octet-stream"})
          // 下载文件名称
          const fileName = file.fileName
          // a标签下载
          const linkNode = document.createElement('a')
          // a标签的download属性规定下载文件的名称
          linkNode.download = fileName
          linkNode.style.display = 'none'
          // 生成一个Blob URL
          linkNode.href = URL.createObjectURL(blob)
          document.body.appendChild(linkNode)
          // 模拟在按钮上的一次鼠标单击
          linkNode.click()
          // 释放URL 对象
          URL.revokeObjectURL(linkNode.href)
          document.body.removeChild(linkNode)
        })
    },

  }
}
</script>
<style scoped lang="scss">
</style>