feat：增加提取title和校验 title
This commit is contained in:
parent
f86192f970
commit
93bb0c62f4
@ -118,3 +118,60 @@ func ConvertReportName(prefix, personName, reportType, locale string) string {
|
||||
return fmt.Sprintf("%s %s %s", prefix, pinyinName, translatedReportType)
|
||||
}
|
||||
|
||||
// isValidPrefix 校验 prefix 是否为 YYYY-MM 格式（年份和月份都是数字）
|
||||
func isValidPrefix(prefix string) bool {
|
||||
if len(prefix) != 7 || prefix[4] != '-' {
|
||||
return false
|
||||
}
|
||||
for i := 0; i < 4; i++ {
|
||||
if prefix[i] < '0' || prefix[i] > '9' {
|
||||
return false
|
||||
}
|
||||
}
|
||||
if prefix[5] < '0' || prefix[5] > '9' || prefix[6] < '0' || prefix[6] > '9' {
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
// isChineseName 校验姓名是否仅包含中文字符（不包含数字和英文字母）
|
||||
func isChineseName(name string) bool {
|
||||
for _, r := range name {
|
||||
if (r >= '0' && r <= '9') || (r >= 'a' && r <= 'z') || (r >= 'A' && r <= 'Z') {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
// ConvertTitleByLocale 将报告标题根据 locale 进行翻译转换。
|
||||
// title: 原始中文标题，格式为 "YYYY-MM" + 姓名 + "数据报告"/"竞品报告"，如 "2026-03王旗老师数据报告"
|
||||
// locale: "zh-CN", "zh-TW", "EN", "de-DE", "ja-JP" 或其他 Accept-Language 值
|
||||
// 返回翻译后的标题，如果格式不符则返回原标题。
|
||||
func ConvertTitleByLocale(title, locale string) string {
|
||||
if len(title) < 7 {
|
||||
return title
|
||||
}
|
||||
|
||||
prefix := title[:7] // YYYY-MM
|
||||
if !isValidPrefix(prefix) {
|
||||
return title
|
||||
}
|
||||
|
||||
var reportType string
|
||||
if strings.HasSuffix(title, "数据报告") {
|
||||
reportType = "数据报告"
|
||||
} else if strings.HasSuffix(title, "竞品报告") {
|
||||
reportType = "竞品报告"
|
||||
} else {
|
||||
return title
|
||||
}
|
||||
|
||||
personName := title[7 : len(title)-len(reportType)]
|
||||
if !isChineseName(personName) {
|
||||
return title
|
||||
}
|
||||
|
||||
return ConvertReportName(prefix, personName, reportType, locale)
|
||||
}
|
||||
|
||||
|
||||
@ -2,6 +2,170 @@ package utils
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestIsValidPrefix(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
prefix string
|
||||
expected bool
|
||||
}{
|
||||
{name: "valid 2026-03", prefix: "2026-03", expected: true},
|
||||
{name: "valid 2026-01", prefix: "2026-01", expected: true},
|
||||
{name: "valid 1999-12", prefix: "1999-12", expected: true},
|
||||
{name: "invalid too short", prefix: "2026-3", expected: false},
|
||||
{name: "invalid too long", prefix: "2026-003", expected: false},
|
||||
{name: "invalid no dash", prefix: "202603", expected: false},
|
||||
{name: "invalid letters in year", prefix: "ab-03", expected: false},
|
||||
{name: "invalid letters in month", prefix: "2026-ab", expected: false},
|
||||
{name: "invalid month > 12", prefix: "2026-13", expected: true}, // 只校验格式，不校验月份合理性
|
||||
{name: "empty string", prefix: "", expected: false},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
result := isValidPrefix(tt.prefix)
|
||||
if result != tt.expected {
|
||||
t.Errorf("isValidPrefix(%q) = %v, want %v", tt.prefix, result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestIsChineseName(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
input string
|
||||
expected bool
|
||||
}{
|
||||
{name: "pure Chinese name", input: "王旗", expected: true},
|
||||
{name: "Chinese name with title", input: "王旗老师", expected: true},
|
||||
{name: "two char Chinese", input: "王明", expected: true},
|
||||
{name: "three char Chinese", input: "王小明", expected: true},
|
||||
{name: "contains digit", input: "王旗1", expected: false},
|
||||
{name: "contains English lower", input: "wang旗", expected: false},
|
||||
{name: "contains English upper", input: "WANG旗", expected: false},
|
||||
{name: "mixed Chinese and digit", input: "王1明", expected: false},
|
||||
{name: "empty string", input: "", expected: true},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
result := isChineseName(tt.input)
|
||||
if result != tt.expected {
|
||||
t.Errorf("isChineseName(%q) = %v, want %v", tt.input, result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestConvertTitleByLocale(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
title string
|
||||
locale string
|
||||
expected string
|
||||
}{
|
||||
{
|
||||
name: "EN Data Report",
|
||||
title: "2026-03王旗老师数据报告",
|
||||
locale: "EN",
|
||||
expected: "2026-03 Wang Qi Data Report",
|
||||
},
|
||||
{
|
||||
name: "EN Competitor Report",
|
||||
title: "2026-03王旗老师竞品报告",
|
||||
locale: "EN",
|
||||
expected: "2026-03 Wang Qi Competitor Report",
|
||||
},
|
||||
{
|
||||
name: "zh-TW Data Report",
|
||||
title: "2026-03王旗老师数据报告",
|
||||
locale: "zh-TW",
|
||||
expected: "2026-03王旗數據報告",
|
||||
},
|
||||
{
|
||||
name: "zh-TW Competitor Report",
|
||||
title: "2026-03王旗老师竞品报告",
|
||||
locale: "zh-TW",
|
||||
expected: "2026-03王旗競合報告",
|
||||
},
|
||||
{
|
||||
name: "de-DE Data Report",
|
||||
title: "2026-03王旗老师数据报告",
|
||||
locale: "de-DE",
|
||||
expected: "2026-03 Wang Qi Datenbericht",
|
||||
},
|
||||
{
|
||||
name: "de-DE Competitor Report",
|
||||
title: "2026-03王旗老师竞品报告",
|
||||
locale: "de-DE",
|
||||
expected: "2026-03 Wang Qi Wettbewerbsanalyse",
|
||||
},
|
||||
{
|
||||
name: "ja-JP Data Report",
|
||||
title: "2026-03王旗老师数据报告",
|
||||
locale: "ja-JP",
|
||||
expected: "2026-03 Wang Qi データレポート",
|
||||
},
|
||||
{
|
||||
name: "ja-JP Competitor Report",
|
||||
title: "2026-03王旗老师竞品报告",
|
||||
locale: "ja-JP",
|
||||
expected: "2026-03 Wang Qi 競合分析レポート",
|
||||
},
|
||||
{
|
||||
name: "zh-CN returns original",
|
||||
title: "2026-03王旗老师数据报告",
|
||||
locale: "zh-CN",
|
||||
expected: "2026-03王旗老师数据报告",
|
||||
},
|
||||
{
|
||||
name: "invalid prefix returns original",
|
||||
title: "202603王旗老师数据报告",
|
||||
locale: "EN",
|
||||
expected: "202603王旗老师数据报告",
|
||||
},
|
||||
{
|
||||
name: "invalid personName with digit returns original",
|
||||
title: "2026-03王旗1老师数据报告",
|
||||
locale: "EN",
|
||||
expected: "2026-03王旗1老师数据报告",
|
||||
},
|
||||
{
|
||||
name: "invalid personName with English returns original",
|
||||
title: "2026-03wang老师数据报告",
|
||||
locale: "EN",
|
||||
expected: "2026-03wang老师数据报告",
|
||||
},
|
||||
{
|
||||
name: "title too short returns original",
|
||||
title: "2026-03",
|
||||
locale: "EN",
|
||||
expected: "2026-03",
|
||||
},
|
||||
{
|
||||
name: "unknown reportType returns original",
|
||||
title: "2026-03王旗老师其他报告",
|
||||
locale: "EN",
|
||||
expected: "2026-03王旗老师其他报告",
|
||||
},
|
||||
{
|
||||
name: "empty title returns original",
|
||||
title: "",
|
||||
locale: "EN",
|
||||
expected: "",
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
result := ConvertTitleByLocale(tt.title, tt.locale)
|
||||
if result != tt.expected {
|
||||
t.Errorf("ConvertTitleByLocale(%q, %q) = %q, want %q", tt.title, tt.locale, result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestConvertReportName(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
|
||||
Loading…
Reference in New Issue
Block a user