From 5504ea28026280308bcd35a2adc4d28d8a1f6573 Mon Sep 17 00:00:00 2001 From: Rene Teigen Date: Wed, 11 Jan 2023 12:49:07 +0000 Subject: [PATCH] Fix cases where the user doesn't exist in the guild Extra: add sqlite files to gitignore --- .gitignore | 1 + cogs/gpt_3_commands_and_converser.py | 12 ++++++++---- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/.gitignore b/.gitignore index 4244f47..a95b25f 100644 --- a/.gitignore +++ b/.gitignore @@ -5,6 +5,7 @@ __pycache__ #user files .env .vscode +*.sqlite bot.pid usage.txt /dalleimages \ No newline at end of file diff --git a/cogs/gpt_3_commands_and_converser.py b/cogs/gpt_3_commands_and_converser.py index a3c1086..4f0d3b7 100644 --- a/cogs/gpt_3_commands_and_converser.py +++ b/cogs/gpt_3_commands_and_converser.py @@ -790,10 +790,14 @@ class GPT3ComCon(discord.Cog, name="GPT3ComCon"): async def replace_mention(self, ctx, message): if discord.utils.raw_mentions(message): - for mention in discord.utils.raw_mentions(message): - user = await discord.utils.get_or_fetch(ctx.guild, 'member', mention, default="User") - message = message.replace(f"<@{str(mention)}>", user.display_name) - return message + for mention in discord.utils.raw_mentions(message): + try: + user = await discord.utils.get_or_fetch(ctx.guild, 'member', mention) + username = user.display_name + except: + username = str(mention) + message = message.replace(f"<@{str(mention)}>", username) + return message else: return message