platypush/platypush/plugins/media/plex.py
2018-11-13 01:29:24 +01:00

464 lines
12 KiB
Python

from plexapi.myplex import MyPlexAccount
from plexapi.video import Movie, Show
from platypush.context import get_plugin
from platypush.plugins import Plugin, action
class MediaPlexPlugin(Plugin):
"""
Plugin to interact with a Plex media server
Requires:
* **plexapi** (``pip install plexapi``)
"""
def __init__(self, server, username, password, *args, **kwargs):
"""
:param server: Plex server name
:type server: str
:param username: Plex username
:type username: str
:param password: Plex password
:type username: str
"""
super().__init__(*args, **kwargs)
self.resource = MyPlexAccount(username, password).resource(server)
self._plex = None
@property
def plex(self):
if not self._plex:
self._plex = self.resource.connect()
return self._plex
@action
def get_clients(self):
"""
Get the list of active clients
"""
return [{
'device': c.device,
'device_class': c.deviceClass,
'local': c.local,
'model': c.model,
'platform': c.platform,
'platform_version': c.platformVersion,
'product': c.product,
'state': c.state,
'title': c.title,
'version': c.version,
} for c in self.plex.clients()]
def _get_client(self, name):
return self.plex.client(name)
@action
def search(self, section=None, title=None, **kwargs):
"""
Return all the items matching the search criteria (default: all library items)
:param section: Section to search (Movies, Shows etc.)
:type section: str
:param title: Full or partial title
:type title: str
:param kwargs: Search criteria - includes e.g. title, unwatched, director, genre etc.
:type kwargs: dict
"""
ret = []
library = self.plex.library
if section:
library = library.section(section)
if title or kwargs:
items = library.search(title, **kwargs)
else:
items = library.all()
for item in items:
ret.append(self._flatten_item(item))
return ret
@action
def playlists(self):
"""
Get the playlists on the server
"""
return [
{
'title': pl.title,
'duration': pl.duration,
'summary': pl.summary,
'viewed_at': pl.viewedAt,
'items': [ self._flatten_item(item) for item in pl.items() ],
} for pl in self.plex.playlists()
]
@action
def history(self):
"""
Get the history of items played on the server
"""
return [
self._flatten_item(item) for item in self.plex.history()
]
def get_chromecast(self, chromecast):
from .lib.plexcast import PlexController
hndl = PlexController()
hndl.namespace = 'urn:x-cast:com.google.cast.sse'
cast = get_plugin('media.chromecast').get_chromecast(chromecast)
cast.register_handler(hndl)
return (cast, hndl)
@action
def play(self, client=None, chromecast=None, **kwargs):
"""
Search and play content on a client or a Chromecast. If no search filter
is specified, a play event will be sent to the specified client.
NOTE: Adding and managing play queues through the Plex API isn't fully
supported yet, therefore in case multiple items are returned from the
search only the first one will be played.
:param client: Client name
:type client: str
:param chromecast: Chromecast name
:type chromecast: str
:param kwargs: Search filter (e.g. title, section, unwatched, director etc.)
:type kwargs: dict
"""
if not client and not chromecast:
raise RuntimeError('No client nor chromecast specified')
if client:
client = plex.client(client)
elif chromecast:
(chromecast, handler) = self.get_chromecast(chromecast)
if not kwargs:
if client:
return client.play()
elif chromecast:
return handler.play()
if 'section' in kwargs:
library = self.plex.library.section(kwargs.pop('section'))
else:
library = self.plex.library
results = library.search(**kwargs)
if not results:
self.logger.info('No results for {}'.format(kwargs))
return
item = results[0]
self.logger.info('Playing {} on {}'.format(item.title, client or chromecast))
if client:
return client.playMedia(item)
elif chromecast:
return handler.play_media(item, self.plex)
@action
def pause(self, client):
"""
Send pause event to a client
"""
return self.client(client).pause()
@action
def stop(self, client):
"""
Send stop event to a client
"""
return self.client(client).stop()
@action
def seek(self, client, offset):
"""
Send seek event to a client
"""
return self.client(client).seekTo(offset)
@action
def forward(self, client):
"""
Forward playback on a client
"""
return self.client(client).stepForward()
@action
def back(self, client):
"""
Backward playback on a client
"""
return self.client(client).stepBack()
@action
def next(self, client):
"""
Play next item on a client
"""
return self.client(client).skipNext()
@action
def previous(self, client):
"""
Play previous item on a client
"""
return self.client(client).skipPrevious()
@action
def set_volume(self, client, volume):
"""
Set the volume on a client between 0 and 100
"""
return self.client(client).setVolume(volume/100)
@action
def repeat(self, client, repeat):
"""
Set the repeat status on a client
"""
return self.client(client).setRepeat(repeat)
@action
def random(self, client, random):
"""
Set the random status on a client
"""
return self.client(client).setShuffle(random)
@action
def up(self, client):
"""
Send an up key event to a client
"""
return self.client(client).moveUp()
@action
def down(self, client):
"""
Send a down key event to a client
"""
return self.client(client).moveDown()
@action
def left(self, client):
"""
Send a left key event to a client
"""
return self.client(client).moveLeft()
@action
def right(self, client):
"""
Send a right key event to a client
"""
return self.client(client).moveRight()
@action
def go_back(self, client):
"""
Send a back key event to a client
"""
return self.client(client).goBack()
@action
def go_home(self, client):
"""
Send a home key event to a client
"""
return self.client(client).goHome()
@action
def go_to_media(self, client):
"""
Send a go to media event to a client
"""
return self.client(client).goToMedia()
@action
def go_to_music(self, client):
"""
Send a go to music event to a client
"""
return self.client(client).goToMusic()
@action
def next_letter(self, client):
"""
Send a next letter event to a client
"""
return self.client(client).nextLetter()
@action
def page_down(self, client):
"""
Send a page down event to a client
"""
return self.client(client).pageDown()
@action
def page_up(self, client):
"""
Send a page up event to a client
"""
return self.client(client).pageUp()
def _flatten_item(self, item):
_item = {
'summary': item.summary,
'title': item.title,
'type': item.type,
'genres': [g.tag for g in getattr(item, 'genres', [])],
'art': getattr(item, 'art', None),
'art_url': getattr(item, 'artUrl', None),
'rating': getattr(item, 'rating', None),
'content_rating': getattr(item, 'content_rating', None),
}
if isinstance(item, Movie):
_item['is_watched'] = item.isWatched
_item['view_offset'] = item.viewOffset
_item['view_count'] = item.viewCount
_item['year'] = item.year
_item['audience_rating'] = item.audienceRating
_item['countries'] = [c.tag for c in item.countries]
_item['media'] = [
{
'duration': (item.media[i].duration or 0)/1000,
'width': item.media[i].width,
'height': item.media[i].height,
'audio_channels': item.media[i].audioChannels,
'audio_codec': item.media[i].audioCodec,
'video_codec': item.media[i].videoCodec,
'video_resolution': item.media[i].videoResolution,
'video_frame_rate': item.media[i].videoFrameRate,
'parts': [
{
'file': part.file,
'duration': (part.duration or 0)/1000,
} for part in item.media[i].parts
]
} for i in range(0, len(item.media))
]
elif isinstance(item, Show):
_item['media'] = [
{
'title': season.title,
'season_number': season.seasonNumber,
'summary': season.summary,
'episodes': [
{
'duration': episode.duration/1000,
'index': episode.index,
'year': episode.year,
'season_number': episode.seasonNumber,
'season_episode': episode.seasonEpisode,
'summary': episode.summary,
'is_watched': episode.isWatched,
'view_count': episode.viewCount,
'view_offset': episode.viewOffset,
'media': [
{
'duration': episode.media[i].duration/1000,
'width': episode.media[i].width,
'height': episode.media[i].height,
'audio_channels': episode.media[i].audioChannels,
'audio_codec': episode.media[i].audioCodec,
'video_codec': episode.media[i].videoCodec,
'video_resolution': episode.media[i].videoResolution,
'video_frame_rate': episode.media[i].videoFrameRate,
'title': episode.title,
'parts': [
{
'file': part.file,
'duration': part.duration/1000,
} for part in episode.media[i].parts
]
} for i in range(0, len(episode.locations))
]
} for episode in season.episodes()
]
} for season in item.seasons()
]
return _item
# vim:sw=4:ts=4:et: