haruhi-dl/haruhi_dl/extractor/senatpl.py
2021-10-23 01:06:54 +02:00

84 lines
2.8 KiB
Python

# coding: utf-8
from .common import InfoExtractor
import datetime
class SenatPlIE(InfoExtractor):
_VALID_URL = r'https://av8\.senat\.pl/(?P<id>\d+[a-zA-Z\d]+)'
IE_NAME = 'senat.gov.pl'
_TESTS = [{
'url': 'https://av8.senat.pl/10Sen221',
'info_dict': {
'id': '10Sen221',
'title': '22. posiedzenie Senatu RP X kadencji - 24.03.2021 r. - cz. 1',
},
'playlist_count': 2,
}]
def _real_extract(self, url):
video_id = self._match_id(url)
webpage = self._download_webpage(url, video_id)
res_type = self._search_regex(
r'<script [^>]+ src="/senat-console/static/js/generated/(vod|live)\.item\.js">',
webpage, 'resource type')
vod = self._download_json(
'https://av8.senat.pl/senat-console/side-menu/transmissions/%s/%s' % (video_id, res_type),
video_id, 'Downloading transmission metadata')
conf = self._download_json(
'https://av8.senat.pl/senat-console/side-menu/transmissions/%s/%s/player-configuration' % (video_id, res_type),
video_id, 'Downloading player configuration')
def unix_milliseconds_to_wtf_atende_wants(date):
date = datetime.datetime.fromtimestamp(date / 1000)
# atende uses timestamp but since 2001 instead of 1970
date = date.replace(year=date.year - 31)
# also it's in milliseconds
return int(date.timestamp() * 1000)
start_time = unix_milliseconds_to_wtf_atende_wants(vod['since'])
if res_type == 'vod':
stop_time = unix_milliseconds_to_wtf_atende_wants(vod['till'])
else:
stop_time = None
if stop_time:
duration = (stop_time - start_time) // 1000
else:
duration = None
entries = []
def add_entry(player):
trans_url = f"https:{player['playlist']['flv']}?startTime={start_time}"
if stop_time:
trans_url += f"&stopTime={stop_time}"
stream_id = self._search_regex(
r'/o2/senat/([^/]+)/[^./]+\.livx', trans_url, 'stream id')
entries.append({
'_type': 'url_transparent',
'url': trans_url,
'ie_key': 'SejmPlVideo',
'id': stream_id,
'title': stream_id,
'duration': duration,
'is_live': res_type == 'live',
})
add_entry(conf['player'])
# PJM translator
if conf.get('sliPlayer'):
add_entry(conf['sliPlayer'])
return {
'_type': 'multi_video',
'entries': entries,
'id': video_id,
'title': vod['title'],
'duration': duration,
'is_live': res_type == 'live',
}