🛠️ Implemented timeout for `requests calls.

This commit is contained in:
Robert Honz
2024-01-12 08:38:41 +01:00
parent 1494490476
commit 69cf0b4707
4 changed files with 9 additions and 4 deletions
+4 -1
View File
@@ -61,6 +61,8 @@ __ERROR_KEY__ = (
},
)
from constants import REQUESTS_TIMEOUT_SEC
def getNum():
return len(__API_KEYS__["keys"])
@@ -94,7 +96,8 @@ def getVersion():
# Load from gist
try:
respond = requests.get("https://api.github.com/gists/48d01f5a24b4b7b37f19443977c22cd6")
respond = requests.get("https://api.github.com/gists/48d01f5a24b4b7b37f19443977c22cd6",
timeout=REQUESTS_TIMEOUT_SEC)
if respond.status_code == 200:
content = respond.json()["files"]["tidal-api-key.json"]["content"]
__API_KEYS__ = json.loads(content)
+1
View File
@@ -1,6 +1,7 @@
from enum import Enum
CTX_TIDAL: str = "tidal"
REQUESTS_TIMEOUT_SEC = 45
class QualityVideo(Enum):
+2 -2
View File
@@ -17,7 +17,7 @@ from rich.progress import Progress
from tidalapi import Album, Mix, Playlist, Session, Track, UserPlaylist, Video
from tidal_dl_ng.config import Settings
from tidal_dl_ng.constants import MediaType
from tidal_dl_ng.constants import REQUESTS_TIMEOUT_SEC, MediaType
from tidal_dl_ng.helper.decryption import decrypt_file, decrypt_security_token
from tidal_dl_ng.helper.exceptions import MediaUnknown
from tidal_dl_ng.helper.path import check_file_exists, format_path_media, path_validate
@@ -134,7 +134,7 @@ class Download:
try:
# Streaming, so we can iterate over the response.
r = requests.get(stream_url, stream=True)
r = requests.get(stream_url, stream=True, timeout=REQUESTS_TIMEOUT_SEC)
r.raise_for_status()
total_size_in_bytes = int(r.headers.get("content-length", 0))
block_size = 4096
+2 -1
View File
@@ -2,6 +2,7 @@ import pathlib
import mutagen
import requests
from constants import REQUESTS_TIMEOUT_SEC
from mutagen import flac, mp4
from mutagen.id3 import APIC, TALB, TCOM, TCOP, TDRC, TIT2, TOPE, TPE1, TRCK, TSRC, USLT
@@ -158,7 +159,7 @@ class Metadata:
if url:
try:
result = requests.get(url).content
result = requests.get(url, timeout=REQUESTS_TIMEOUT_SEC).content
except:
pass
elif path_file: