45 lines
1.0 KiB
Go
45 lines
1.0 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/Jinnrry/pmail/config"
|
||
"github.com/Jinnrry/pmail/db"
|
||
"github.com/Jinnrry/pmail/hooks/spam_block/tools"
|
||
"github.com/Jinnrry/pmail/models"
|
||
"os"
|
||
)
|
||
|
||
func main() {
|
||
config.Init()
|
||
err := db.Init("test")
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
fmt.Println(config.Instance.DbDSN)
|
||
|
||
fmt.Println("文件第一列是分类，0表示正常邮件，1表示垃圾邮件，2表示诈骗邮件")
|
||
|
||
var start int
|
||
file, err := os.OpenFile("data.csv", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0777)
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
}
|
||
defer file.Close()
|
||
for {
|
||
var emails []models.Email
|
||
db.Instance.Table(&models.Email{}).Where("id > ?", start).OrderBy("id").Find(&emails)
|
||
if len(emails) == 0 {
|
||
break
|
||
}
|
||
for _, email := range emails {
|
||
start = email.Id
|
||
_, err = file.WriteString(fmt.Sprintf("0 \t%s %s\n", email.Subject, tools.Trim(tools.TrimHtml(email.Html.String))))
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
}
|
||
//fmt.Printf("0 \t%s %s\n", email.Subject, trim(trimHtml(email.Html.String)))
|
||
}
|
||
}
|
||
|
||
}
|