diff options
Diffstat (limited to 'modules/youtube.py')
| -rw-r--r-- | modules/youtube.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/modules/youtube.py b/modules/youtube.py index 3d075f64..20dc072b 100644 --- a/modules/youtube.py +++ b/modules/youtube.py @@ -24,6 +24,8 @@ class Module(object): help="Find a video on youtube", usage="[query]") events.on("received.message.channel").hook(self.channel_message) + events.on("get.youtubefromlastfm").hook(self.get_yt_from_lastfm) + exports.add("channelset", {"setting": "auto-youtube", "help": "Disable/Enable automatically getting info from " "youtube URLs", "validate": Utils.bool_or_none}) @@ -63,6 +65,22 @@ class Module(object): video_title, video_duration, video_uploader, "{:,}".format( int(video_views)), video_opinions, URL_YOUTUBESHORT % video_id) + def get_yt_from_lastfm(self, event): + search = event["query"] + video_id = "" + + search_page = Utils.get_url(URL_YOUTUBESEARCH, + get_params={"q": search, "part": "snippet", + "maxResults": "1", "type": "video", + "key": self.bot.config["google-api-key"]}, + json=True) + + if search_page: + if search_page["pageInfo"]["totalResults"] > 0: + video_id = search_page["items"][0]["id"]["videoId"] + return "https://youtu.be/%s" % video_id + + def yt(self, event): video_id = None search = None |