[nrk] Improve series metadata extraction

This commit is contained in:
Sergey M․ 2021-01-03 07:30:45 +07:00
parent 7dfd966848
commit 85de33b04e
No known key found for this signature in database
GPG key ID: 2C393E0F18A9236D

View file

@ -223,7 +223,9 @@ class NRKIE(NRKBaseIE):
age_limit = int_or_none(try_get( age_limit = int_or_none(try_get(
data, lambda x: x['legalAge']['body']['rating']['code'])) data, lambda x: x['legalAge']['body']['rating']['code']))
return { is_series = try_get(data, lambda x: x['_links']['series']['name']) == 'series'
info = {
'id': video_id, 'id': video_id,
'title': title, 'title': title,
'alt_title': alt_title, 'alt_title': alt_title,
@ -235,6 +237,27 @@ class NRKIE(NRKBaseIE):
'subtitles': subtitles, 'subtitles': subtitles,
} }
if is_series:
series = title
if alt_title:
title += ' - %s' % alt_title
season_number = int_or_none(self._search_regex(
r'Sesong\s+(\d+)', description or '', 'season number',
default=None))
episode = alt_title if is_series else None
episode_number = int_or_none(self._search_regex(
r'(\d+)\.\s+episode', episode or '', 'episode number',
default=None))
info.update({
'title': title,
'series': series,
'season_number': season_number,
'episode': episode,
'episode_number': episode_number,
})
return info
class NRKTVIE(InfoExtractor): class NRKTVIE(InfoExtractor):
IE_DESC = 'NRK TV and NRK Radio' IE_DESC = 'NRK TV and NRK Radio'