deemix-py/deemix/__init__.py

64 lines
2.2 KiB
Python

#!/usr/bin/env python3
import re
from urllib.request import urlopen
from deemix.itemgen import generateTrackItem, generateAlbumItem, generatePlaylistItem, generateArtistItem, generateArtistDiscographyItem, generateArtistTopItem
__version__ = "2.0.16"
USER_AGENT_HEADER = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) " \
"Chrome/79.0.3945.130 Safari/537.36"
# Returns the Resolved URL, the Type and the ID
def parseLink(link):
if 'deezer.page.link' in link: link = urlopen(url).url # Resolve URL shortner
# Remove extra stuff
if '?' in link: link = link[:link.find('?')]
if '&' in link: link = link[:link.find('&')]
if link.endswith('/'): link = link[:-1] # Remove last slash if present
type = None
id = None
if not 'deezer' in link: return (link, type, id) # return if not a deezer link
if '/track' in link:
type = 'track'
id = link[link.rfind("/") + 1:]
elif '/playlist' in link:
type = 'playlist'
id = re.search("\/playlist\/(\d+)", link)[0]
elif '/album' in link:
type = 'album'
id = link[link.rfind("/") + 1:]
elif re.search("\/artist\/(\d+)\/top_track", link):
type = 'artist_top'
id = re.search("\/artist\/(\d+)\/top_track", link)[0]
elif re.search("\/artist\/(\d+)\/discography", link):
type = 'artist_discography'
id = re.search("\/artist\/(\d+)\/discography", link)[0]
elif '/artist' in link:
type = 'artist'
id = re.search("\/artist\/(\d+)", link)[0]
return (link, type, id)
def generateDownloadItem(dz, link, bitrate):
(link, type, id) = parseLink(link)
if type == None or id == None: return None
if type == "track":
return generateTrackItem(dz, id, bitrate)
elif type == "album":
return generateAlbumItem(dz, id, bitrate)
elif type == "playlist":
return generatePlaylistItem(dz, id, bitrate)
elif type == "artist":
return generateArtistItem(dz, id, bitrate)
elif type == "artist_discography":
return generateArtistDiscographyItem(dz, id, bitrate)
elif type == "artist_top":
return generateArtistTopItem(dz, id, bitrate)
return None