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

Refactor #2

Merged
KevinMidboe merged 70 commits from refactor into master 2018-10-16 21:11:45 +00:00
Conversation 1 Commits 70 Files Changed 20 +10 -3
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 3d63a3bf08 - Show all commits

9
src/utils.py
View File

@@ -7,7 +7,7 @@ import re
import struct import struct
from babelfish import Error as BabelfishError, Language from babelfish import Error as BabelfishError, Language
from enzyme import MKV from enzyme import MalformedMKVError, MKV
def sanitize(string, ignore_characters=None): def sanitize(string, ignore_characters=None):
"""Sanitize a string to strip special characters. """Sanitize a string to strip special characters.
@@ -58,7 +58,14 @@ def refine(video, embedded_subtitles=True, **kwargs):
extension = os.path.splitext(video.name)[1] extension = os.path.splitext(video.name)[1]
if extension == '.mkv': if extension == '.mkv':
with open(video.name, 'rb') as f: with open(video.name, 'rb') as f:
try:
mkv = MKV(f) mkv = MKV(f)
except MalformedMKVError:
logging.error('Failed to parse mkv, malformed file')
return
except KeyError:
logging.error('Key error while opening file, uncompatible mkv container')
return
# main video track # main video track
if mkv.video_tracks: if mkv.video_tracks: