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 = "%s: %s" % (self.sanitize_string(col), self.sanitize_string(