🎨 群欢迎迁移优化

This commit is contained in:
HibiKier 2025-01-24 17:17:49 +08:00
parent 91f97133e2
commit e992a85ab7
3 changed files with 14 additions and 7 deletions

View File

@ -54,6 +54,8 @@ def migrate(path: Path):
path: 路径 path: 路径
""" """
text_file = path / "text.json" text_file = path / "text.json"
if not text_file.exists():
return
with text_file.open(encoding="utf8") as f: with text_file.open(encoding="utf8") as f:
json_data = json.load(f) json_data = json.load(f)
new_data = {} new_data = {}

View File

@ -116,19 +116,19 @@ async def _(
session: Uninfo, session: Uninfo,
event: GroupIncreaseNoticeEvent | GroupMemberIncreaseEvent, event: GroupIncreaseNoticeEvent | GroupMemberIncreaseEvent,
): ):
user_id = str(event.user_id) if session.user.id == bot.self_id:
group_id = str(event.group_id)
if user_id == bot.self_id:
"""新成员为bot本身""" """新成员为bot本身"""
group, _ = await GroupConsole.get_or_create( group, _ = await GroupConsole.get_or_create(
group_id=group_id, channel_id__isnull=True group_id=str(event.group_id), channel_id__isnull=True
) )
try: try:
await GroupManager.add_bot(bot, str(event.operator_id), group_id, group) await GroupManager.add_bot(
bot, str(event.operator_id), str(event.group_id), group
)
except ForceAddGroupError as e: except ForceAddGroupError as e:
await PlatformUtils.send_superuser(bot, e.get_info()) await PlatformUtils.send_superuser(bot, e.get_info())
else: else:
await GroupManager.add_user(session, bot, user_id, group_id) await GroupManager.add_user(session, bot)
@group_decrease_handle.handle() @group_decrease_handle.handle()

View File

@ -234,7 +234,12 @@ class GroupManager:
bot: Bot bot: Bot
""" """
user_id = session.user.id user_id = session.user.id
group_id = session.group.id if session.group else "" group_id = ""
if session.group:
if session.group.parent:
group_id = session.group.parent.id
else:
group_id = session.group.id
join_time = datetime.now() join_time = datetime.now()
try: try:
user_info = await bot.get_group_member_info( user_info = await bot.get_group_member_info(