158 lines
6.0 KiB
Python
158 lines
6.0 KiB
Python
from deezer.gw import LyricsStatus
|
|
|
|
from deemix.utils import removeDuplicateArtists, removeFeatures
|
|
from deemix.types.Artist import Artist
|
|
from deemix.types.Date import Date
|
|
from deemix.types.Picture import Picture
|
|
from deemix.types import VARIOUS_ARTISTS
|
|
|
|
class Album:
|
|
def __init__(self, alb_id="0", title="", pic_md5=""):
|
|
self.id = alb_id
|
|
self.title = title
|
|
self.pic = Picture(pic_md5, "cover")
|
|
self.artist = {"Main": []}
|
|
self.artists = []
|
|
self.mainArtist = None
|
|
self.date = Date()
|
|
self.dateString = ""
|
|
self.trackTotal = "0"
|
|
self.discTotal = "0"
|
|
self.embeddedCoverPath = ""
|
|
self.embeddedCoverURL = ""
|
|
self.explicit = False
|
|
self.genre = []
|
|
self.barcode = "Unknown"
|
|
self.label = "Unknown"
|
|
self.copyright = ""
|
|
self.recordType = "album"
|
|
self.bitrate = 0
|
|
self.rootArtist = None
|
|
self.variousArtists = None
|
|
|
|
self.playlistId = None
|
|
self.owner = None
|
|
self.isPlaylist = False
|
|
|
|
def parseAlbum(self, albumAPI):
|
|
self.title = albumAPI['title']
|
|
|
|
# Getting artist image ID
|
|
# ex: https://e-cdns-images.dzcdn.net/images/artist/f2bc007e9133c946ac3c3907ddc5d2ea/56x56-000000-80-0-0.jpg
|
|
art_pic = albumAPI['artist']['picture_small']
|
|
art_pic = art_pic[art_pic.find('artist/') + 7:-24]
|
|
self.mainArtist = Artist(
|
|
albumAPI['artist']['id'],
|
|
albumAPI['artist']['name'],
|
|
"Main",
|
|
art_pic
|
|
)
|
|
if albumAPI.get('root_artist'):
|
|
art_pic = albumAPI['root_artist']['picture_small']
|
|
art_pic = art_pic[art_pic.find('artist/') + 7:-24]
|
|
self.rootArtist = Artist(
|
|
albumAPI['root_artist']['id'],
|
|
albumAPI['root_artist']['name'],
|
|
"Root",
|
|
art_pic
|
|
)
|
|
|
|
for artist in albumAPI['contributors']:
|
|
isVariousArtists = str(artist['id']) == VARIOUS_ARTISTS
|
|
isMainArtist = artist['role'] == "Main"
|
|
|
|
if isVariousArtists:
|
|
self.variousArtists = Artist(
|
|
art_id = artist['id'],
|
|
name = artist['name'],
|
|
role = artist['role']
|
|
)
|
|
continue
|
|
|
|
if artist['name'] not in self.artists:
|
|
self.artists.append(artist['name'])
|
|
|
|
if isMainArtist or artist['name'] not in self.artist['Main'] and not isMainArtist:
|
|
if not artist['role'] in self.artist:
|
|
self.artist[artist['role']] = []
|
|
self.artist[artist['role']].append(artist['name'])
|
|
|
|
self.trackTotal = albumAPI['nb_tracks']
|
|
self.recordType = albumAPI['record_type']
|
|
|
|
self.barcode = albumAPI.get('upc', self.barcode)
|
|
self.label = albumAPI.get('label', self.label)
|
|
self.explicit = bool(albumAPI.get('explicit_lyrics', False))
|
|
if 'release_date' in albumAPI:
|
|
self.date.day = albumAPI["release_date"][8:10]
|
|
self.date.month = albumAPI["release_date"][5:7]
|
|
self.date.year = albumAPI["release_date"][0:4]
|
|
self.date.fixDayMonth()
|
|
|
|
self.discTotal = albumAPI.get('nb_disk')
|
|
self.copyright = albumAPI.get('copyright')
|
|
|
|
if self.pic.md5 == "" and albumAPI.get('cover_small'):
|
|
# Getting album cover MD5
|
|
# ex: https://e-cdns-images.dzcdn.net/images/cover/2e018122cb56986277102d2041a592c8/56x56-000000-80-0-0.jpg
|
|
alb_pic = albumAPI['cover_small']
|
|
self.pic.md5 = alb_pic[alb_pic.find('cover/') + 6:-24]
|
|
|
|
if albumAPI.get('genres') and len(albumAPI['genres'].get('data', [])) > 0:
|
|
for genre in albumAPI['genres']['data']:
|
|
self.genre.append(genre['name'])
|
|
|
|
def parseAlbumGW(self, albumAPI_gw):
|
|
self.title = albumAPI_gw['ALB_TITLE']
|
|
self.mainArtist = Artist(
|
|
art_id = albumAPI_gw['ART_ID'],
|
|
name = albumAPI_gw['ART_NAME'],
|
|
role = "Main"
|
|
)
|
|
|
|
self.artists = [albumAPI_gw['ART_NAME']]
|
|
self.trackTotal = albumAPI_gw['NUMBER_TRACK']
|
|
self.discTotal = albumAPI_gw['NUMBER_DISK']
|
|
self.label = albumAPI_gw.get('LABEL_NAME', self.label)
|
|
|
|
explicitLyricsStatus = albumAPI_gw.get('EXPLICIT_ALBUM_CONTENT', {}).get('EXPLICIT_LYRICS_STATUS', LyricsStatus.UNKNOWN)
|
|
self.explicit = explicitLyricsStatus in [LyricsStatus.EXPLICIT, LyricsStatus.PARTIALLY_EXPLICIT]
|
|
|
|
self.addExtraAlbumGWData(albumAPI_gw)
|
|
|
|
def addExtraAlbumGWData(self, albumAPI_gw):
|
|
if self.pic.md5 == "" and albumAPI_gw.get('ALB_PICTURE'):
|
|
self.pic.md5 = albumAPI_gw['ALB_PICTURE']
|
|
if 'PHYSICAL_RELEASE_DATE' in albumAPI_gw:
|
|
self.date.day = albumAPI_gw["PHYSICAL_RELEASE_DATE"][8:10]
|
|
self.date.month = albumAPI_gw["PHYSICAL_RELEASE_DATE"][5:7]
|
|
self.date.year = albumAPI_gw["PHYSICAL_RELEASE_DATE"][0:4]
|
|
self.date.fixDayMonth()
|
|
|
|
def makePlaylistCompilation(self, playlist):
|
|
self.variousArtists = playlist.variousArtists
|
|
self.mainArtist = playlist.mainArtist
|
|
self.title = playlist.title
|
|
self.rootArtist = playlist.rootArtist
|
|
self.artist = playlist.artist
|
|
self.artists = playlist.artists
|
|
self.trackTotal = playlist.trackTotal
|
|
self.recordType = playlist.recordType
|
|
self.barcode = playlist.barcode
|
|
self.label = playlist.label
|
|
self.explicit = playlist.explicit
|
|
self.date = playlist.date
|
|
self.discTotal = playlist.discTotal
|
|
self.playlistId = playlist.playlistId
|
|
self.owner = playlist.owner
|
|
self.pic = playlist.pic
|
|
self.isPlaylist = True
|
|
|
|
def removeDuplicateArtists(self):
|
|
"""Removes duplicate artists for both artist array and artists dict"""
|
|
(self.artist, self.artists) = removeDuplicateArtists(self.artist, self.artists)
|
|
|
|
def getCleanTitle(self):
|
|
"""Removes featuring from the album name"""
|
|
return removeFeatures(self.title)
|