Repository: yt-dlp/yt-dlp
File: yt_dlp/extractor/zaiko.py (Python)

Code before:
import json
import html
import re

from .common import InfoExtractor
from ..utils import traverse_obj


class ZaikoIE(InfoExtractor):
    _VALID_URL = r'https://zaiko\.io/event/(?P<id>\d+)/stream(?:/\d+)+'
    _TESTS = [{
        'url': 'https://zaiko.io/event/324868/stream/20571/20571',
        'info_dict': {
            'id': 324868,
            'ext': 'mp4',
            'title': 'ZAIKO STREAMING TEST',
            'uploader_id': 454,
            'timestamp': 1583809200,
        }
    }]

    def _parse_vue_element_attr(self, name, string):
        page_elem = self._search_regex(r'<' + name + '([^>]+)>', string, name)
        attr = {}
        for m in re.finditer(r'(:?[\w\-_]+)="([^"]+)"', page_elem):
            key, value = m.groups()
            if key[0] == ':':
                key = key[1:]
                value = json.loads(html.unescape(value))
            attr[key] = value
        return attr

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

        webpage = self._download_webpage(url, video_id)
        stream_meta = self._parse_vue_element_attr('stream-page', webpage)

        headers = {'referer': 'https://zaiko.io/'}
        player_page = self._download_webpage(stream_meta['stream-access']['video_source'], video_id, headers=headers)
        player_meta = self._parse_vue_element_attr('player', player_page)

        return {

Review comment (bug):
make sure to actually run this test, right now it fails without an `upload_date` value

use `test/local_parameters.json` to pass cookies params for the test download, e.g.:
```json
{
  "cookiesfrombrowser": [
    "firefox",
    null,
    null,
    null
  ]
}
```

Code after:
from .common import InfoExtractor
from ..utils import (
    ExtractorError,
    extract_attributes,
    int_or_none,
    str_or_none,
    traverse_obj,
    unescapeHTML,
    url_or_none,
)


class ZaikoIE(InfoExtractor):
    _VALID_URL = r'https?://(?:[\w-]+\.)?zaiko\.io/event/(?P<id>\d+)/stream(?:/\d+)+'
    _TESTS = [{
        'url': 'https://zaiko.io/event/324868/stream/20571/20571',
        'info_dict': {
            'id': '324868',
            'ext': 'mp4',
            'title': 'ZAIKO STREAMING TEST',
            'alt_title': '[VOD] ZAIKO STREAMING TEST_20210603(Do Not Delete)',
            'uploader_id': '454',
            'uploader': 'ZAIKO ZERO',
            'release_timestamp': 1583809200,
            'thumbnail': r're:https://[a-z0-9]+.cloudfront.net/[a-z0-9_]+/[a-z0-9_]+',
            'release_date': '20200310',
            'categories': ['Tech House'],
            'live_status': 'was_live',
        },
        'params': {'skip_download': 'm3u8'},
    }]

    def _parse_vue_element_attr(self, name, string, video_id):
        page_elem = self._search_regex(rf'(<{name}[^>]+>)', string, name)
        attrs = {}
        for key, value in extract_attributes(page_elem).items():
            if key.startswith(':'):
                attrs[key[1:]] = self._parse_json(
                    value, video_id, transform_source=unescapeHTML, fatal=False)
        return attrs

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