Commit 2437154c authored by Liam Brenner's avatar Liam Brenner
Browse files

Do a type check for jinja2 tag

parent c28cc1bd
from __future__ import absolute_import, unicode_literals from __future__ import absolute_import, unicode_literals
import mimetypes
from jinja2.ext import Extension from jinja2.ext import Extension
from django import template from .models import Video
from django.forms.widgets import flatatt
from django.utils.text import mark_safe
def video(video, attrs):
if not video:
return ''
if attrs: def video(video, attrs='preload controls'):
return video.video_tag(attrs) if type(video) != Video:
else: raise TypeError('Expected type {0}, received {1}.'.format(Video, type(video)))
return video return video.video_tag(attrs)
class WagtailVideosExtension(Extension): class WagtailVideosExtension(Extension):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment