reformat (and add bandcamp)

5 files changed, 157 insertions, 62 deletions

diff --git a/bandcamp.py b/bandcamp.py
new file mode 100755
index 0000000..7b32b3f
--- /dev/null
+++ b/bandcamp.py
@@ -0,0 +1,86 @@
+#!/usr/bin/env python3
+from html.parser import HTMLParser
+from urllib.request import urlopen
+import json
+
+
+class Bandcamp:
+
+    def __init__(self):
+        self.ldjson = None
+
+    def mesg(self, msg, t=None):
+        self.util.mesg(msg, t)
+
+    def match_urls(self, str):
+        r = [
+            i
+            for i in str.split()
+            if "https://" in i
+            and ("bandcamp.com/album/" in i or "bandcamp.com/track/" in i)
+        ]
+        r = list(dict.fromkeys(r))
+        n = 0
+        for i in r:
+            if not i.startswith("http"):
+                r.pop(n)
+            n += 1
+
+        return r
+
+    class parseprop(HTMLParser):
+        def __init__(self):
+            # print("bandcamp parse init")
+            HTMLParser.__init__(self)
+            self.ldjson = False
+
+        def handle_starttag(self, tag, attrs):
+            if tag == "script" and ("type", "application/ld+json") in attrs:
+                self.ldjson = True
+            else:
+                self.ldjson = False
+
+        def handle_endtag(self, tag):
+            self.ldjson = False
+
+        def handle_data(self, data):
+            if self.ldjson:
+                Bandcamp.ldjson = data
+                return
+
+    def bandcamp(self, url):
+        url = url.rstrip("\x01")
+        p = self.parseprop()
+        data = urlopen(url).read().decode()
+        p.feed(data)
+        irc_string = "[\x0304BandCamp\x03] \x0307ERROR:\x0308 got no data from server! \x0315(check your URL for typos!)\x03"
+        ansi_string = "[\x1b[31mBandCamp\x1b[0m] \x1b[33;2mERROR:\x1b[33;1m got no data from server! \x1b[37;2m(check your URL for typos!)\x1b[0m"
+        data = json.loads(Bandcamp.ldjson)
+        # print(data)
+        # for i in data:
+        #  print(i)
+        try:
+            type = data["@type"]
+        except KeyError:
+            print(ansi_string)
+            return irc_string, True
+        id = data["@id"]
+        name = data["name"]
+        date = data["datePublished"]
+        artists = data["byArtist"]["name"]
+        # artists=artists.removeprefix(f'Listen to {name} on Spotify').removeprefix('.').strip().removeprefix('· ')
+        # if artists.startswith("Song · "): artists=artists.removeprefix("Song · ")
+        # elif artists.startswith("Album · "):
+        #  artists=artists.removeprefix("Album · ")[::-1].split(" · ",1)[1][::-1] #removes the "10 songs" part from album
+        # artists=artists.removesuffix(f" · {date[:4]}")  #remove the year too
+        irc_string = f"[\x0303BandCamp\x03] \x02{name}\x02 by \x1d{artists}\x1d published on {date}"
+        ansi_string = f"[\x1b[32mBandCamp\x1b[0m] \x1b[1m{name}\x1b[0m by \x1b[03m{artists}\x1b[0m published on {date}"
+        # print(("Song: " if type=="MusicRecording" else "Album: " if type=="MusicAlbum" else f"Unknown type ({type}): ")+'"'+name+'"'+" by "+'"'+artists+'"'+" released on "+date)
+        print(ansi_string)
+        return irc_string, False
+
+
+if __name__ == "__main__":
+    import sys
+
+    Bandcamp.bandcamp(Bandcamp, sys.argv[1])
diff --git a/config.py b/config.py
index 27cf5c4..28902ee 100644
--- a/config.py
+++ b/config.py
@@ -4,6 +4,7 @@ import importlib, sys
 if __name__ == "local_config":
     config = importlib.reload(sys.modules["config"]).config
 else:
+
     class config:  # dummy, so the local config can simply be a copy of this template
         ...
 
@@ -13,7 +14,7 @@ class config(config):
         nick = "pawbot"
         username = "pawky_bot"
         # you should probably indicate yourself to be the owner of the bot, in username, or realname, or both
-        realname = "pawky's bot"
+        realname = "pawky's song-of-the-day bot, use ~help"
         source = "/home/pawky/git/sotdbot"  # so far only used for ctcp response
         gitdir = "./"  # where is the bot git dir? currently only used for version (git commit hash, assuming latest commit == latest version)
 
@@ -65,6 +66,7 @@ class config(config):
         "chghost",  #         uh, same, I think?
     ]
 
+
 # cap-notify draft/account-registration draft/channel-rename draft/persistence draft/read-marker echo-message ergo.chat/nope extended-join extended-monitor invite-notify labeled-response sasl=PLAIN,EXTERNAL server-time setname userhost-in-names znc.in/self-message
 # message-tags multi-prefix account-tag batch away-notify account-notify chghost
 
