From 81fa3aa4dfc22998f618e9541d57003777aafb7c Mon Sep 17 00:00:00 2001
From: UnCLAS-Prommer <gitunclasprommer@gmail.com>
Date: Tue, 22 Jul 2025 22:10:00 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=AF=B9=E5=A4=9A=E6=AC=A1?=
 =?UTF-8?q?=E7=A6=81=E8=A8=80=E7=9A=84=E5=85=BC=E5=AE=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pyproject.toml  |  2 +-
 src/database.py | 22 +++++++++++++++++-----
 2 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index f1acff6..e4abfd0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "MaiBotNapcatAdapter"
-version = "0.4.5"
+version = "0.4.6"
 description = "A MaiBot adapter for Napcat"
 
 [tool.ruff]
diff --git a/src/database.py b/src/database.py
index f9a94d1..af193da 100644
--- a/src/database.py
+++ b/src/database.py
@@ -123,14 +123,26 @@ class DatabaseManager:
         其同时还是简化版的更新方式。
         """
         with Session(self.engine) as session:
-            db_record = DB_BanUser(
-                user_id=ban_record.user_id, group_id=ban_record.group_id, lift_time=ban_record.lift_time
+            # 检查记录是否已存在
+            statement = select(DB_BanUser).where(
+                DB_BanUser.user_id == ban_record.user_id, DB_BanUser.group_id == ban_record.group_id
             )
-            session.add(db_record)
+            existing_record = session.exec(statement).first()
+            if existing_record:
+                # 如果记录已存在，更新 lift_time
+                existing_record.lift_time = ban_record.lift_time
+                session.add(existing_record)
+                logger.debug(f"更新禁言记录: {ban_record}")
+            else:
+                # 如果记录不存在，创建新记录
+                db_record = DB_BanUser(
+                    user_id=ban_record.user_id, group_id=ban_record.group_id, lift_time=ban_record.lift_time
+                )
+                session.add(db_record)
+                logger.debug(f"创建新禁言记录: {ban_record}")
             session.commit()
-            logger.debug(f"创建/更新禁言记录: {ban_record}")
 
-    def delete_ban_record(self, ban_record: BanUser) -> bool:
+    def delete_ban_record(self, ban_record: BanUser):
         """
         删除特定用户在特定群组中的禁言记录。
         一个简化版本的删除方式，防止 update_ban_record 方法的复杂性。