[NotSoBot] Don't try to look through guild for user names when in DM's

TrustyJAID · Mar 9, 2024 · a79d38b · a79d38b
1 parent 734e9f1
commit a79d38b
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/notsobot/converter.py b/notsobot/converter.py
@@ -67,16 +67,17 @@ async def convert(
             for attachment in attachments:
                 urls.append(attachment.url)
         if not urls:
-            for m in ctx.guild.members:
-                if argument.lower() in unidecode.unidecode(m.display_name.lower()):
-                    # display_name so we can get the nick of the user first
-                    # without being NoneType and then check username if that matches
-                    # what we're expecting
-                    urls.append(m.display_avatar.replace(format="png").url)
-                    continue
-                if argument.lower() in unidecode.unidecode(m.name.lower()):
-                    urls.append(m.display_avatar.replace(format="png").url)
-                    continue
+            if ctx.guild:
+                for m in ctx.guild.members:
+                    if argument.lower() in unidecode.unidecode(m.display_name.lower()):
+                        # display_name so we can get the nick of the user first
+                        # without being NoneType and then check username if that matches
+                        # what we're expecting
+                        urls.append(m.display_avatar.replace(format="png").url)
+                        continue
+                    if argument.lower() in unidecode.unidecode(m.name.lower()):
+                        urls.append(m.display_avatar.replace(format="png").url)
+                        continue
 
         if not urls:
             raise BadArgument("No images provided.")