# HG changeset patch
# User neko259 <neko259@gmail.com>
# Date 2015-08-24 19:20:24
# Node ID ba607abe881619815d23bfb5566a64a50e058130
# Parent  f9fccc44f7df799621b9f25edab8ca687c6902a0

Video formats are not limited to webm

diff --git a/boards/models/attachment/viewers.py b/boards/models/attachment/viewers.py
--- a/boards/models/attachment/viewers.py
+++ b/boards/models/attachment/viewers.py
@@ -3,7 +3,10 @@ from django.contrib.staticfiles.template
 
 FILE_STUB_IMAGE = 'images/file.png'
 
-FILE_TYPE_WEBM = 'webm'
+FILE_TYPES_VIDEO = (
+    'webm',
+    'mp4',
+)
 FILE_TYPE_SVG = 'svg'
 FILE_TYPES_AUDIO = (
     'ogg',
@@ -37,10 +40,10 @@ class AbstractViewer:
                '</a>'.format(self.file.url,static(FILE_STUB_IMAGE))
 
 
-class WebmViewer(AbstractViewer):
+class VideoViewer(AbstractViewer):
     @staticmethod
     def supports(file_type):
-        return file_type == FILE_TYPE_WEBM
+        return file_type in FILE_TYPES_VIDEO
 
     def get_format_view(self):
         return '<video width="200" height="150" controls src="{}"></video>'\