aboutsummaryrefslogtreecommitdiffstats
path: root/youtube_dl/extractor/gfycat.py
blob: 6de78c49d18a8bac1b9071360e503cf2e8eb30c8 (plain)
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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
# coding: utf-8

from __future__ import unicode_literals

import datetime

from .common import InfoExtractor

class GfycatIE(InfoExtractor):
    _VALID_URL = r'https?://(?:www\.)?gfycat\.com/(?P<id>[^/?#]+)'
    _TESTS = [
        {
            'url': 'http://gfycat.com/DeadlyDecisiveGermanpinscher',
            'info_dict': {
                'id':          'DeadlyDecisiveGermanpinscher',
                'title':       'Ghost in the Shell',
                'ext':         'mp4',
                'upload_date': '20140913'
            }
        },{
            'url': 'http://gfycat.com/pleasinghilariouskusimanse',
            'info_dict': {
                'id':          'pleasinghilariouskusimanse',
                'title':       'PleasingHilariousKusimanse',
                'ext':         'webm',
                'upload_date': '20150412'
            },
            'params': {
                'format': 'webm',
            },
        },{
            'url': 'http://gfycat.com/requiredunkemptbuzzard',
            'info_dict': {
                'id':          'requiredunkemptbuzzard',
                'title':       'Headshot!',
                'ext':         'gif',
                'upload_date': '20150129'
            },
            'params': {
                'format': 'gif',
            },
        },
    ]

    def _real_extract(self, url):
        video_id = self._match_id(url)
        json     = self._download_json("http://gfycat.com/cajax/get/" + video_id, video_id, 'Downloading video info')['gfyItem']
        
        # Title
        # Use user title first, else fallback to url formated name
        if json['title']:
            video_title = json['title']
        else:
            video_title = json['gfyName']
        
        # Formats
        # Pref: mp4, webm, gif
        formats = [{
            'format_id':  'mp4',
            'ext':        'mp4',
            'url':        json['mp4Url'],
            'width':      json['width'],
            'height':     json['height'],
            'fps':        json['frameRate'],
            'filesize':   json['mp4Size'],
            'preference': '-1'
        }, {
            'format_id': 'webm',
            'ext':       'webm',
            'url':        json['webmUrl'],
            'width':      json['width'],
            'height':     json['height'],
            'fps':        json['frameRate'],
            'filesize':   json['webmSize'],
            'preference': 0
        }, {
            'format_id':  'gif',
            'ext':        'gif',
            'url':        json['gifUrl'],
            'width':      json['width'],
            'height':     json['height'],
            'fps':        json['frameRate'],
            'filesize':   json['gifSize'],
            'preference': 1
        }]
        
        self._sort_formats(formats)
        
        # Date
        date = datetime.datetime.fromtimestamp(
            int(json['createDate'])
        ).strftime('%Y%m%d')
        
        # Length
        duration = json['numFrames'] / json['frameRate']
        
        # Age limit
        # 1 = nsfw / 0 = sfw
        if json['nsfw'] == 1:
            age_limit = 18
        else:
            age_limit = 0
        
        return {
            'id':          video_id,
            'title':       video_title,
            'formats':     formats,
            'creator':     json['userName'],
            'description': json['description'],
            'upload_date': date,
            'categories':  json['tags'],
            'age_limit':   age_limit,
            'duration':    duration,
            'view_count':  json['views']
        }