PaiGramTeam/PaiGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PaiGram.git synced 2024-11-22 15:36:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

✨ Add Allow Bind User Invite Bot

Browse Source
This commit is contained in:
洛水居室 2023-01-23 21:34:13 +08:00
parent 28e0890390
commit 7ab92b630c
No known key found for this signature in database
GPG Key ID: C9DE87DA724B88FC
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
core/config.py
View File

@ -21,6 +21,7 @@ dotenv.load_dotenv()
class JoinGroups(str, Enum): class JoinGroups(str, Enum):
NO_ALLOW = "NO_ALLOW" NO_ALLOW = "NO_ALLOW"
ALLOW_AUTH_USER = "ALLOW_AUTH_USER" ALLOW_AUTH_USER = "ALLOW_AUTH_USER"
ALLOW_USER = "ALLOW_USER"
ALLOW_ALL = "ALLOW_ALL" ALLOW_ALL = "ALLOW_ALL"

9
plugins/system/chat_member.py
View File

@ -71,6 +71,15 @@ class ChatMember(Plugin):
logger.error("获取信息出现错误", exc_info=exc) logger.error("获取信息出现错误", exc_info=exc)
else: else:
quit_status = False quit_status = False
elif config.join_groups == JoinGroups.ALLOW_USER:
try:
await self.user_service.get_user_by_id(user.id)
except UserNotFoundError:
logger.warning("用户 %s[%s] 邀请请求被拒绝", user.full_name, user.id)
except Exception as exc:
logger.error("获取信息出现错误", exc_info=exc)
else:
quit_status = False
elif config.join_groups == JoinGroups.ALLOW_ALL: elif config.join_groups == JoinGroups.ALLOW_ALL:
quit_status = False quit_status = False
else: else: