aboutsummaryrefslogtreecommitdiff
path: root/youtube_dl/extractor/douyutv.py
blob: e9b92eb3b291cd610bd168058ec606d96cbca308 (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
# coding: utf-8
from __future__ import unicode_literals

from .common import InfoExtractor

from ..utils import (
    ExtractorError,
)

class DouyutvIE(InfoExtractor):
    _VALID_URL = r'http://(?:www\.)?douyutv\.com/(?P<id>[A-Za-z0-9]+)'

    '''
    show_status: 1 直播中 ,2 没有直播
    '''

    _TEST = {
        'url': 'http://www.douyutv.com/iseven',
        'info_dict': {
            'id': 'iseven',
            'title': '清晨醒脑!T-ara根本停不下来!',
            'ext': 'flv',
            'thumbnail': 're:^https?://.*\.jpg$',
            'is_live': True,
        }
    }

    def _real_extract(self, url):
        video_id = self._match_id(url)
        info_url = 'http://www.douyutv.com/api/client/room/' + video_id

        config = self._download_json(info_url, video_id)

        error_code = config.get('error')
        show_status = config['data'].get('show_status')
        if error_code is not 0:
            raise ExtractorError('Server reported error %i' % error_code,
                                 expected=True)

        if show_status == '2':
            raise ExtractorError('The live show has not yet started',
                                 expected=True)

        title = config['data'].get('room_name')
        rtmp_url = config['data'].get('rtmp_url')
        rtmp_live = config['data'].get('rtmp_live')
        thumbnail = config['data'].get('room_src') 

        url = rtmp_url+'/'+rtmp_live

        return {
            'id': video_id,
            'title': title,
            'ext':'flv',
            'url': url,
            'thumbnail': thumbnail,
            'is_live': True,
            # TODO more properties (see youtube_dl/extractor/common.py)
        }