fix: musify downloading

This commit is contained in:
Hellow 2024-04-11 20:13:12 +02:00
parent bdb4a46c1a
commit 44d1f0d7db
2 changed files with 15 additions and 5 deletions

View File

@ -119,7 +119,7 @@ class Connection:
) -> Dict[str, str]:
headers = self.get_header(**(headers or {}))
if not refer_from_origin:
headers["Referer"] = self.base_url(url=url)
headers["Referer"] = self.base_url(url=url)
return headers
@ -145,6 +145,7 @@ class Connection:
disable_cache: bool = None,
method: str = None,
name: str = "",
exclude_headers: List[str] = None,
**kwargs
) -> Optional[requests.Response]:
if method is None:
@ -158,7 +159,6 @@ class Connection:
current_kwargs.pop("kwargs")
current_kwargs.update(**kwargs)
parsed_url = urlparse(url)
if not raw_headers:
@ -195,6 +195,10 @@ class Connection:
if timeout is None:
timeout = self.TIMEOUT
for header in exclude_headers or []:
if header in headers:
del headers[header]
r = None
connection_failed = False
try:
@ -308,7 +312,7 @@ class Connection:
return DownloadResult(error_message=f"Could not establish a stream from: {url}")
target.create_path()
total_size = int(r.headers.get('content-length'))
total_size = int(r.headers.get('content-length', r.headers.get('Content-Length', chunk_size)))
progress = 0
retry = False

View File

@ -118,7 +118,13 @@ class Musify(Page):
def __init__(self, *args, **kwargs):
self.connection: Connection = Connection(
host="https://musify.club/",
logger=self.LOGGER
logger=self.LOGGER,
)
self.stream_connection: Connection = Connection(
host="https://musify.club/",
logger=self.LOGGER,
semantic_not_found=False,
)
super().__init__(*args, **kwargs)
@ -1121,4 +1127,4 @@ class Musify(Page):
self.LOGGER.warning(f"The source has no audio link. Falling back to {endpoint}.")
return self.connection.stream_into(endpoint, target, raw_url=True)
return self.stream_connection.stream_into(endpoint, target, raw_url=True, exclude_headers=["Host"])