forked from platypush/platypush
62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
from platypush.context import get_plugin
|
|
from platypush.plugins.media.search import MediaSearcher
|
|
|
|
|
|
class PlexMediaSearcher(MediaSearcher):
|
|
def search(self, query, **kwargs):
|
|
"""
|
|
Performs a Plex search using the configured :class:`platypush.plugins.media.plex.MediaPlexPlugin` instance if
|
|
it is available.
|
|
"""
|
|
|
|
try:
|
|
plex = get_plugin('media.plex')
|
|
except RuntimeError:
|
|
return []
|
|
|
|
self.logger.info('Searching Plex for "{}"'.format(query))
|
|
results = []
|
|
|
|
for result in plex.search(title=query).output:
|
|
results.extend(self._flatten_result(result))
|
|
|
|
self.logger.info('{} Plex results found for the search query "{}"'.format(len(results), query))
|
|
return results
|
|
|
|
@staticmethod
|
|
def _flatten_result(result):
|
|
def parse_part(media, part, episode=None, sub_media=None):
|
|
if 'episodes' in media:
|
|
del media['episodes']
|
|
|
|
return {
|
|
**{k: v for k, v in result.items() if k not in ['media', 'type']},
|
|
'media_type': result.get('type'),
|
|
'type': 'plex',
|
|
**{k: v for k, v in media.items() if k not in ['parts']},
|
|
**part,
|
|
'title': '{}{}{}'.format(
|
|
result.get('title', ''),
|
|
' [{}]'.format(episode['season_episode']) if (episode or {}).get('season_episode') else '',
|
|
' {}'.format(sub_media['title']) if (sub_media or {}).get('title') else '',
|
|
),
|
|
'summary': episode['summary'] if (episode or {}).get('summary') else media.get('summary'),
|
|
}
|
|
|
|
results = []
|
|
|
|
for media in result.get('media', []):
|
|
if 'episodes' in media:
|
|
for episode in media['episodes']:
|
|
for sub_media in episode.get('media', []):
|
|
for part in sub_media.get('parts', []):
|
|
results.append(parse_part(media=media, episode=episode, sub_media=sub_media, part=part))
|
|
else:
|
|
for part in media.get('parts', []):
|
|
results.append(parse_part(media=media, part=part))
|
|
|
|
return results
|
|
|
|
|
|
# vim:sw=4:ts=4:et:
|