From e5df488268e29b272932e6cc1d2b1e7034590ba0 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 12 六月 2024 11:40:45 +0800
Subject: [PATCH] weedfs包放到项目里

---
 controllers/operation.go |  200 +++++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 172 insertions(+), 28 deletions(-)

diff --git a/controllers/operation.go b/controllers/operation.go
index 216c091..e334da1 100644
--- a/controllers/operation.go
+++ b/controllers/operation.go
@@ -2,7 +2,6 @@
 
 import (
 	"context"
-	"encoding/json"
 	"errors"
 	"fmt"
 	"github.com/gin-gonic/gin"
@@ -10,16 +9,15 @@
 	"github.com/shopspring/decimal"
 	"github.com/xuri/excelize/v2"
 	"gorm.io/gorm"
-	"os"
 	"sort"
 	"strconv"
 	"strings"
 	"time"
+	"wms/conf"
 	"wms/constvar"
 	"wms/extend/code"
 	"wms/extend/util"
 	"wms/models"
-	"wms/opa"
 	"wms/pkg/logx"
 	"wms/pkg/mysqlx"
 	"wms/pkg/structx"
@@ -541,23 +539,23 @@
 
 		}
 
-		if operation.BaseOperationType == constvar.BaseOperationTypeOutgoing {
+		if operation.BaseOperationType == constvar.BaseOperationTypeOutgoing || operation.BaseOperationType == constvar.BaseOperationTypeDisuse {
 			for k, v := range listDetails {
 				//todo 婕旂ず娴嬭瘯鏁版嵁
-				data, err := os.ReadFile("conf/input.json")
-				if err != nil {
-					return errors.New("鏂囦欢璇诲彇澶辫触")
-				}
-				m := make(map[string]interface{})
-				err = json.Unmarshal(data, &m)
-				if err != nil {
-					return errors.New("鏍煎紡杞崲澶辫触")
-				}
-				if opa.OpaCheck(c, m, "operation") {
-					if v.Product.Amount.LessThan(v.Amount) {
-						return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,鍑哄簱锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愬嚭搴撴搷浣�", v.Product.Name, v.Product.Amount.String(), v.Amount.String()))
-					}
-				}
+				//data, err := os.ReadFile("conf/input.json")
+				//if err != nil {
+				//	return errors.New("鏂囦欢璇诲彇澶辫触")
+				//}
+				//m := make(map[string]interface{})
+				//err = json.Unmarshal(data, &m)
+				//if err != nil {
+				//	return errors.New("鏍煎紡杞崲澶辫触")
+				//}
+				//if opa.OpaCheck(c, m, "operation") {
+				//	if v.Product.Amount.LessThan(v.Amount) {
+				//		return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,鍑哄簱锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愬嚭搴撴搷浣�", v.Product.Name, v.Product.Amount.String(), v.Amount.String()))
+				//	}
+				//}
 				//todo ================end===============================
 				if v.Product.Amount.LessThan(v.Amount) {
 					return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,鍑哄簱锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愬嚭搴撴搷浣�", v.Product.Name, v.Product.Amount.String(), v.Amount.String()))
@@ -599,6 +597,27 @@
 					toLocAmount.Amount = toLocAmount.Amount.Add(v.Amount)
 					if err := models.NewLocationProductAmountSearch().SetID(toLocAmount.Id).Update(toLocAmount); err != nil {
 						return err
+					}
+				} else {
+					if err := models.NewLocationProductAmountSearch().Create(&models.LocationProductAmount{
+						LocationId:        v.ToLocationID,
+						ProductCategoryID: v.Product.CategoryId,
+						ProductId:         v.ProductId,
+						Amount:            v.Amount,
+						CreateDate:        time.Now().Format("2006-01-02 15:04:05"),
+					}); err != nil {
+						return err
+					}
+				}
+			}
+		}
+
+		if operation.BaseOperationType == constvar.BaseOperationTypeAdjust {
+			for _, v := range listDetails {
+				if locAmount, aok := mapLocAmount[strconv.Itoa(v.ToLocationID)+v.ProductId]; aok {
+					locAmount.Amount = v.Amount
+					if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
+						return res.Error
 					}
 				} else {
 					if err := models.NewLocationProductAmountSearch().Create(&models.LocationProductAmount{
@@ -870,24 +889,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) {
@@ -1106,7 +1250,7 @@
 		Select("wms_operation.id as operation_id,wms_operation.number,wms_operation.base_operation_type,material.id AS product_id," +
 			"material.`name` AS product_name,wms_operation_details.amount,material.unit,wms_operation_details.from_location_id," +
 			"from_location.`name` AS from_location,wms_operation_details.to_location_id,to_location.`name` AS to_location," +
-			"wms_operation.operation_date as date,wms_operation.`status`,material.weight").
+			"wms_operation.operation_date as date,wms_operation.`status`,material.weight,wms_operation.operation_type_name").
 		InnerJoins("inner join wms_operation_details ON wms_operation_details.operation_id = wms_operation.id").
 		InnerJoins("INNER JOIN material ON material.id = wms_operation_details.product_id").
 		InnerJoins("INNER JOIN wms_location AS from_location ON from_location.id = wms_operation_details.from_location_id").

--
Gitblit v1.8.0