Edm0nd/edmond/plugins/youtube_parser.py

59 lines
1.6 KiB
Python

import re
try:
from googleapiclient.errors import Error as GoogleApiError
DEPENDENCIES_FOUND = True
except ImportError:
DEPENDENCIES_FOUND = False
from edmond.plugin import Plugin
class YoutubeParserPlugin(Plugin):
VIDEO_URL_RE = re.compile(
r"https:\/\/www\.youtube\.com\/watch\?(?:.*&)?v=([^&]+)"
)
def __init__(self, bot):
super().__init__(bot)
self.priority = -3
self._youtube_plugin = None
@property
def youtube_plugin(self):
if self._youtube_plugin is None:
self._youtube_plugin = self.bot.get_plugin("youtube")
return self._youtube_plugin
def on_welcome(self, _):
if not (self.youtube_plugin and self.youtube_plugin.is_ready):
self.log_w("Youtube plugin is not available.")
self.is_ready = False
def on_pubmsg(self, event):
match = self.VIDEO_URL_RE.match(event.arguments[0])
if match:
self.tell_video_title(match.group(1), event.target)
return True
return False
def tell_video_title(self, video_id, target):
try:
search_response = self.youtube_plugin.youtube.videos().list(
id=video_id,
part="snippet",
).execute()
except GoogleApiError:
self.signal_failure(target)
return
title = ""
for result in search_response.get("items", []):
if result["kind"] == "youtube#video":
title = result["snippet"]["title"]
break
else:
self.signal_failure(target)
return
self.bot.say(target, title)