diff options
Diffstat (limited to 'youtube.py')
| -rwxr-xr-x | youtube.py | 68 |
1 files changed, 50 insertions, 18 deletions
diff --git a/youtube.py b/youtube.py index a75f52b..f15613e 100755 --- a/youtube.py +++ b/youtube.py @@ -4,7 +4,14 @@ from urllib.error import HTTPError from urllib.parse import urlencode, urlparse, parse_qs from json import loads as json_loads + class YouTube: + def __init__(self): + try: + YouTube.prefer_playlist = YouTube.prefer_playlist + except AttributeError: + YouTube.prefer_playlist = False + def mesg(self, msg, t=None): self.util.mesg(msg, t) @@ -21,8 +28,6 @@ class YouTube: or "https://music.youtube.com/watch?v=" in i or "https://youtube.com/shorts/" in i or "https://www.youtube.com/shorts/" in i - or "https://www.youtube.com/clip/" in i - or "https://youtube.com/clip/" in i ] r = list(dict.fromkeys(r)) n = 0 @@ -33,20 +38,36 @@ class YouTube: return r - def is_embed(str): + def is_embed(self, *str): + if type(self) == type("a"): + str = self + else: + str = str[0] return str.startswith("https://www.youtube.com/embed/") or str.startswith( "https://www.youtube-nocookie.com/embed/" ) - def is_ytmusic(str): + def is_ytmusic(self, *str): + if type(self) == type("a"): + str = self + else: + str = str[0] return str.startswith("https://music.youtube.com/watch?v=") - def is_ytshorts(str): + def is_ytshorts(self, *str): + if type(self) == type("a"): + str = self + else: + str = str[0] return str.startswith("https://youtube.com/shorts/") or str.startswith( "https://www.youtube.com/shorts/" ) - def is_clip(str): + def is_clip(self, *str): + if type(self) == type("a"): + str = self + else: + str = str[0] return str.startswith("https://youtube.com/clip/") or str.startswith( "https://www.youtube.com/clip/" ) @@ -68,26 +89,34 @@ class YouTube: videoId = url.split("?")[0].split("/")[-1] url = f"https://www.youtube.com/watch?v={videoId}" url = urlparse(url) - qs=parse_qs(url.query);video_id=qs['v'][0] - try: playlist_id=qs['list'][0] - except KeyError: playlist_id=None - if self.prefer_playlist and playlist_id: - url = url.scheme + "://" + url.netloc + "/playlist?list=" + playlist_id + qs = parse_qs(url.query) + try: + video_id = qs["v"][0] + except KeyError: + video_id = None + try: + playlist_id = qs["list"][0] + except KeyError: + playlist_id = None + if (self.prefer_playlist and playlist_id) or (playlist_id and not video_id): + url = url.scheme + "://" + url.netloc + "/playlist?list=" + playlist_id else: - url = url.scheme + "://" + url.netloc + url.path + "?v=" + video_id + url = url.scheme + "://" + url.netloc + url.path + "?v=" + video_id url = f"https://www.youtube.com/oembed?{urlencode([('url',url),('format','json')])}" try: - print(url," and ",playlist_id) + #print(url, " and ", playlist_id) data = urlopen(url).read().decode() data = json_loads(data) - title=data['title'] - channelName=data['author_name'] + title = data["title"] + channelName = data["author_name"] except HTTPError as e: irc_string = f"[\x0304Youtube\x03] \x0307ERROR:\x0308 {e} \x0315\x03" ansi_string = f"[\x1b[31mYoutube\x1b[0m] \x1b[33;2mERROR:\x1b[33;1m {e} \x1b[37;2m\x1b[0m" print(ansi_string) return irc_string, True - irc_string = f"[\x0303Youtube\x03] \x02{title}\x02 uploaded by \x1d{channelName}\x1d" + irc_string = ( + f"[\x0303Youtube\x03] \x02{title}\x02 uploaded by \x1d{channelName}\x1d" + ) ansi_string = f"[\x1b[32mYoutube\x1b[0m] \x1b[1m{title}\x1b[0m uploaded by \x1b[03m{channelName}\x1b[0m" print(ansi_string) return irc_string, False @@ -96,5 +125,8 @@ class YouTube: if __name__ == "__main__": import sys - YouTube.prefer_playlist=False - YouTube.yt(YouTube, sys.argv[1]) + #if url is a video that's part of a playlist, + #return playlist (True) or video (False, default)? + #YouTube.prefer_playlist=False + YouTube().yt(sys.argv[1]) + #YouTube.yt(YouTube, sys.argv[1]) |