diff --git a/soundcloud.py b/soundcloud.py
index de0e930..ae17f9d 100755
--- a/soundcloud.py
+++ b/soundcloud.py
@@ -1,8 +1,9 @@
 #!/usr/bin/env python3
-from urllib.parse import urlencode,urlparse
+from urllib.parse import urlencode, urlparse
 from urllib.request import urlopen
 from json import loads as json_loads
 
+
 class SoundCloud:
     video_type = ""
 
@@ -10,11 +11,7 @@ class SoundCloud:
         self.util.mesg(msg, t)
 
     def match_urls(self, str):
-        r = [
-            i
-            for i in str.split()
-            if "https://soundcloud.com" in i
-        ]
+        r = [i for i in str.split() if "https://soundcloud.com" in i]
         r = list(dict.fromkeys(r))
         n = 0
         for i in r:
@@ -27,27 +24,31 @@ class SoundCloud:
     def soundcloud(self, url):
         # self.util.mesg("dbg hello")
         url = url.rstrip("\x01")
-        url=urlparse(url)
-        url=url.scheme+"://"+url.netloc+url.path
+        url = urlparse(url)
+        url = url.scheme + "://" + url.netloc + url.path
         url = f"https://soundcloud.com/oembed?{urlencode([('url',url),('format','json')])}"
-        data=urlopen(url).read().decode()
-        data=json_loads(data)
-        """ {'version': 1.0, 'type': 'rich', 'provider_name': 'SoundCloud', 'provider_url': 'https://soundcloud.com', 'height': 400, 'width': '100%', 'title': 'Doses And Mimosas - Cherub (Mocha Remix) by Mocha Music', 'description': 'One of my favorite songs ever with a dubstep twist! Hope you enjoy :)', 'thumbnail_url': 'https://i1.sndcdn.com/artworks-768QwHQ4tGr0P4wc-iH0Zww-t500x500.jpg', 'html': '<iframe width="100%" height="400" scrolling="no" frameborder="no" src="https://w.soundcloud.com/player/?visual=true&url=https%3A%2F%2Fapi.soundcloud.com%2Ftracks%2F1881169554&show_artwork=true"></iframe>', 'author_name': 'Mocha Music', 'author_url': 'https://soundcloud.com/mochamusic11' }"""
-        #print(data["title"].removesuffix(" by "+data["author_name"]),data["author_name"])
+        data = urlopen(url).read().decode()
+        data = json_loads(data)
+        print(data)
+        # print(data["title"].removesuffix(" by "+data["author_name"]),data["author_name"])
         try:
-          artist=data["author_name"];title=data["title"].removesuffix(" by "+artist)
+            artist = data["author_name"]
+            title = data["title"].removesuffix(" by " + artist)
         except KeyError:
-          title="";artist=""
-        print(title.removesuffix(" by "+artist),"|",artist)
+            title = ""
+            artist = ""
+        print(title.removesuffix(" by " + artist), "|", artist)
         if title == "":
             irc_string = "[\x0304SoundCloud\x03] \x0307ERROR:\x0308 got no data from server! \x0315(check your URL for typos!)\x03"
             ansi_string = "[\x1b[31mSoundCloud\x1b[0m] \x1b[33;2mERROR:\x1b[33;1m got no data from server! \x1b[37;2m(check your URL for typos!)\x1b[0m"
             print(ansi_string)
             return irc_string, True
-        irc_string = f"[\x0303SoundCloud\x03] \x02{title}\x02 uploaded by \x1d{artist}\x1d"
+        irc_string = (
+            f"[\x0303SoundCloud\x03] \x02{title}\x02 uploaded by \x1d{artist}\x1d"
+        )
         ansi_string = f"[\x1b[32mSoundCloud\x1b[0m] \x1b[1m{title}\x1b[0m uploaded by \x1b[03m{artist}\x1b[0m"
-        #"""
-        #irc_string="dummy";ansi_string="dummy"
+        # """
+        # irc_string="dummy";ansi_string="dummy"
         print(ansi_string)
         return irc_string, False
 
diff --git a/spotify.py b/spotify.py
index f376bc1..1586784 100755
--- a/spotify.py
+++ b/spotify.py
@@ -3,20 +3,17 @@ from html.parser import HTMLParser
 from urllib.request import urlopen
 import json
 
+
 class Spotify:
 
     def __init__(self):
-        self.ldjson=None
+        self.ldjson = None
 
     def mesg(self, msg, t=None):
         self.util.mesg(msg, t)
 
     def match_urls(self, str):
-        r = [
-            i
-            for i in str.split()
-            if "https://open.spotify.com/" in i
-        ]
+        r = [i for i in str.split() if "https://open.spotify.com/" in i]
         r = list(dict.fromkeys(r))
         n = 0
         for i in r:
@@ -28,51 +25,58 @@ class Spotify:
 
     class parseprop(HTMLParser):
         def __init__(self):
