From 151a2b4a2a037967ffd3126a48f432a7fbf3e260 Mon Sep 17 00:00:00 2001
From: Nichole Mattera <697668+NicholeMattera@users.noreply.github.com>
Date: Tue, 5 May 2020 20:39:31 -0400
Subject: [PATCH] Changed how file extensions are checked.

---
 cogs/lists.py | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/cogs/lists.py b/cogs/lists.py
index 5c174f1..3444e07 100644
--- a/cogs/lists.py
+++ b/cogs/lists.py
@@ -2,6 +2,7 @@ import config
 import discord
 import io
 import urllib.parse
+import os.path
 from discord.ext import commands
 from discord.ext.commands import Cog
 
@@ -101,9 +102,7 @@ class Lists(Cog):
                 (
                     a
                     for a in message.attachments
-                    if a.filename.endswith(".png")
-                    or a.filename.endswith(".jpg")
-                    or a.filename.endswith(".jpeg")
+                    if os.path.splitext(a.filename)[1] in [".png", ".jpg", ".jpeg"]
                 ),
                 None,
             )
@@ -265,9 +264,7 @@ class Lists(Cog):
                 (
                     a
                     for a in message.attachments
-                    if a.filename.endswith(".png")
-                    or a.filename.endswith(".jpg")
-                    or a.filename.endswith(".jpeg")
+                    if os.path.splitext(a.filename)[1] in [".png", ".jpg", ".jpeg"]
                 ),
                 None,
             )