From 51b0152c2a9264a496cf88d77425768bdaa039a3 Mon Sep 17 00:00:00 2001
From: trigg <triggerhapp@gmail.com>
Date: Wed, 21 Oct 2020 11:09:56 +0000
Subject: [PATCH] - Don't attempt to load non-image attachments

---
 discover_overlay/text_overlay.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/discover_overlay/text_overlay.py b/discover_overlay/text_overlay.py
index 99a337b..6cc6e62 100644
--- a/discover_overlay/text_overlay.py
+++ b/discover_overlay/text_overlay.py
@@ -198,12 +198,18 @@ class TextOverlayWindow(OverlayWindow):
             if line['attach'] and self.show_attach:
                 attachment = line['attach'][0]
                 url = attachment['url']
-                if url in self.attachment:
-                    current_y = self.draw_attach(current_y, url)
+                extension = attachment['filename']
+                extension = extension.rsplit(".", 1)[1]
+                extension = extension.lower()
+                if extension in ['jpeg', 'jpg', 'png', 'gif']:
+                    if url in self.attachment:
+                        current_y = self.draw_attach(current_y, url)
+                    else:
+                        get_surface(self.recv_attach,
+                                    url,
+                                    url, None)
                 else:
-                    get_surface(self.recv_attach,
-                                url,
-                                url, None)
+                    logging.warning("Unknown file extension '%s'", extension)
                 # cy = self.draw_text(cy, "%s" % (line['attach']))
             message = "<span foreground='%s'>%s</span>: %s" % (self.sanitize_string(col),
                                                                self.sanitize_string(