-            print("spotify parse init")
+            # print("spotify parse init")
             HTMLParser.__init__(self)
-            self.ldjson=False
+            self.ldjson = False
 
         def handle_starttag(self, tag, attrs):
-            if (tag == "script" and ('type', 'application/ld+json') in attrs):
-            	self.ldjson=True
-            else: self.ldjson=False
+            if tag == "script" and ("type", "application/ld+json") in attrs:
+                self.ldjson = True
+            else:
+                self.ldjson = False
 
-        def handle_endtag(self,tag): self.ldjson=False
+        def handle_endtag(self, tag):
+            self.ldjson = False
 
-        def handle_data(self,data):
-        	if self.ldjson:
-        		Spotify.ldjson=data
-        		return
+        def handle_data(self, data):
+            if self.ldjson:
+                Spotify.ldjson = data
+                return
 
     def spotify(self, url):
         url = url.rstrip("\x01")
         p = self.parseprop()
         data = urlopen(url).read().decode()
         p.feed(data)
-        irc_string = "[\x0304Youtube\x03] \x0307ERROR:\x0308 got no data from server! \x0315(check your URL for typos!)\x03"
-        ansi_string = "[\x1b[31mYoutube\x1b[0m] \x1b[33;2mERROR:\x1b[33;1m got no data from server! \x1b[37;2m(check your URL for typos!)\x1b[0m"
-        data=(json.loads(Spotify.ldjson))
-        try: type=data["@type"]
+        irc_string = "[\x0304Spotify\x03] \x0307ERROR:\x0308 got no data from server! \x0315(check your URL for typos!)\x03"
+        ansi_string = "[\x1b[31mSpotify\x1b[0m] \x1b[33;2mERROR:\x1b[33;1m got no data from server! \x1b[37;2m(check your URL for typos!)\x1b[0m"
+        data = json.loads(Spotify.ldjson)
+        try:
+            type = data["@type"]
         except KeyError:
-          print(ansi_string)
-          return irc_string, True
-        id=data["@id"]
-        name=data["name"]
-        date=data["datePublished"]
-        artists=data["description"]
-        artists=artists.removeprefix(f'Listen to {name} on Spotify')
-        artists=artists.removeprefix('.').strip()
-        artists=artists.removeprefix('· ')
-        if artists.startswith("Song · "): artists=artists.removeprefix("Song · ")
+            print(ansi_string)
+            return irc_string, True
+        id = data["@id"]
+        name = data["name"]
+        date = data["datePublished"]
+        artists = data["description"]
+        artists = (
+            artists.removeprefix(f"Listen to {name} on Spotify")
+            .removeprefix(".")
+            .strip()
+            .removeprefix("· ")
+        )
+        if artists.startswith("Song · "):
+            artists = artists.removeprefix("Song · ")
         elif artists.startswith("Album · "):
-          artists=artists.removeprefix("Album · ")
-          artists=artists[::-1].split(" · ",1)[1][::-1] #removes the "10 songs" part from album
-        artists=artists.removesuffix(f" · {date[:4]}")  #remove the year too
-        #print(type,id,name,"|"+artists+"|",date)
+            artists = artists.removeprefix("Album · ")[::-1].split(" · ", 1)[1][
+                ::-1
+            ]  # removes the "10 songs" part from album
+        artists = artists.removesuffix(f" · {date[:4]}")  # remove the year too
         irc_string = f"[\x0303Spotify\x03] \x02{name}\x02 by \x1d{artists}\x1d published on {date}"
         ansi_string = f"[\x1b[32mSpotify\x1b[0m] \x1b[1m{name}\x1b[0m by \x1b[03m{artists}\x1b[0m published on {date}"
-        #print(("Song: " if type=="MusicRecording" else "Album: " if type=="MusicAlbum" else f"Unknown type ({type}): ")+'"'+name+'"'+" by "+'"'+artists+'"'+" released on "+date)
-        #irc_string="dummy"
+        # print(("Song: " if type=="MusicRecording" else "Album: " if type=="MusicAlbum" else f"Unknown type ({type}): ")+'"'+name+'"'+" by "+'"'+artists+'"'+" released on "+date)
+        # irc_string="dummy"
         print(ansi_string)
         return irc_string, False
 
diff --git a/youtube.py b/youtube.py
index 475a36d..41dd18b 100755
--- a/youtube.py
+++ b/youtube.py
@@ -114,13 +114,15 @@ class YouTube:
         self.video_type = (
             "clip"
             if self.is_clip(url)
-            else "shorts"
-            if self.is_ytshorts(url)
-            else "music"
-            if self.is_ytmusic(url)
-            else "embed"
-            if self.is_embed(url)
-            else "video"
+            else (
+                "shorts"
+                if self.is_ytshorts(url)
+                else (
+                    "music"
+                    if self.is_ytmusic(url)
+                    else "embed" if self.is_embed(url) else "video"
+                )
+            )
         )
         video_type = self.video_type
         if video_type == "embed":