ahhh fuck me :(

This commit is contained in:
Hellow2
2023-03-21 12:46:32 +01:00
parent 0f47cdadb8
commit c648a330e0
5 changed files with 142 additions and 14 deletions

View File

@@ -15,15 +15,24 @@ from ..objects import (
Target,
MusicObject,
Options,
SourcePages
SourcePages,
Collection
)
class PageCache(Collection):
def clear(self):
self.__init__(element_type=self.element_type)
class Page:
"""
This is an abstract class, laying out the
functionality for every other class fetching something
"""
SONG_CACHE = PageCache(element_type=Song)
ALBUM_CACHE = PageCache(element_type=Album)
ARTIST_CACHE = PageCache(element_type=Artist)
API_SESSION: requests.Session = requests.Session()
API_SESSION.proxies = shared.proxies
@@ -151,6 +160,10 @@ class Page:
tracklist of every album of the artist.
:return detailed_music_object: IT MODIFIES THE INPUT OBJ
"""
cls.ARTIST_CACHE.clear()
cls.ALBUM_CACHE.clear()
cls.SONG_CACHE.clear()
if type(music_object) == Song:
song = cls.fetch_song_details(music_object, flat=flat)

View File

@@ -151,11 +151,11 @@ class Musify(Page):
artist_thumbnail = image_soup.get("src")
return Artist(
return cls.ARTIST_CACHE.append(Artist(
_id=_id,
name=name,
source_list=source_list
)
))
@classmethod
def parse_album_contact(cls, contact: BeautifulSoup) -> Album:
@@ -257,13 +257,13 @@ class Musify(Page):
else:
LOGGER.warning("got an unequal ammount than 3 small elements")
return Album(
return cls.ALBUM_CACHE.append(Album(
_id=_id,
title=title,
source_list=source_list,
date=ID3Timestamp(year=year),
artist_list=artist_list
)
))
@classmethod
def parse_contact_container(cls, contact_container_soup: BeautifulSoup) -> List[Union[Artist, Album]]:
@@ -535,14 +535,14 @@ class Musify(Page):
else:
LOGGER.debug("there is not even 1 footer in the album card")
return Album(
return cls.ALBUM_CACHE.append(Album(
_id=_id,
title=name,
source_list=source_list,
date=timestamp,
album_type=album_type,
album_status=album_status
)
))
@classmethod
def get_discography(cls, url: MusifyUrl, artist_name: str = None, flat=False) -> List[Album]:
@@ -700,13 +700,13 @@ class Musify(Page):
if note_soup is not None:
notes.html = note_soup.decode_contents()
return Artist(
return cls.ARTIST_CACHE.append(Artist(
_id=url.musify_id,
name=name,
country=country,
source_list=source_list,
notes=notes
)
))
@classmethod
def fetch_artist_from_source(cls, source: Source, flat: bool = False) -> Artist:
@@ -842,7 +842,7 @@ class Musify(Page):
_artist_name = meta_artist_name_text
if _artist_name is not None or _artist_src is not None:
artist_list.append(Artist(name=_artist_name, source_list=_artist_src))
artist_list.append(cls.ARTIST_CACHE.append(Artist(name=_artist_name, source_list=_artist_src)))
return Song(
title=song_name,