KevinMidboe/seasonedParser
1
0
Fork 0
Code Issues 7 Pull Requests Packages Projects Releases 1 Wiki Activity

Video gets a new parameter embeded_subtitles to represent the subtitle files found in the mkv container.

Browse Source
This commit is contained in:
Kevin
2018-09-16 23:26:06 +02:00
parent 1ff34630ac
commit 3e2faadc2f
2 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/utils.py
View File

@@ -47,7 +47,7 @@ def refine(video, embedded_subtitles=True, **kwargs):
* :attr:`~subliminal.video.Video.resolution`
* :attr:`~subliminal.video.Video.video_codec`
* :attr:`~subliminal.video.Video.audio_codec`
* :attr:`~subliminal.video.Video.subtitle_languages`
* :attr:`~subliminal.video.Video.embeded_subtitles`
:param bool embedded_subtitles: search for embedded subtitles.
"""
# skip non existing videos
@@ -111,24 +111,24 @@ def refine(video, embedded_subtitles=True, **kwargs):
# subtitle tracks
if mkv.subtitle_tracks:
if embedded_subtitles:
embedded_subtitle_languages = set()
embeded_subtitles = set()
for st in mkv.subtitle_tracks:
if st.language:
try:
embedded_subtitle_languages.add(Language.fromalpha3b(st.language))
embeded_subtitles.add(Language.fromalpha3b(st.language))
except BabelfishError:
logging.error('Embedded subtitle track language %r is not a valid language', st.language)
embedded_subtitle_languages.add(Language('und'))
embeded_subtitles.add(Language('und'))
elif st.name:
try:
embedded_subtitle_languages.add(Language.fromname(st.name))
embeded_subtitles.add(Language.fromname(st.name))
except BabelfishError:
logging.debug('Embedded subtitle track name %r is not a valid language', st.name)
embedded_subtitle_languages.add(Language('und'))
embeded_subtitles.add(Language('und'))
else:
embedded_subtitle_languages.add(Language('und'))
logging.debug('Found embedded subtitle %r', embedded_subtitle_languages)
video.subtitle_languages |= embedded_subtitle_languages
embeded_subtitles.add(Language('und'))
logging.debug('Found embedded subtitle %r', embeded_subtitles)
video.embeded_subtitles |= embeded_subtitles
else:
logging.debug('MKV has no subtitle track')
else:

5
src/video.py
View File

@@ -33,7 +33,7 @@ class Video(object):
:param set subtitles: existing subtitle languages.
"""
def __init__(self, name, hash=None, size=None, format=None, release_group=None, resolution=None, video_codec=None, audio_codec=None,
imdb_id=None, subtitles=None):
imdb_id=None, subtitles=None, embeded_subtitles=None):
#: Name or path of the video
self.name = name
@@ -64,6 +64,9 @@ class Video(object):
#: Existing subtitle languages
self.subtitles = subtitles or set()
#: Embeded subtitle languages
self.embeded_subtitles = embeded_subtitles or set()
@property
def exists(self):
"""Test whether the video exists"""