fix(bot): 🐛 DiscordUserID onverter not working with mentions.

This commit is contained in:
Nyeki.py
2024-10-19 02:04:31 -03:00
parent 0b9c4ececa
commit 398fc2cb93
2 changed files with 7 additions and 7 deletions
+1 -1
View File
@@ -49,7 +49,7 @@ class Admin(base.Cog):
for call in calls:
for rs in call.call_rss:
if rs.channel.server_id in servers_count_map:
servers_count_map[rs.channel.server.id] += 1
servers_count_map[rs.channel.server_id] += 1
else:
servers_count_map[rs.channel.server_id] = 0
+6 -6
View File
@@ -48,18 +48,13 @@ class DiscordUserID(converter.Converter):
return int(argument)
if len(argument) > 17 and argument[2:-1].isdecimal():
return int(argument)
return int(argument[2:-1])
if argument.startswith("https://discord.com/users/") or argument.startswith(
"discord://-/users/"
):
return int(argument.split("?", 1)[0].rsplit("/", 1)[1])
user = await models.User.get_or_none(username=argument)
if user is not None:
return user.id
if argument.lower() in ["reply", "."]:
if ctx.message.reference is not None:
ref = ctx.message.reference
@@ -80,4 +75,9 @@ class DiscordUserID(converter.Converter):
return msg.author.id
user = await models.User.get_or_none(username=argument)
if user is not None:
return user.id
raise converter.BadArgument()