第一次提交

excel/index.go

@@ -0,0 +1,141 @@
+package excel
+
+import (
+	"fmt"
+	"log"
+	"path/filepath"
+
+	"github.com/gogf/gf/v2/container/gmap"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gfile"
+	"github.com/xuri/excelize/v2"
+)
+
+func CreateExcel(sheetName string, headers, cols []string, drops []DropdownCol, fileName string, centerCol []string, creatCol func(f *excelize.File, cols []string) *excelize.File) string {
+	f := excelize.NewFile()
+	defer func() {
+		if err := f.Close(); err != nil {
+			panic(fmt.Sprintf("关闭Excel文件错误:%s", err))
+		}
+	}()
+	index, err := f.NewSheet(sheetName)
+	if err != nil {
+		panic(fmt.Sprintf("创建工作表错误:%s", err))
+	}
+	f.SetActiveSheet(index)
+	_ = f.SetRowHeight(sheetName, 0, 150)
+	for i, v := range headers {
+		_ = f.SetCellValue("sheet1", cols[i]+"1", v)
+	}
+	f = creatCol(f, cols)
+	if !g.IsEmpty(drops) {
+		ExcelDownCol(f, drops)
+	}
+	if !g.IsEmpty(centerCol) {
+		for _, col := range centerCol {
+			_ = f.SetColStyle("sheet1", col, CenterCol(f))
+		}
+	}
+	basePath := filepath.Join("excel", fileName)
+	fPath := filepath.Join(gfile.Pwd(), "resource", basePath)
+	if err := f.SaveAs(fPath); err != nil {
+		panic(fmt.Sprintf("导出房源Excel模板文件错误:%s", err))
+	}
+	return "/uploads/" + basePath
+}
+
+// ExcelDownCol 导出模板列下拉数据
+// 添加下拉框和注释
+// @param f *excelize.File
+// @param dropdownCols []model.DropdownCol
+// @return 返回文件
+func ExcelDownCol(f *excelize.File, dropdownCols []DropdownCol) *excelize.File {
+	// 为导入模板的列添加下拉数据
+	for _, dropdown := range dropdownCols {
+		if dropdown.EndRow < dropdown.StartRow {
+			// 默认设置1000行下拉框
+			dropdown.EndRow = dropdown.StartRow + 999
+		}
+
+		// 构建范围字符串 (如 "B2:B1000")
+		dvRange := fmt.Sprintf(
+			"%s%d:%s%d",
+			dropdown.Column,
+			dropdown.StartRow,
+			dropdown.Column,
+			dropdown.EndRow,
+		)
+
+		// 创建数据验证对象
+		dv := excelize.NewDataValidation(true) // 允许空值
+		dv.SetSqref(dvRange)                   // 设置应用范围
+
+		// 设置下拉列表选项
+		if err := dv.SetDropList(dropdown.Options); err != nil {
+			fmt.Errorf("设置下拉选项失败: %w", err)
+			continue
+		}
+
+		// 添加数据验证到工作表
+		if err := f.AddDataValidation("sheet1", dv); err != nil {
+			fmt.Errorf("添加下拉框失败: %w", err)
+			continue
+		}
+
+		// 添加注释提示（使用新版API）
+		commentCell := fmt.Sprintf("%s1", dropdown.Column)
+		if err := f.AddComment("sheet1", excelize.Comment{
+			Cell:   commentCell,
+			Author: "系统提示",
+			Paragraph: []excelize.RichTextRun{
+				{Text: "请从下拉列表中选择:\n"},
+				{Text: JoinOptions(dropdown.Options)},
+			},
+		}); err != nil {
+			log.Printf("添加注释失败: %v", err)
+			continue
+		}
+	}
+	return f
+}
+
+// JoinOptions 将选项列表转换为逗号分隔的字符串（用于注释）
+func JoinOptions(options []string) string {
+	result := ""
+	for i, opt := range options {
+		if i > 0 {
+			result += ", "
+		}
+		result += opt
+	}
+	return result
+}
+
+// TransMap 闯入MAP 并根据isCn查询key 返回value
+func TransMap(maps map[string]string, key string, isCn bool) string {
+	tagMap := gmap.NewStrStrMap()
+	tagMap.Sets(maps)
+	if !isCn {
+		tagMap.Flip()
+	}
+	val, ok := tagMap.Search(key)
+	if ok {
+		return val
+	} else {
+		return ""
+	}
+}
+
+func CenterCol(f *excelize.File) int {
+	style, err := f.NewStyle(&excelize.Style{
+		Alignment: &excelize.Alignment{
+			Horizontal: "center",
+			Vertical:   "center",
+		},
+	})
+	if err != nil {
+		fmt.Println("创建样式失败:", err)
+		return 0
+	}
+	return style
+}

excel/model.go

@@ -0,0 +1,8 @@
+package excel
+
+type DropdownCol struct {
+	Column   string   // 需要下拉框的列标识（如B, C）
+	Options  []string // 下拉选项列表
+	StartRow int      // 下拉框起始行（从1开始）
+	EndRow   int      // 下拉框结束行
+}