django-magicembed/magicembed/providers.py at 0569e462c005b97e09c106b36413d276dc8319b9 · kronoscode/django-magicembed · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
# -*- coding: utf-8 -*-
import json
import re
import urllib

from urlparse import parse_qs

from django.core.exceptions import ImproperlyConfigured
from django.conf import settings

class Provider(object):

    def __init__(self, url, size=(640, 480)):
        self.url = url
        self.size = size

    def render_video(self):
        '''simply returns html to render embed
        video in a template'''
        raise NotImplementedError

    def render_thumbnail(self, link_to):
        '''Renders the html with a link_to a parameter'''
        raise NotImplementedError


class Youtube(Provider):

    def __init__(self, url, size=(640, 480)):
        super(Youtube, self).__init__(url, size)
        qs = url.split('?')
        self.video_id = parse_qs(qs[1])['v'][0]

    def render_video(self):
        html = '''<iframe title="YouTube video player" width="%d" height="%d" src="https://www.youtube.com/embed/%s" frameborder="0" allowfullscreen></iframe>''' # noqa
        return html % (self.size[0], self.size[1], self.video_id)

    def render_thumbnail(self, link_to='#'):
        return '''http://img.youtube.com/vi/%s/0.jpg''' % (self.video_id)


class Vimeo(Provider):

    def __init__(self, url, size=(640, 480)):
        super(Vimeo, self).__init__(url, size)
        pattern = re.compile('(http|https)://(?:www\.)?vimeo\.com/([0-9]{1,12})')
        self.video_id = pattern.match(url).groups()[1]
        self.api_url = 'http://vimeo.com/api/v2/video/%s.json' % self.video_id

    def render_video(self):
        html = '''<iframe title="Vimeo video player" width="%d" height="%d" src="http://player.vimeo.com/video/%s" frameborder="0" allowfullscreen></iframe>''' # noqa
        return html % (self.size[0], self.size[1], self.video_id)

    def render_thumbnail(self, link_to="#"):
        api_response = json.loads(urllib.urlopen(self.api_url).read())
        return api_response[0]['thumbnail_medium']


class Embedly(Provider):

    def __init__(self, url, size=(640, 480)):
        super(Embedly, self).__init__(url, size)
        key = getattr(settings, "EMBEDLY_KEY", None)
        if key is not None:
            self.api_url = 'http://api.embed.ly/1/oembed?key=%s&url=%s&maxwidth=%s&format=json' % (key, url, size[0])
        else:
            raise ImproperlyConfigured("If you want to use this please set the Embedly api key")

    def render_video(self):
        return self._call_api()['html']

    def render_thumbnail(self):
        return self._call_api()['thumbnail_url']

    def _call_api(self):
        try:
            data = json.loads(urllib.urlopen(self.api_url).read())
        except IOError:
            raise ImproperlyConfigured("Please set the Embedly api key correctly")
        return data


def get_provider(url, size=None):
    '''returns a provider instance according to the url'''
    provider_domain = dict(youtube=Youtube, vimeo=Vimeo)
    for domain, provider in provider_domain.items():
        if domain in url:
            return provider(url, size) if size else provider(url)

    return Embedly(url, size) if size else Embedly(url)