feat: 批量识别结果导出CSV

- 新增导出CSV文件功能,包含文本、判定结果、置信度、风险关键词
- 新增复制CSV内容到剪贴板功能
- CSV字段:文本、判定结果、置信度、垃圾概率、正常概率、风险关键词

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
刘正航
2026-04-21 23:31:49 +08:00
parent f7d0601c4e
commit 5279816452
2 changed files with 82 additions and 0 deletions

View File

@@ -79,5 +79,83 @@ Page({
showCancel: false,
confirmText: '关闭'
})
},
generateCSV() {
const items = this.data.items
if (!items.length) return ''
const headers = ['文本', '判定结果', '置信度', '垃圾概率', '正常概率', '风险关键词']
const rows = items.map((item) => {
const prediction = item.prediction === 'spam' ? '垃圾信息' : '正常信息'
const confidence = item.confidence_text || '0%'
const spamProb = this.formatPercent(item.spam_probability, 4)
const hamProb = this.formatPercent(item.ham_probability, 4)
const tokens = (item.reason_tokens || []).map((t) => t.token || t).join('; ')
// CSV 转义:文本中的逗号和换行需要处理
const text = (item.text || '').replace(/"/g, '""')
const tokensEscaped = tokens.replace(/"/g, '""')
return `"${text}","${prediction}","${confidence}","${spamProb}","${hamProb}","${tokensEscaped}"`
})
return [headers.join(','), ...rows].join('\n')
},
exportCSV() {
const items = this.data.items
if (!items.length) {
wx.showToast({ title: '暂无识别结果可导出', icon: 'none' })
return
}
const csvContent = this.generateCSV()
const timestamp = new Date().toISOString().slice(0, 19).replace(/[T:]/g, '-')
const filename = `batch_detect_${timestamp}.csv`
// 写入临时文件
const fs = wx.getFileSystemManager()
const tempPath = `${wx.env.USER_DATA_PATH}/${filename}`
try {
fs.writeFileSync(tempPath, csvContent, 'utf8')
wx.showModal({
title: '导出成功',
content: `CSV文件已生成是否打开查看\n文件名:${filename}`,
confirmText: '打开',
cancelText: '关闭',
success: (res) => {
if (res.confirm) {
wx.openDocument({
filePath: tempPath,
fileType: 'csv',
showMenu: true,
fail: (err) => {
console.error('打开文件失败', err)
wx.showToast({ title: '打开失败,请检查文件管理器', icon: 'none' })
}
})
}
}
})
} catch (err) {
console.error('写入文件失败', err)
wx.showToast({ title: '导出失败', icon: 'none' })
}
},
copyCSVToClipboard() {
const items = this.data.items
if (!items.length) {
wx.showToast({ title: '暂无识别结果可复制', icon: 'none' })
return
}
const csvContent = this.generateCSV()
wx.setClipboardData({
data: csvContent,
success: () => {
wx.showToast({ title: 'CSV内容已复制到剪贴板', icon: 'success' })
}
})
}
})