diff --git a/brozzler/ydl.py b/brozzler/ydl.py index 563842f..01c8b7b 100644 --- a/brozzler/ydl.py +++ b/brozzler/ydl.py @@ -117,9 +117,11 @@ def _build_youtube_dl(worker, destdir, site, page): def process_ie_result(self, ie_result, download=True, extra_info=None): if extra_info is None: extra_info = {} - if 'redirect_count' in extra_info: - self.logger.info(f"Following redirect URL: {ie_result['url']} redirect_count: {extra_info['redirect_count']}") - extra_info['redirect_count'] = 1 + extra_info.get('redirect_count', 0) + if "redirect_count" in extra_info: + self.logger.info( + f"Following redirect URL: {ie_result['url']} redirect_count: {extra_info['redirect_count']}" + ) + extra_info["redirect_count"] = 1 + extra_info.get("redirect_count", 0) if extra_info["redirect_count"] > YTDLP_MAX_REDIRECTS: raise ExtractorError( f"Too many redirects for URL: {ie_result['url']}", @@ -128,7 +130,6 @@ def _build_youtube_dl(worker, destdir, site, page): super().process_ie_result(ie_result, download, extra_info) - def add_default_extra_info(self, ie_result, ie, url): # hook in some logging super().add_default_extra_info(ie_result, ie, url)