Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2025-12-20 06:20:08 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'master' of https://github.com/FloatTech/ZeroBot-Plugin

Browse Source
This commit is contained in:
fumiama 2022-01-06 12:26:08 +08:00
commit 2dc1fbde96
1 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
plugin_vtb_quotation/model/model.go
View File

@ -115,11 +115,10 @@ func (vdb *VtbDB) GetAllSecondCategoryMessageByFirstIndex(firstIndex int) string
db := (*gorm.DB)(vdb) db := (*gorm.DB)(vdb)
secondStepMessage := "请选择一个语录类别并发送序号:\n" secondStepMessage := "请选择一个语录类别并发送序号:\n"
var scl []SecondCategory var scl []SecondCategory
var count int
var fc FirstCategory var fc FirstCategory
db.Model(FirstCategory{}).Where("first_category_index = ?", firstIndex).First(&fc) db.Model(FirstCategory{}).Where("first_category_index = ?", firstIndex).First(&fc)
err := db.Debug().Model(&SecondCategory{}).Find(&scl, "first_category_uid = ?", fc.FirstCategoryUID).Count(&count).Error err := db.Debug().Model(&SecondCategory{}).Find(&scl, "first_category_uid = ?", fc.FirstCategoryUID).Error
if err != nil || count == 0 { if err != nil || len(scl) == 0 {
log.Errorln("[vtb/model]数据库读取错误", err) log.Errorln("[vtb/model]数据库读取错误", err)
return "" return ""
} }
@ -135,10 +134,9 @@ func (vdb *VtbDB) GetAllThirdCategoryMessageByFirstIndexAndSecondIndex(firstInde
thirdStepMessage := "请选择一个语录并发送序号:\n" thirdStepMessage := "请选择一个语录并发送序号:\n"
var fc FirstCategory var fc FirstCategory
db.Model(FirstCategory{}).Where("first_category_index = ?", firstIndex).First(&fc) db.Model(FirstCategory{}).Where("first_category_index = ?", firstIndex).First(&fc)
var count int
var tcl []ThirdCategory var tcl []ThirdCategory
err := db.Debug().Model(&ThirdCategory{}).Find(&tcl, "first_category_uid = ? and second_category_index = ?", fc.FirstCategoryUID, secondIndex).Count(&count).Error err := db.Debug().Model(&ThirdCategory{}).Find(&tcl, "first_category_uid = ? and second_category_index = ?", fc.FirstCategoryUID, secondIndex).Error
if err != nil || count == 0 { if err != nil || len(tcl) == 0 {
log.Errorln("[vtb/model]数据库读取错误", err) log.Errorln("[vtb/model]数据库读取错误", err)
return "" return ""
} }
@ -172,7 +170,7 @@ func (vdb *VtbDB) RandomVtb() ThirdCategory {
func (vdb *VtbDB) GetFirstCategoryByFirstUID(firstUID string) FirstCategory { func (vdb *VtbDB) GetFirstCategoryByFirstUID(firstUID string) FirstCategory {
db := (*gorm.DB)(vdb) db := (*gorm.DB)(vdb)
var fc FirstCategory var fc FirstCategory
db.Model(FirstCategory{}).Where("first_category_uid = ?", firstUID).Take(&fc) db.Model(FirstCategory{}).Take(&fc, "first_category_uid = ?", firstUID)
return fc return fc
} }
@ -227,7 +225,7 @@ func (vdb *VtbDB) GetVtbList() (uidList []string) {
} }
log.Println(fc) log.Println(fc)
if err := db.Debug().Model(&FirstCategory{}).Where("first_category_uid = ?", fc.FirstCategoryUID).First(&fc).Error; err != nil { if err := db.Debug().Model(&FirstCategory{}).First(&fc, "first_category_uid = ?", fc.FirstCategoryUID).Error; err != nil {
if gorm.IsRecordNotFoundError(err) { if gorm.IsRecordNotFoundError(err) {
db.Debug().Model(&FirstCategory{}).Create(&fc) // newUser not user db.Debug().Model(&FirstCategory{}).Create(&fc) // newUser not user
} }
@ -290,7 +288,7 @@ func (vdb *VtbDB) StoreVtb(uid string) {
FirstCategoryUID: uid, FirstCategoryUID: uid,
} }
if err := db.Debug().Model(&SecondCategory{}).Where("first_category_uid = ? and second_category_index = ?", uid, secondIndex).First(&sc).Error; err != nil { if err := db.Debug().Model(&SecondCategory{}).First(&sc, "first_category_uid = ? and second_category_index = ?", uid, secondIndex).Error; err != nil {
// error handling... // error handling...
if gorm.IsRecordNotFoundError(err) { if gorm.IsRecordNotFoundError(err) {
db.Debug().Model(&SecondCategory{}).Create(&sc) // newUser not user db.Debug().Model(&SecondCategory{}).Create(&sc) // newUser not user
@ -319,8 +317,8 @@ func (vdb *VtbDB) StoreVtb(uid string) {
} }
log.Println(tc) log.Println(tc)
if err := db.Debug().Model(&ThirdCategory{}).Where("first_category_uid = ? and second_category_index = ? and third_category_index = ?", if err := db.Debug().Model(&ThirdCategory{}).First(&tc, "first_category_uid = ? and second_category_index = ? and third_category_index = ?",
uid, secondIndex, thirdIndex).First(&tc).Error; err != nil { uid, secondIndex, thirdIndex).Error; err != nil {
if gorm.IsRecordNotFoundError(err) { if gorm.IsRecordNotFoundError(err) {
db.Debug().Model(&ThirdCategory{}).Create(&tc) // newUser not user db.Debug().Model(&ThirdCategory{}).Create(&tc) // newUser not user
} }