depau/youtube-dl
1
0
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl.git synced 2024-05-29 00:19:32 +00:00
Code Issues Releases Wiki Activity

Retry loop with itertools

Browse source
This commit is contained in:
dirkf 2024-02-21 13:07:25 +00:00 committed by GitHub
parent f7173cca4d
commit 2bd001fa8c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
test/test_download.py
View file

@ -20,6 +20,7 @@ from test.helper import (
import hashlib
import itertools
import json
import socket
import re
@ -175,8 +176,7 @@ def generator(test_case, tname):
try_rm_tcs_files()
try:
try_num = 1
while True:
for try_num in itertools.count(1):
try:
# We're not using .download here since that is just a shim
# for outside error handling, and returns the exit code
@ -185,7 +185,7 @@ def generator(test_case, tname):
test_case['url'],
force_generic_extractor=params.get('force_generic_extractor', False))
except (DownloadError, ExtractorError) as err:
# Check if the exception is not a network related one
# Retry, or raise if the exception is not network-related
if not err.exc_info[0] in (compat_urllib_error.URLError, socket.timeout, UnavailableVideoError, compat_http_client.BadStatusLine) or (err.exc_info[0] == compat_HTTPError and err.exc_info[1].code == 503):
msg = getattr(err, 'msg', error_to_compat_str(err))
err.msg = '%s (%s)' % (msg, tname, )
@ -196,8 +196,6 @@ def generator(test_case, tname):
return
print('Retrying: {0} failed tries\n\n##########\n\n'.format(try_num))
try_num += 1
else:
break