package service
|
|
import (
|
"errors"
|
"github.com/shopspring/decimal"
|
"github.com/xuri/excelize/v2"
|
"strconv"
|
"wms/models"
|
)
|
|
func CreateMoreUnit(amount decimal.Decimal, units []models.UnitItems) []models.UnitItems {
|
moreValueArr := make([]models.UnitItems, 0, len(units))
|
for _, unitItem := range units {
|
if unitItem.Amount.IsZero() {
|
continue
|
}
|
moreValueArr = append(moreValueArr, models.UnitItems{
|
Amount: amount.Div(unitItem.Amount),
|
Unit: unitItem.Unit,
|
Floating: unitItem.Floating,
|
})
|
}
|
return moreValueArr
|
}
|
|
func FillMoreUnitToExcel(amount decimal.Decimal, units []models.UnitItems, startIndex int, column int, f *excelize.File) {
|
columnStr := strconv.Itoa(column)
|
for _, v := range units {
|
switch v.Unit {
|
case "匹":
|
f.SetCellValue("Sheet1", getColumnAlphabet(startIndex)+columnStr, v.Amount.Mul(amount))
|
case "米":
|
f.SetCellValue("Sheet1", getColumnAlphabet(startIndex+1)+columnStr, v.Amount.Mul(amount))
|
case "重量":
|
f.SetCellValue("Sheet1", getColumnAlphabet(startIndex+2)+columnStr, v.Amount.Mul(amount))
|
}
|
}
|
return
|
}
|
|
func SetExcelHeader(headers []interface{}, f *excelize.File) (err error) {
|
// 设置表头
|
var i int
|
for _, h := range headers {
|
if v, ok := h.(string); ok {
|
err = f.SetCellValue("Sheet1", getColumnAlphabet(i)+"1", v)
|
if err != nil {
|
return
|
}
|
i++
|
} else if childHeaders, ok := h.(map[string][]string); ok {
|
for title, list := range childHeaders {
|
err = f.SetCellValue("Sheet1", getColumnAlphabet(i)+"1", title)
|
if err != nil {
|
return err
|
}
|
err = f.MergeCell("Sheet1", getColumnAlphabet(i)+"1", getColumnAlphabet(i-1+len(list))+"1") // 合并单元格
|
for _, t := range list {
|
err = f.SetCellValue("Sheet1", getColumnAlphabet(i)+"2", t)
|
if err != nil {
|
return err
|
}
|
i++
|
}
|
}
|
} else {
|
return errors.New("unsupported header value")
|
}
|
}
|
return nil
|
}
|
|
func SetHeaderStyle(f *excelize.File) (styleInt int, err error) {
|
// 设置表头样式
|
style := &excelize.Style{
|
Border: nil,
|
Fill: excelize.Fill{
|
Type: "pattern",
|
Pattern: 1,
|
Shading: 0,
|
},
|
Font: &excelize.Font{
|
Bold: true,
|
},
|
Alignment: &excelize.Alignment{
|
Horizontal: "center",
|
},
|
}
|
return f.NewStyle(style)
|
}
|
