feature: erp mat
parent
265394a690
commit
a03d58eb1a
@ -0,0 +1,28 @@
|
|||||||
|
package cmd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cu-helper/erp-mat/common"
|
||||||
|
"github.com/mizuki1412/go-core-kit/init/initkit"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
)
|
||||||
|
|
||||||
|
//erp剩余材料按类型统计
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
rootCmd.AddCommand(erpMatCmd)
|
||||||
|
defFlagsErp(erpMatCmd)
|
||||||
|
}
|
||||||
|
|
||||||
|
var erpMatCmd = &cobra.Command{
|
||||||
|
Use: "mat",
|
||||||
|
Short: "Generate erp materials remain list",
|
||||||
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
|
initkit.BindFlags(cmd)
|
||||||
|
common.GenExcelFile()
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func defFlagsErp(cmd *cobra.Command) {
|
||||||
|
cmd.Flags().String("folderPath", "", "erp导出文件夹路径")
|
||||||
|
cmd.Flags().String("templatePath", "", "模板路径")
|
||||||
|
}
|
||||||
@ -0,0 +1,101 @@
|
|||||||
|
package common
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/mizuki1412/go-core-kit/class/exception"
|
||||||
|
"github.com/mizuki1412/go-core-kit/library/mathkit"
|
||||||
|
"github.com/mizuki1412/go-core-kit/service/configkit"
|
||||||
|
"github.com/mizuki1412/go-core-kit/service/logkit"
|
||||||
|
"github.com/spf13/cast"
|
||||||
|
"github.com/xuri/excelize/v2"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type summary struct {
|
||||||
|
Quantity float64
|
||||||
|
Amount float64
|
||||||
|
}
|
||||||
|
|
||||||
|
func GenExcelFile() {
|
||||||
|
folderPath := cast.ToString(configkit.Get("folderPath", ""))
|
||||||
|
if folderPath == "" {
|
||||||
|
panic(exception.New("folder path is empty"))
|
||||||
|
}
|
||||||
|
templatePath := cast.ToString(configkit.Get("templatePath", ""))
|
||||||
|
if templatePath == "" {
|
||||||
|
panic(exception.New("templatePath is empty"))
|
||||||
|
}
|
||||||
|
files, err := os.ReadDir(folderPath)
|
||||||
|
if err != nil {
|
||||||
|
panic(exception.New(err.Error()))
|
||||||
|
}
|
||||||
|
f, err := excelize.OpenFile(templatePath)
|
||||||
|
if err != nil {
|
||||||
|
panic(exception.New(err.Error()))
|
||||||
|
}
|
||||||
|
// 存储按列号汇总的数据
|
||||||
|
index := 3
|
||||||
|
for _, file := range files {
|
||||||
|
if !file.IsDir() && strings.HasSuffix(file.Name(), ".xlsx") {
|
||||||
|
filePath := filepath.Join(folderPath, file.Name())
|
||||||
|
logkit.Info("处理文件：" + filePath)
|
||||||
|
summaryMap, projectName := processFile(filePath)
|
||||||
|
for col, s := range summaryMap {
|
||||||
|
f.SetCellValue("汇总表", fmt.Sprintf("A%d", index), projectName)
|
||||||
|
f.SetCellValue("汇总表", col+fmt.Sprintf("%d", index), cast.ToFloat64(mathkit.FloatRound(s.Quantity, 2)))
|
||||||
|
f.SetCellValue("汇总表", col+fmt.Sprintf("%d", index+1), cast.ToFloat64(mathkit.FloatRound(s.Amount, 2)))
|
||||||
|
}
|
||||||
|
index += 2
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Save the file
|
||||||
|
if err := f.SaveAs("/Users/leo/Documents/ERP材料核对/材料汇总(" + time.Now().Format("1月2日") + ").xlsx"); err != nil {
|
||||||
|
panic(exception.New(err.Error()))
|
||||||
|
}
|
||||||
|
logkit.Info("生成结束...")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 处理单个excel
|
||||||
|
func processFile(filePath string) (map[string]*summary, string) {
|
||||||
|
f, err := excelize.OpenFile(filePath)
|
||||||
|
if err != nil {
|
||||||
|
panic(exception.New(err.Error()))
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
sheetList := f.GetSheetList()
|
||||||
|
if len(sheetList) == 0 {
|
||||||
|
panic(exception.New("无Sheet"))
|
||||||
|
}
|
||||||
|
sheet := sheetList[0] //获取第一个Sheet
|
||||||
|
projectName, _ := f.GetCellValue(sheet, "B3")
|
||||||
|
summaryMap := make(map[string]*summary)
|
||||||
|
row := 5
|
||||||
|
for {
|
||||||
|
eCell, err := f.GetCellValue(sheet, fmt.Sprintf("E%d", row))
|
||||||
|
if err != nil {
|
||||||
|
panic(exception.New(err.Error()))
|
||||||
|
}
|
||||||
|
if eCell == "" {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
kCell, _ := f.GetCellValue(sheet, fmt.Sprintf("K%d", row))
|
||||||
|
mCell, _ := f.GetCellValue(sheet, fmt.Sprintf("M%d", row))
|
||||||
|
quantity := cast.ToFloat64(kCell)
|
||||||
|
amount := cast.ToFloat64(mCell)
|
||||||
|
col := getMatType(eCell)
|
||||||
|
if strings.Contains(col, "~") {
|
||||||
|
quantity /= 1000
|
||||||
|
col = strings.ReplaceAll(col, "~", "")
|
||||||
|
}
|
||||||
|
if summaryMap[col] == nil {
|
||||||
|
summaryMap[col] = &summary{}
|
||||||
|
}
|
||||||
|
summaryMap[col].Quantity += quantity
|
||||||
|
summaryMap[col].Amount += amount
|
||||||
|
row++
|
||||||
|
}
|
||||||
|
return summaryMap, projectName
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue