From 76e0c13f891037de589ac060762879aafcfb0fec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=96=B9=E6=9F=B3=E7=85=9C?=
 <101934327+fangliuyu@users.noreply.github.com>
Date: Fri, 23 Sep 2022 11:11:19 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A5=BD=E6=84=9F=E5=BA=A6=E9=9C=80=E5=8F=91?=
 =?UTF-8?q?=E2=80=9C=E9=87=8D=E7=BD=AE=E6=89=80=E6=9C=89=E8=8A=B1=E5=90=8D?=
 =?UTF-8?q?=E5=86=8C=E2=80=9D=E6=89=8D=E8=83=BD=E4=BD=BF=E7=94=A8=20(#423)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* Update function.go

* Update command.go
---
 plugin/qqwife/command.go  |  2 +-
 plugin/qqwife/function.go | 25 ++++++++++++++++++++-----
 2 files changed, 21 insertions(+), 6 deletions(-)

diff --git a/plugin/qqwife/command.go b/plugin/qqwife/command.go
index 2372ab87..d71ed4d6 100644
--- a/plugin/qqwife/command.go
+++ b/plugin/qqwife/command.go
@@ -61,7 +61,7 @@ func init() {
 		DisableOnDefault:  false,
 		PrivateDataFolder: "qqwife",
 		Help: "一群一天一夫一妻制群老婆\n（每天凌晨刷新CP）\n" +
-			"- 娶群友\n- 群老婆列表\n- 允许/禁止自由恋爱\n- 允许/禁止牛头人\n- 设置CD为xx小时    →(默认12小时)\n- 重置花名册\n" +
+			"- 娶群友\n- 群老婆列表\n- 允许/禁止自由恋爱\n- 允许/禁止牛头人\n- 设置CD为xx小时    →(默认12小时)\n- 重置花名册\n- 重置所有花名册(用于清除所有群数据及其设置)\n" +
 			"--------------------------------\n以下指令存在CD,不跨天刷新,前两个受指令开关\n--------------------------------\n" +
 			"- (娶|嫁)@对方QQ\n自由选择对象，自由恋爱(好感度越高成功率越高,保底30%概率)\n" +
 			"- 当[对方Q号|@对方QQ]的小三\n我和你才是真爱，为了你我愿意付出一切(好感度越高成功率越高,保底10%概率)\n" +
diff --git a/plugin/qqwife/function.go b/plugin/qqwife/function.go
index be5eba37..ee0ec9a0 100644
--- a/plugin/qqwife/function.go
+++ b/plugin/qqwife/function.go
@@ -180,19 +180,34 @@ func (sql *婚姻登记) 清理花名册(gid string) error {
 		return err
 	}
 	if gid != "0" {
-		grouplist = []string{gid}
+		grouplist = []string{"group" + gid}
 	}
 	for _, gid := range grouplist {
-		err = sql.db.Drop("group" + gid)
-		if err != nil {
+		if gid == "favorability" {
+			continue
+		}
+		err = sql.db.Drop(gid)
+		if err != nil || gid == "updateinfo" {
 			continue
 		}
 		gidint, _ := strconv.ParseInt(gid, 10, 64)
-		updateinfo := updateinfo{
+		upinfo := updateinfo{
 			GID:        gidint,
 			Updatetime: time.Now().Format("2006/01/02"),
+			CanMatch:   1,
+			CanNtr:     1,
+			CDtime:     12,
 		}
-		err = sql.db.Insert("updateinfo", &updateinfo)
+		err = sql.db.Create("updateinfo", &updateinfo{})
+		if err != nil {
+			if err = sql.db.Drop("updateinfo"); err == nil {
+				err = sql.db.Create("updateinfo", &updateinfo{})
+			}
+			if err != nil {
+				return err
+			}
+		}
+		err = sql.db.Insert("updateinfo", &upinfo)
 	}
 	return err
 }