[cleanup] Add more ruff rules (#10149)

[yt-dlp.git] / yt_dlp / extractor / ku6.py

from .common import InfoExtractor


class Ku6IE(InfoExtractor):
    _VALID_URL = r'https?://v\.ku6\.com/show/(?P<id>[a-zA-Z0-9\-\_]+)(?:\.)*html'
    _TEST = {
        'url': 'http://v.ku6.com/show/JG-8yS14xzBr4bCn1pu0xw...html',
        'md5': '01203549b9efbb45f4b87d55bdea1ed1',
        'info_dict': {
            'id': 'JG-8yS14xzBr4bCn1pu0xw',
            'ext': 'f4v',
            'title': 'techniques test',
        },
    }

    def _real_extract(self, url):
        video_id = self._match_id(url)
        webpage = self._download_webpage(url, video_id)

        title = self._html_search_regex(
            r'<h1 title=.*>(.*?)</h1>', webpage, 'title')
        data_url = f'http://v.ku6.com/fetchVideo4Player/{video_id}.html'
        json_data = self._download_json(data_url, video_id)
        download_url = json_data['data']['f']

        return {
            'id': video_id,
            'title': title,
            'url': download_url,
        }