diff --git a/src/music_kraken/pages/musify.py b/src/music_kraken/pages/musify.py index 50295dd..ba7f513 100644 --- a/src/music_kraken/pages/musify.py +++ b/src/music_kraken/pages/musify.py @@ -8,6 +8,7 @@ import pycountry import requests from bs4 import BeautifulSoup +from ..connection import Connection from .abstract import Page from ..utils.enums.source import SourcePages from ..utils.enums.album import AlbumType, AlbumStatus @@ -22,6 +23,7 @@ from ..objects import ( Options, Target ) +from ..utils.shared import MUSIFY_LOGGER from ..utils import string_processing, shared from .support_classes.download_result import DownloadResult @@ -76,6 +78,11 @@ class Musify(Page): TRIES = 5 HOST = "https://musify.club" + CONNECTION = Connection( + host="https://musify.club/", + logger=MUSIFY_LOGGER + ) + SOURCE_TYPE = SourcePages.MUSIFY LOGGER = shared.MUSIFY_LOGGER @@ -355,7 +362,7 @@ class Musify(Page): def plaintext_search(cls, query: str) -> Options: search_results = [] - r = cls.get_request(f"https://musify.club/search?searchText={query}") + r = cls.CONNECTION.get(f"https://musify.club/search?searchText={query}") if r is None: return Options() search_soup: BeautifulSoup = BeautifulSoup(r.content, features="html.parser") @@ -580,7 +587,7 @@ class Musify(Page): :return: """ - r = cls.get_request(f"https://musify.club/{url.source_type.value}/{url.name_with_id}?_pjax=#bodyContent") + r = cls.CONNECTION.get(f"https://musify.club/{url.source_type.value}/{url.name_with_id}?_pjax=#bodyContent") if r is None: return Artist(_id=url.musify_id) @@ -976,7 +983,7 @@ class Musify(Page): url = cls.parse_url(source.url) endpoint = cls.HOST + "/release/" + url.name_with_id - r = cls.get_request(endpoint) + r = cls.CONNECTION.get(endpoint) if r is None: return Album() @@ -1032,7 +1039,7 @@ class Musify(Page): cls.LOGGER.warning(f"The source has no audio link. Falling back to {endpoint}.") - r = cls.get_request(endpoint, stream=True) + r = cls.CONNECTION.get(endpoint, stream=True) if r is None: return DownloadResult(error_message=f"couldn't connect to {endpoint}")