Different images for different URL protocols

boards/models/attachment/viewers.py

@@ -3,7 +3,7 b' from django.template.defaultfilters impo'
 from django.contrib.staticfiles.templatetags.staticfiles import static
 
 FILE_STUB_IMAGE = 'images/file.png'
-FILE_STUB_URL = 'images/url.png'
+FILE_STUB_URL = 'url'
 
 FILE_TYPES_VIDEO = (
     'webm',
@@ -33,6 +33,10 b' PLAIN_FILE_FORMATS = {'
     'xcf': 'xcf',
 }
 
+URL_PROTOCOLS = {
+    'magnet': 'magnet',
+}
+
 CSS_CLASS_IMAGE = 'image'
 CSS_CLASS_THUMB = 'thumb'
 
@@ -143,7 +147,12 b' class UrlViewer(AbstractViewer):'
         return '<div class="image">' \
                '{}' \
                '</div>'.format(self.get_format_view())
+
     def get_format_view(self):
+        protocol = self.url.split('://')[0]
+        url_image_name = URL_PROTOCOLS.get(protocol, FILE_STUB_URL)
+        image = static('images/' + url_image_name + '.png')
+
         return '<a href="{}">' \
                '<img class="url-image" src="{}" width="200" height="150"/>' \
-               '</a>'.format(self.url, static(FILE_STUB_URL))
+               '</a>'.format(self.url, image)