ha-mes-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-form-item label="BU">
          <el-select v-model="bu" placeholder="请选择"  style="width: 295px">
            <el-option
              v-for = "i in userBuList"
              :key = "i.buNo"
              :label = "i.sitename"
              :value = "i.buNo">
              <span style="float: left;width: 100px">{{ i.sitename }}</span>
              <span style="float: right; color: #8492a6;white-space:nowrap;overflow:hidden;text-overflow:ellipsis; font-size: 11px;">
                {{ i.buDesc }}
              </span>
            </el-option>
          </el-select>
        </el-form-item>
        <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" @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
  getSiteAndBuByUserName
} 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: [],
      bu: '',
      userBuList:[],
      pageData: {
        flag: '',
        createBy: '',
        site: ''
      },
    }
  },
  created(){
    this.getSiteAndBu();
  },
  methods: {
    // 获取用户的bu
    getSiteAndBu () {
      let tempData = {
        username: this.$store.state.user.name,
      }
      getSiteAndBuByUserName(tempData).then(({data}) => {
        if (data.code === 0) {
          this.userBuList = data.rows
        }
      })
    },
    //初始化组件的参数
    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.$emit('refreshPageTables');
      // 关闭当前的页面
      this.visible = false;
    },
    /*保修当前的数据*/
    saveUploadFile(){
      if (this.bu==''||this.bu == null){
        this.$message.warning("请选择BU!");
        return false;
      }
      /*判断文件是否上传*/
      if(null == this.fileList || 0 == this.fileList.length){
        this.$message.error("请先上传文件!");
        return false;
      }
      const formData = new FormData();
      formData.append("file", this.fileList[0].raw);
      formData.append("createBy", this.pageData.createBy);
      formData.append("site", this.bu.split('_')[0]);
      if (this.pageData.flag === 'item'){
        uploadExcel(formData).then(({data}) => {
          if (data.code === 0) {
            this.$message.success(data.msg);
            // 清空文件上传记录
            this.$refs.uploadFile.clearFiles();
            // 关闭窗口并刷新页面
            this.closeDialog();
          }else {
            this.$message.warning(data.msg);
          }
        })
      } else if (this.pageData.flag === 'template'){
        formData.set('site',this.bu)
        uploadTemplateExcel(formData).then(({data}) => {
          if (data.code === 0) {
            this.$message.success(data.msg);
            // 清空文件上传记录
            this.$refs.uploadFile.clearFiles();
            // 关闭窗口并刷新页面
            this.closeDialog();
          }else {
            this.$message.warning(data.msg);
          }
        })
      }else if (this.pageData.flag === 'partAttribute'){
        formData.set('site',this.bu)
        uploadPartAttributeExcel(formData).then(({data}) => {
          if (data.code === 0) {
            this.$message.success(data.msg);
            // 清空文件上传记录
            this.$refs.uploadFile.clearFiles();
            // 关闭窗口并刷新页面
            this.closeDialog();
          }else {
            this.$message.warning(data.msg);
          }
        })
      }else if (this.pageData.flag === 'familyAttribute'){
        uploadFamilyAttributeExcel(formData).then(({data}) => {
          if (data.code === 0) {
            this.$message.success(data.msg);
            // 清空文件上传记录
            this.$refs.uploadFile.clearFiles();
            // 关闭窗口并刷新页面
            this.closeDialog();
          }else {
            this.$message.warning(data.msg);
          }
        })
      }
      this.bu = '';
    },
    // 下载
    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>