From 10806eefe017093646a671af0b321b9df0a2ebc5 Mon Sep 17 00:00:00 2001
From: jiangshuai <291802688@qq.com>
Date: 星期一, 20 十一月 2023 16:02:39 +0800
Subject: [PATCH] 打印入库-自制导出excel 增加对取消的统计

---
 controllers/operation.go |  129 +++++++++++++++++++++++++++++++++++++-----
 1 files changed, 113 insertions(+), 16 deletions(-)

diff --git a/controllers/operation.go b/controllers/operation.go
index b673452..dee8bd9 100644
--- a/controllers/operation.go
+++ b/controllers/operation.go
@@ -831,14 +831,14 @@
 	util.ResponseFormat(c, code.Success, "鎿嶄綔鎴愬姛")
 }
 
-// inputSelfmade
+// outputOperation
 //
 //	@Tags		鍏ュ簱/鍑哄簱
-//	@Summary	鎵撳嵃-鑷埗-鍏ュ簱
+//	@Summary	鎵撳嵃
 //	@Produce	application/json
 //	@Param		id	path		int			true	"id"
 //	@Success	200	{object}	util.Response	"鎴愬姛"
-//	@Router		/api-wms/v1/operation/inputSelfmade/{id} [put]
+//	@Router		/api-wms/v1/operation/outputOperation/{id} [put]
 func (slf OperationController) InputSelfmade(c *gin.Context) {
 	id, err := strconv.Atoi(c.Param("id"))
 	if err != nil {
@@ -854,17 +854,45 @@
 		util.ResponseFormat(c, code.RequestParamError, "鏈壘鍒扮浉鍏冲嚭鍏ュ簱淇℃伅:"+err.Error())
 		return
 	}
-
-	readerCloser, err := http.HttpGetWithReadCloser(conf.FileTemplateConf.InputSelfmadeAddr)
-	if err != nil {
-		util.ResponseFormat(c, code.RequestParamError, "鑾峰彇妯$増澶辫触:"+err.Error())
+	if len(operation.Details) == 0 {
+		util.ResponseFormat(c, code.RequestParamError, "娌℃湁鐩稿叧浜у搧鏄庣粏鐨勪俊鎭�")
 		return
+	}
+
+	var fileUrl string
+	if operation.BaseOperationType == constvar.BaseOperationTypeIncoming {
+		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
+		}
+	} else {
+		util.ResponseFormat(c, code.RequestParamError, "璇ヨ褰曚笉鏀寔鎵撳嵃")
+		return
+	}
+
+	util.ResponseFormat(c, code.Success, fileUrl)
+}
+
+func ExportInputSelfmade(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.Attachment.FileUrl)
+	if err != nil {
+		return "", errors.New("鑾峰彇妯$増澶辫触:" + err.Error())
 	}
 
 	f, err := excelize.OpenReader(readerCloser)
 	if err != nil {
-		util.ResponseFormat(c, code.RequestParamError, "璇诲彇excel妯$増澶辫触锛�"+err.Error())
-		return
+		return "", errors.New("璇诲彇excel妯$増澶辫触锛�" + err.Error())
 	}
 	readerCloser.Close()
 
@@ -905,8 +933,7 @@
 		totalAmount = totalAmount.Add(v.Amount)
 	}
 	if err := f.MergeCell("Sheet1", "A"+strconv.Itoa(rowIndex), "C"+strconv.Itoa(rowIndex)); err != nil {
-		util.ResponseFormat(c, code.RequestParamError, err.Error())
-		return
+		return "", err
 	}
 	f.SetCellValue("Sheet1", "A"+strconv.Itoa(rowIndex), "鍚堣")
 	f.SetCellValue("Sheet1", "E"+strconv.Itoa(rowIndex), totalAmount.String())
@@ -920,16 +947,86 @@
 
 	buf, err := f.WriteToBuffer()
 	if err != nil {
-		util.ResponseFormat(c, code.RequestParamError, err.Error())
-		return
+		return "", err
 	}
 
 	fileUrl, err := upload.UploadFileToSeaWeed(string(constvar.FileType_File), "xlsx", buf.Bytes())
 	if err != nil {
 		logx.Errorf("file upload err: %v", err)
-		util.ResponseFormat(c, code.RequestParamError, err.Error())
-		return
+		return "", err
 	}
 
-	util.ResponseFormat(c, code.Success, fileUrl)
+	return fileUrl, nil
+}
+
+func ExportOutputOperation(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.Attachment.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()
+
+	style, _ := f.NewStyle(&excelize.Style{
+		Border: []excelize.Border{
+			{Type: "left", Color: "#000000", Style: 1},
+			{Type: "top", Color: "#000000", Style: 1},
+			{Type: "bottom", Color: "#000000", Style: 1},
+			{Type: "right", Color: "#000000", Style: 1},
+		},
+		Alignment: &excelize.Alignment{
+			Horizontal: "center",
+			Vertical:   "center",
+		},
+	})
+
+	f.SetCellValue("Sheet1", "C2", operation.FromLocation.Name)
+	f.SetCellValue("Sheet1", "H2", operation.WaybillNumber)
+	f.SetCellValue("Sheet1", "H3", operation.SourceNumber)
+	f.SetCellValue("Sheet1", "K3", operation.Number)
+	f.SetCellValue("Sheet1", "C4", operation.CompanyName)
+
+	rowIndex := 8
+	totalAmount := decimal.NewFromInt(0)
+	totalWeight := decimal.NewFromInt(0)
+
+	addRows := len(operation.Details) - 1
+	if addRows > 0 {
+		f.InsertRows("Sheet1", 9, addRows)
+		f.SetCellStyle("Sheet1", "A9", "K"+strconv.Itoa(rowIndex+addRows), style)
+	}
+
+	for k, v := range operation.Details {
+		productTotalWeight := v.Amount.Mul(v.Product.Weight).Div(decimal.NewFromInt(1000))
+		f.SetCellValue("Sheet1", "A"+strconv.Itoa(rowIndex), k+1)
+		f.SetCellValue("Sheet1", "B"+strconv.Itoa(rowIndex), v.Product.Name)
+		f.SetCellValue("Sheet1", "H"+strconv.Itoa(rowIndex), v.Amount)
+		f.SetCellValue("Sheet1", "I"+strconv.Itoa(rowIndex), v.Product.Weight)
+		f.SetCellValue("Sheet1", "J"+strconv.Itoa(rowIndex), productTotalWeight)
+		rowIndex++
+		totalAmount = totalAmount.Add(v.Amount)
+		totalWeight = totalWeight.Add(productTotalWeight)
+	}
+
+	buf, err := f.WriteToBuffer()
+	if err != nil {
+		return "", err
+	}
+
+	fileUrl, err := upload.UploadFileToSeaWeed(string(constvar.FileType_File), "xlsx", buf.Bytes())
+	if err != nil {
+		logx.Errorf("file upload err: %v", err)
+		return "", err
+	}
+
+	return fileUrl, nil
+
 }

--
Gitblit v1.8.0