mirror of
https://github.com/Dvorinka/SpotifyRecAlg.git
synced 2026-06-04 12:33:03 +00:00
Tomas Dvorak
27 lines
519 B
Python
27 lines
519 B
Python
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class SimilarArtistEntry:
|
|
artisthash: str
|
|
name: str
|
|
weight: float
|
|
scrobbles: int
|
|
listeners: int
|
|
|
|
|
|
@dataclass
|
|
class SimilarArtist:
|
|
artisthash: str
|
|
similar_artists: list[SimilarArtistEntry]
|
|
|
|
def get_artist_hash_set(self) -> set[str]:
|
|
"""
|
|
Returns a set of similar artists.
|
|
"""
|
|
if not self.similar_artists:
|
|
return set()
|
|
|
|
# INFO:
|
|
return {a["artisthash"] for a in self.similar_artists}
|