From 1ae01ed1fd19925c0829f825720a748896b00514 Mon Sep 17 00:00:00 2001 From: Hellow Date: Sat, 18 Mar 2023 17:06:12 +0100 Subject: [PATCH] added wikipedia --- src/music_kraken/objects/source.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/music_kraken/objects/source.py b/src/music_kraken/objects/source.py index 3214513..b48cb64 100644 --- a/src/music_kraken/objects/source.py +++ b/src/music_kraken/objects/source.py @@ -26,6 +26,7 @@ class SourcePages(Enum): SPOTIFY = "spotify" # This has nothing to do with audio, but bands can be here + WIKIPEDIA = "wikipedia" INSTAGRAM = "instagram" FACEBOOK = "facebook" TWITTER = "twitter" # I will use nitter though lol @@ -45,7 +46,8 @@ class SourcePages(Enum): cls.FACEBOOK: "https://www.facebook.com/", cls.SPOTIFY: "https://open.spotify.com/", cls.TWITTER: "https://twitter.com/", - cls.MYSPACE: "https://myspace.com/" + cls.MYSPACE: "https://myspace.com/", + cls.WIKIPEDIA: "https://en.wikipedia.org/wiki/Main_Page" } return homepage_map[attribute] @@ -94,6 +96,9 @@ class Source(DatabaseObject): if "bandcamp" in url: return cls(SourcePages.BANDCAMP, url) + if "wikipedia" in parsed.netloc: + return cls(SourcePages.WIKIPEDIA, url) + if url.startswith("https://www.metal-archives.com/"): return cls(SourcePages.ENCYCLOPAEDIA_METALLUM, url)