music-kraken/music-kraken-core
3
2
Fork 0
Code Issues 10 Pull Requests 1 Packages Projects 2 Releases Wiki Activity

made the referer in sources more concistent

Browse Source
This commit is contained in:
Hellow2 2023-04-18 13:35:00 +02:00
parent ef08dca088
commit 58c58b2141
3 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/music_kraken/objects/source.py
View File

@ -30,7 +30,7 @@ class Source(DatabaseObject):
page_enum: SourcePages, page_enum: SourcePages,
url: str, url: str,
id_: str = None, id_: str = None,
referer_page: SourceTypes = None, referer_page: SourcePages = None,
adio_url: str = None adio_url: str = None
) -> None: ) -> None:
DatabaseObject.__init__(self, id_=id_) DatabaseObject.__init__(self, id_=id_)
@ -42,7 +42,7 @@ class Source(DatabaseObject):
self.audio_url = adio_url self.audio_url = adio_url
@classmethod @classmethod
def match_url(cls, url: str) -> Optional["Source"]: def match_url(cls, url: str, referer_page: SourcePages) -> Optional["Source"]:
""" """
this shouldn't be used, unlesse you are not certain what the source is for this shouldn't be used, unlesse you are not certain what the source is for
the reason is that it is more inefficient the reason is that it is more inefficient
@ -51,38 +51,38 @@ class Source(DatabaseObject):
url = parsed.geturl() url = parsed.geturl()
if "musify" in parsed.netloc: if "musify" in parsed.netloc:
return cls(SourcePages.MUSIFY, url) return cls(SourcePages.MUSIFY, url, referer_page=referer_page)
if url.startswith("https://www.youtube"): if url.startswith("https://www.youtube"):
return cls(SourcePages.YOUTUBE, url) return cls(SourcePages.YOUTUBE, url, referer_page=referer_page)
if url.startswith("https://www.deezer"): if url.startswith("https://www.deezer"):
return cls(SourcePages.DEEZER, url) return cls(SourcePages.DEEZER, url, referer_page=referer_page)
if url.startswith("https://open.spotify.com"): if url.startswith("https://open.spotify.com"):
return cls(SourcePages.SPOTIFY, url) return cls(SourcePages.SPOTIFY, url, referer_page=referer_page)
if "bandcamp" in url: if "bandcamp" in url:
return cls(SourcePages.BANDCAMP, url) return cls(SourcePages.BANDCAMP, url, referer_page=referer_page)
if "wikipedia" in parsed.netloc: if "wikipedia" in parsed.netloc:
return cls(SourcePages.WIKIPEDIA, url) return cls(SourcePages.WIKIPEDIA, url, referer_page=referer_page)
if url.startswith("https://www.metal-archives.com/"): if url.startswith("https://www.metal-archives.com/"):
return cls(SourcePages.ENCYCLOPAEDIA_METALLUM, url) return cls(SourcePages.ENCYCLOPAEDIA_METALLUM, url, referer_page=referer_page)
# the less important once # the less important once
if url.startswith("https://www.facebook"): if url.startswith("https://www.facebook"):
return cls(SourcePages.FACEBOOK, url) return cls(SourcePages.FACEBOOK, url, referer_page=referer_page)
if url.startswith("https://www.instagram"): if url.startswith("https://www.instagram"):
return cls(SourcePages.INSTAGRAM, url) return cls(SourcePages.INSTAGRAM, url, referer_page=referer_page)
if url.startswith("https://twitter"): if url.startswith("https://twitter"):
return cls(SourcePages.TWITTER, url) return cls(SourcePages.TWITTER, url, referer_page=referer_page)
if url.startswith("https://myspace.com"): if url.startswith("https://myspace.com"):
return cls(SourcePages.MYSPACE, url) return cls(SourcePages.MYSPACE, url, referer_page=referer_page)
def get_song_metadata(self) -> Metadata: def get_song_metadata(self) -> Metadata:
return Metadata({ return Metadata({

2
src/music_kraken/pages/encyclopaedia_metallum.py
View File

@ -285,7 +285,7 @@ class EncyclopaediaMetallum(Page):
if url is None: if url is None:
continue continue
source_list.append(Source.match_url(url)) source_list.append(Source.match_url(url, referer_page=cls.SOURCE_TYPE))
return source_list return source_list

2
src/music_kraken/pages/musify.py
View File

@ -681,7 +681,7 @@ class Musify(Page):
href = additional_source.get("href") href = additional_source.get("href")
if href is None: if href is None:
continue continue
new_src = Source.match_url(href) new_src = Source.match_url(href, referer_page=cls.SOURCE_TYPE)
if new_src is None: if new_src is None:
continue continue
source_list.append(new_src) source_list.append(new_src)