liujiandao
2024-04-22 8cb1da78fd4d237b278ed4d512c6c4f04b663cfc
controllers/operation.go
@@ -15,6 +15,7 @@
   "strconv"
   "strings"
   "time"
   "wms/conf"
   "wms/constvar"
   "wms/extend/code"
   "wms/extend/util"
@@ -870,24 +871,149 @@
   }
   var fileUrl string
   companyName := conf.WebConf.CompanyName
   if companyName == "" {
      util.ResponseFormat(c, code.RequestParamError, "缺少工厂名称配置")
      return
   }
   if operation.BaseOperationType == constvar.BaseOperationTypeIncoming {
      fileUrl, err = ExportInputSelfmade(constvar.FileTemplateCategory_Selfmade, operation)
      if err != nil {
         util.ResponseFormat(c, code.RequestParamError, err.Error())
         return
      if companyName == "jialian" {
         fileUrl, err = JialianOperation(constvar.FileTemplateCategory_JialianInput, operation)
         if err != nil {
            util.ResponseFormat(c, code.RequestParamError, err.Error())
            return
         }
      } else if companyName == "geruimi" {
         fileUrl, err = ExportInputSelfmade(constvar.FileTemplateCategory_Selfmade, operation)
         if err != nil {
            util.ResponseFormat(c, code.RequestParamError, err.Error())
            return
         }
      }
   } else if operation.BaseOperationType == constvar.BaseOperationTypeOutgoing {
      fileUrl, err = ExportOutputOperation(constvar.FileTemplateCategory_Output, operation)
      if err != nil {
         util.ResponseFormat(c, code.RequestParamError, err.Error())
         return
      if companyName == "jialian" {
         fileUrl, err = JialianOperation(constvar.FileTemplateCategory_JialianOutput, operation)
         if err != nil {
            util.ResponseFormat(c, code.RequestParamError, err.Error())
            return
         }
      } else if companyName == "geruimi" {
         fileUrl, err = ExportInputSelfmade(constvar.FileTemplateCategory_Output, operation)
         if err != nil {
            util.ResponseFormat(c, code.RequestParamError, err.Error())
            return
         }
      }
   } else {
      util.ResponseFormat(c, code.RequestParamError, "该记录不支持打印")
      return
   }
   util.ResponseFormat(c, code.Success, fileUrl)
   m := make(map[string]string)
   m["url"] = fileUrl
   m["name"] = companyName
   util.ResponseFormat(c, code.Success, m)
}
func JialianOperation(category constvar.FileTemplateCategory, operation *models.Operation) (string, error) {
   template, err := models.NewFileTemplateAttachmentSearch().SetCategory(category).First()
   if err != nil {
      return "", errors.New("获取模版记录失败:" + err.Error())
   }
   readerCloser, err := http.HttpGetWithReadCloser(template.FileUrl)
   if err != nil {
      return "", errors.New("获取模版失败:" + err.Error())
   }
   f, err := excelize.OpenReader(readerCloser)
   if err != nil {
      return "", errors.New("读取excel模版失败:" + err.Error())
   }
   readerCloser.Close()
   defer f.Close()
   f.SetCellValue("Sheet1", "H2", operation.OperationDate)
   f.SetCellValue("Sheet1", "O2", operation.Number)
   rowIndex := 5
   totalAmount := decimal.NewFromInt(0)
   totalPrice := decimal.NewFromInt(0)
   for i, v := range operation.Details {
      //设置表单最多9条数据
      if i > 8 {
         break
      }
      f.SetCellValue("Sheet1", "A"+strconv.Itoa(rowIndex), v.Product.Name)
      f.SetCellValue("Sheet1", "B"+strconv.Itoa(rowIndex), v.Product.Type)
      f.SetCellValue("Sheet1", "C"+strconv.Itoa(rowIndex), v.Product.Unit)
      f.SetCellValue("Sheet1", "D"+strconv.Itoa(rowIndex), v.Amount.String())
      f.SetCellValue("Sheet1", "E"+strconv.Itoa(rowIndex), v.Product.SalePrice.String())
      if !v.Product.SalePrice.IsZero() {
         ap := v.Amount.Mul(v.Product.SalePrice)
         totalPrice = totalPrice.Add(ap)
         price := ap.String()
         split := strings.Split(price, ".")
         if len(split) == 2 {
            for n, a := range split[1] {
               f.SetCellValue("Sheet1", string('M'+n)+strconv.Itoa(rowIndex), string(a))
            }
         }
         s := split[0]
         n := 0
         for j := len(s) - 1; j >= 0; j-- {
            if n == 6 {
               f.SetCellValue("Sheet1", string('L'-n)+strconv.Itoa(rowIndex), s[:j])
               break
            }
            f.SetCellValue("Sheet1", string('L'-n)+strconv.Itoa(rowIndex), string(s[j]))
            n++
         }
      }
      rowIndex++
      totalAmount = totalAmount.Add(v.Amount)
   }
   //合计
   if !totalPrice.IsZero() {
      price := totalPrice.String()
      split := strings.Split(price, ".")
      if len(split) == 2 {
         for n, a := range split[1] {
            f.SetCellValue("Sheet1", string('M'+n)+"14", string(a))
         }
      } else {
         f.SetCellValue("Sheet1", "M14", "0")
         f.SetCellValue("Sheet1", "N14", "0")
      }
      s := split[0]
      n := 0
      for j := len(s) - 1; j >= 0; j-- {
         if n == 6 {
            f.SetCellValue("Sheet1", string('L'-n)+"14", s[:j])
            break
         }
         f.SetCellValue("Sheet1", string('L'-n)+"14", string(s[j]))
         n++
      }
   }
   f.SetCellValue("Sheet1", "D14", totalAmount)
   f.SetCellValue("Sheet1", "B15", operation.Manager)
   f.SetCellValue("Sheet1", "D15", operation.Accountant)
   f.SetCellValue("Sheet1", "F15", operation.Custodian)
   buf, err := f.WriteToBuffer()
   if err != nil {
      return "", err
   }
   fileUrl, err := upload.UploadFileToSeaWeed(string(constvar.FileType_File), uuid.NewV4().String()+".xlsx", buf.Bytes())
   if err != nil {
      logx.Errorf("file upload err: %v", err)
      return "", err
   }
   return fileUrl, nil
}
func ExportInputSelfmade(category constvar.FileTemplateCategory, operation *models.Operation) (string, error) {