111 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			111 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
# coding: utf-8
 | 
						|
from __future__ import unicode_literals
 | 
						|
 | 
						|
from .common import InfoExtractor
 | 
						|
from ..utils import (
 | 
						|
    int_or_none,
 | 
						|
    float_or_none,
 | 
						|
    qualities,
 | 
						|
    ExtractorError,
 | 
						|
)
 | 
						|
 | 
						|
 | 
						|
class GfycatIE(InfoExtractor):
 | 
						|
    _VALID_URL = r'https?://(?:www\.)?gfycat\.com/(?:ifr/)?(?P<id>[^/?#]+)'
 | 
						|
    _TESTS = [{
 | 
						|
        'url': 'http://gfycat.com/DeadlyDecisiveGermanpinscher',
 | 
						|
        'info_dict': {
 | 
						|
            'id': 'DeadlyDecisiveGermanpinscher',
 | 
						|
            'ext': 'mp4',
 | 
						|
            'title': 'Ghost in the Shell',
 | 
						|
            'timestamp': 1410656006,
 | 
						|
            'upload_date': '20140914',
 | 
						|
            'uploader': 'anonymous',
 | 
						|
            'duration': 10.4,
 | 
						|
            'view_count': int,
 | 
						|
            'like_count': int,
 | 
						|
            'dislike_count': int,
 | 
						|
            'categories': list,
 | 
						|
            'age_limit': 0,
 | 
						|
        }
 | 
						|
    }, {
 | 
						|
        'url': 'http://gfycat.com/ifr/JauntyTimelyAmazontreeboa',
 | 
						|
        'info_dict': {
 | 
						|
            'id': 'JauntyTimelyAmazontreeboa',
 | 
						|
            'ext': 'mp4',
 | 
						|
            'title': 'JauntyTimelyAmazontreeboa',
 | 
						|
            'timestamp': 1411720126,
 | 
						|
            'upload_date': '20140926',
 | 
						|
            'uploader': 'anonymous',
 | 
						|
            'duration': 3.52,
 | 
						|
            'view_count': int,
 | 
						|
            'like_count': int,
 | 
						|
            'dislike_count': int,
 | 
						|
            'categories': list,
 | 
						|
            'age_limit': 0,
 | 
						|
        }
 | 
						|
    }]
 | 
						|
 | 
						|
    def _real_extract(self, url):
 | 
						|
        video_id = self._match_id(url)
 | 
						|
 | 
						|
        gfy = self._download_json(
 | 
						|
            'http://gfycat.com/cajax/get/%s' % video_id,
 | 
						|
            video_id, 'Downloading video info')
 | 
						|
        if 'error' in gfy:
 | 
						|
            raise ExtractorError('Gfycat said: ' + gfy['error'], expected=True)
 | 
						|
        gfy = gfy['gfyItem']
 | 
						|
 | 
						|
        title = gfy.get('title') or gfy['gfyName']
 | 
						|
        description = gfy.get('description')
 | 
						|
        timestamp = int_or_none(gfy.get('createDate'))
 | 
						|
        uploader = gfy.get('userName')
 | 
						|
        view_count = int_or_none(gfy.get('views'))
 | 
						|
        like_count = int_or_none(gfy.get('likes'))
 | 
						|
        dislike_count = int_or_none(gfy.get('dislikes'))
 | 
						|
        age_limit = 18 if gfy.get('nsfw') == '1' else 0
 | 
						|
 | 
						|
        width = int_or_none(gfy.get('width'))
 | 
						|
        height = int_or_none(gfy.get('height'))
 | 
						|
        fps = int_or_none(gfy.get('frameRate'))
 | 
						|
        num_frames = int_or_none(gfy.get('numFrames'))
 | 
						|
 | 
						|
        duration = float_or_none(num_frames, fps) if num_frames and fps else None
 | 
						|
 | 
						|
        categories = gfy.get('tags') or gfy.get('extraLemmas') or []
 | 
						|
 | 
						|
        FORMATS = ('gif', 'webm', 'mp4')
 | 
						|
        quality = qualities(FORMATS)
 | 
						|
 | 
						|
        formats = []
 | 
						|
        for format_id in FORMATS:
 | 
						|
            video_url = gfy.get('%sUrl' % format_id)
 | 
						|
            if not video_url:
 | 
						|
                continue
 | 
						|
            filesize = gfy.get('%sSize' % format_id)
 | 
						|
            formats.append({
 | 
						|
                'url': video_url,
 | 
						|
                'format_id': format_id,
 | 
						|
                'width': width,
 | 
						|
                'height': height,
 | 
						|
                'fps': fps,
 | 
						|
                'filesize': filesize,
 | 
						|
                'quality': quality(format_id),
 | 
						|
            })
 | 
						|
        self._sort_formats(formats)
 | 
						|
 | 
						|
        return {
 | 
						|
            'id': video_id,
 | 
						|
            'title': title,
 | 
						|
            'description': description,
 | 
						|
            'timestamp': timestamp,
 | 
						|
            'uploader': uploader,
 | 
						|
            'duration': duration,
 | 
						|
            'view_count': view_count,
 | 
						|
            'like_count': like_count,
 | 
						|
            'dislike_count': dislike_count,
 | 
						|
            'categories': categories,
 | 
						|
            'age_limit': age_limit,
 | 
						|
            'formats': formats,
 | 
						|
        }
 |