fix: add friend request handling

2 files changed, 4 insertions, 2 deletions

diff --git a/ATRI/plugins/essential/__init__.py b/ATRI/plugins/essential/__init__.py
index 4635583..020653e 100644
--- a/ATRI/plugins/essential/__init__.py
+++ b/ATRI/plugins/essential/__init__.py
@@ -45,7 +45,7 @@ friend_add_event = plugin.on_request("好友申请", "好友申请检测")
 
 
 @friend_add_event.handle()
-async def _(event: FriendRequestEvent):
+async def _(bot: Bot, event: FriendRequestEvent):
     path = __ESSENTIAL_DIR / "friend_add.json"
     file = FileDealer(path)
     if not path.is_file():
@@ -62,7 +62,7 @@ async def _(event: FriendRequestEvent):
         user_id=user_id,
         comment=apply_comment,
         time=now_time,
-    )
+    ).dict()
     await file.write(json.dumps(data))
 
     result = (
diff --git a/ATRI/plugins/manage/__init__.py b/ATRI/plugins/manage/__init__.py
index 0268ec9..03de2fe 100644
--- a/ATRI/plugins/manage/__init__.py
+++ b/ATRI/plugins/manage/__init__.py
@@ -290,6 +290,8 @@ async def _deal_approve_friend_add(
     except Exception:
         await approve_friend_add.finish("同意失败...尝试下手动？")
     data = Manage().load_friend_apply_list()
+    if apply_code not in data:
+        await approve_friend_add.reject("申请码不存在...请检查是否输入正确")
     data.pop(apply_code)
     Manage().save_friend_apply_list(data)
     await approve_friend_add.finish("好欸！申请已通过！")