Add seasons to search query

This commit is contained in:
Jonathan Wong 2015-09-24 22:09:14 -07:00
parent 25249e7538
commit 28d05ba9fe
2 changed files with 18 additions and 9 deletions

View file

@ -803,21 +803,23 @@ class DataFactory(object):
media_type = item['media_type']
elif str(item['parent_rating_key']) == rating_key:
title = item['parent_title']
if item['media_type'] == episode:
if item['media_type'] == 'episode':
title = item['grandparent_title']
media_type = 'season'
elif item['media_type'] == track:
elif item['media_type'] == 'track':
title = item['parent_title']
media_type = 'album'
elif str(item['grandparent_rating_key']) == rating_key:
if item['media_type'] == 'episode':
title = item['grandparent_title']
if item['media_type'] == episode:
media_type = 'show'
elif item['media_type'] == track:
elif item['media_type'] == 'track':
title = item['grandparent_title']
media_type = 'artist'
if title and media_type:
query = {'title': title,
query = {'title': title.replace('"', ''),
'media_type': media_type,
'rating_key': rating_key}
else:

View file

@ -1421,8 +1421,15 @@ class PmsConnect(object):
metadata = self.get_metadata_details(rating_key=rating_key)
if metadata['metadata']['type'] == 'show':
search_results_list['show'].append(metadata['metadata'])
elif metadata['metadata']['type'] == 'season':
show_seasons = self.get_item_children(rating_key=metadata['metadata']['rating_key'])
if show_seasons['children_count'] != '0':
for season in show_seasons['children_list']:
if season['rating_key']:
rating_key = season['rating_key']
metadata = self.get_metadata_details(rating_key=rating_key)
search_results_list['season'].append(metadata['metadata'])
elif metadata['metadata']['type'] == 'artist':
search_results_list['artist'].append(metadata['metadata'])
elif metadata['metadata']['type'] == 'album':