diff --git a/platypush/plugins/media/chromecast.py b/platypush/plugins/media/chromecast.py index 32ba7f0488..ae8c80d206 100644 --- a/platypush/plugins/media/chromecast.py +++ b/platypush/plugins/media/chromecast.py @@ -1,5 +1,8 @@ +import re import pychromecast +from pychromecast.controllers.youtube import YouTubeController + from platypush.plugins import Plugin, action @@ -7,6 +10,11 @@ class MediaChromecastPlugin(Plugin): """ Plugin to interact with Chromecast devices + Supported formats: + + * YouTube URLs + * Plex (through ``media.plex`` plugin, experimental) + Requires: * **pychromecast** (``pip install pychromecast``) @@ -67,16 +75,19 @@ class MediaChromecastPlugin(Plugin): if chromecast not in self.chromecasts: - self.chromecasts = pychromecast.get_chromecasts() - cast = next(cc for cc in self.chromecasts - if cc.device.friendly_name == chromecast) - else: - cast = self.chromecasts[chromecast] + self.chromecasts = { + cast.device.friendly_name: cast + for cast in pychromecast.get_chromecasts() + } + + if chromecast not in self.chromecasts: + raise RuntimeError('Device {} not found'.format(chromecast)) + + return self.chromecasts[chromecast] - return cast @action - def cast_media(self, media, content_type, chromecast=None, title=None, + def cast_media(self, media, content_type=None, chromecast=None, title=None, image_url=None, autoplay=True, current_time=0, stream_type=STREAM_TYPE_BUFFERED, subtitles=None, subtitles_lang='en-US', subtitles_mime='text/vtt', @@ -121,10 +132,29 @@ class MediaChromecastPlugin(Plugin): :type subtitle_id: int """ + if not chromecast: + chromecast = self.chromecast + cast = self.get_chromecast(chromecast) cast.wait() + mc = cast.media_controller - mc.namespace = 'urn:x-cast:com.google.cast.sse' + yt = self._get_youtube_url(media) + + if yt: + self.logger.info('Playing YouTube video {} on {}'.format( + yt, chromecast)) + + hndl = YouTubeController() + cast.register_handler(hndl) + hndl.update_screen_id() + return hndl.play_video(yt) + + if not content_type: + raise RuntimeError('content_type required to process media {}'. + format(media)) + + self.logger.info('Playing {} on {}'.format(media, chromecast)) mc.play_media(media, content_type, title=title, thumb=image_url, current_time=current_time, autoplay=autoplay, @@ -134,6 +164,20 @@ class MediaChromecastPlugin(Plugin): mc.block_until_active() + + @classmethod + def _get_youtube_url(cls, url): + m = re.match('https?://www.youtube.com/watch\?v=([^&]+).*', url) + if m: + return m.group(1) + + m = re.match('youtube:video:(.*)', url) + if m: + return m.group(1) + + return None + + @action def disconnect(self, chromecast=None, timeout=None, blocking=True): """