From 8aba97ec5c4d9dd001e01427defd9bbd6bdb47cf Mon Sep 17 00:00:00 2001
From: Ruoqi Yang <martinyangrq@outlook.com>
Date: Sat, 12 Nov 2022 18:56:50 +0800
Subject: fix: add friend request handling

---
 ATRI/plugins/manage/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'ATRI/plugins/manage')

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("好欸！申请已通过！")
-- 
cgit v1.2.3