github-repos/blrec
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: refactor segment_fetcher

Browse Source
This commit is contained in:
acgnhik
2022-10-01 15:52:01 +08:00
parent f828ffd885
commit a64d2d7153
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/blrec/hls/operators/segment_fetcher.py
View File

@@ -11,7 +11,14 @@ import urllib3
from m3u8.model import InitializationSection
from reactivex import Observable, abc
from reactivex.disposable import CompositeDisposable, Disposable, SerialDisposable
from tenacity import retry, retry_if_exception_type, stop_after_delay, wait_exponential
from tenacity import (
retry,
retry_all,
retry_if_exception_type,
retry_if_not_exception_type,
stop_after_delay,
wait_exponential,
)
from blrec.bili.live import Live
from blrec.utils.hash import cksum
@@ -137,8 +144,11 @@ class SegmentFetcher:
@retry(
reraise=True,
retry=retry_if_exception_type(
(requests.exceptions.RequestException, urllib3.exceptions.HTTPError)
retry=retry_all(
retry_if_exception_type(
(requests.exceptions.RequestException, urllib3.exceptions.HTTPError)
),
retry_if_not_exception_type(requests.exceptions.HTTPError),
),
wait=wait_exponential(max=10),
stop=stop_after_delay(60),