feat: server connect mongo

server/service/cleaner.go

@@ -0,0 +1,84 @@
+package service
+
+import (
+	"fmt"
+	"os/exec"
+	"strings"
+
+	"billai-server/config"
+)
+
+// CleanOptions 清洗选项
+type CleanOptions struct {
+	Year   string // 年份筛选
+	Month  string // 月份筛选
+	Start  string // 起始日期
+	End    string // 结束日期
+	Format string // 输出格式: csv/json
+}
+
+// CleanResult 清洗结果
+type CleanResult struct {
+	BillType string // 检测到的账单类型: alipay/wechat
+	Output   string // Python 脚本输出
+}
+
+// RunCleanScript 执行 Python 清洗脚本
+// inputPath: 输入文件路径
+// outputPath: 输出文件路径
+// opts: 清洗选项
+func RunCleanScript(inputPath, outputPath string, opts *CleanOptions) (*CleanResult, error) {
+	// 构建命令参数
+	cleanScriptAbs := config.ResolvePath(config.Global.CleanScript)
+	args := []string{cleanScriptAbs, inputPath, outputPath}
+
+	if opts != nil {
+		if opts.Year != "" {
+			args = append(args, "--year", opts.Year)
+		}
+		if opts.Month != "" {
+			args = append(args, "--month", opts.Month)
+		}
+		if opts.Start != "" {
+			args = append(args, "--start", opts.Start)
+		}
+		if opts.End != "" {
+			args = append(args, "--end", opts.End)
+		}
+		if opts.Format != "" {
+			args = append(args, "--format", opts.Format)
+		}
+	}
+
+	// 执行 Python 脚本
+	fmt.Printf("🐍 执行清洗脚本...\n")
+	pythonPathAbs := config.ResolvePath(config.Global.PythonPath)
+	cmd := exec.Command(pythonPathAbs, args...)
+	cmd.Dir = config.Global.ProjectRoot
+
+	output, err := cmd.CombinedOutput()
+	outputStr := string(output)
+
+	if err != nil {
+		return nil, fmt.Errorf("清洗脚本执行失败: %w\n输出: %s", err, outputStr)
+	}
+
+	// 从输出中检测账单类型
+	billType := DetectBillTypeFromOutput(outputStr)
+
+	return &CleanResult{
+		BillType: billType,
+		Output:   outputStr,
+	}, nil
+}
+
+// DetectBillTypeFromOutput 从 Python 脚本输出中检测账单类型
+func DetectBillTypeFromOutput(output string) string {
+	if strings.Contains(output, "支付宝") {
+		return "alipay"
+	}
+	if strings.Contains(output, "微信") {
+		return "wechat"
+	}
+	return ""
+}