[postprocessor] Fix `_restrict_to` when a codec is not set

4 years ago · 7e87e27c52
parent d0fb4bd16f
commit 7e87e27c52
1 changed files with 2 additions and 2 deletions
--- a/yt_dlp/postprocessor/common.py
+++ b/yt_dlp/postprocessor/common.py
@ -76,8 +76,8 @@ class PostProcessor(object):
            @functools.wraps(func)
            def wrapper(self, info):
                format_type = (
-                    'video' if info['vcodec'] != 'none'
-                    else 'audio' if info['acodec'] != 'none'
+                    'video' if info.get('vcodec') != 'none'
+                    else 'audio' if info.get('acodec') != 'none'
                    else 'images')
                if allowed[format_type]:
                    return func(self, info)