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'] media_type = item['media_type']
elif str(item['parent_rating_key']) == rating_key: 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' media_type = 'season'
elif item['media_type'] == track: elif item['media_type'] == 'track':
title = item['parent_title']
media_type = 'album' media_type = 'album'
elif str(item['grandparent_rating_key']) == rating_key: elif str(item['grandparent_rating_key']) == rating_key:
title = item['grandparent_title'] if item['media_type'] == 'episode':
if item['media_type'] == episode: title = item['grandparent_title']
media_type = 'show' media_type = 'show'
elif item['media_type'] == track: elif item['media_type'] == 'track':
title = item['grandparent_title']
media_type = 'artist' media_type = 'artist'
if title and media_type: if title and media_type:
query = {'title': title, query = {'title': title.replace('"', ''),
'media_type': media_type, 'media_type': media_type,
'rating_key': rating_key} 'rating_key': rating_key}
else: else:

View file

@ -1421,8 +1421,15 @@ class PmsConnect(object):
metadata = self.get_metadata_details(rating_key=rating_key) metadata = self.get_metadata_details(rating_key=rating_key)
if metadata['metadata']['type'] == 'show': if metadata['metadata']['type'] == 'show':
search_results_list['show'].append(metadata['metadata']) search_results_list['show'].append(metadata['metadata'])
elif metadata['metadata']['type'] == 'season':
search_results_list['season'].append(metadata['metadata']) 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': elif metadata['metadata']['type'] == 'artist':
search_results_list['artist'].append(metadata['metadata']) search_results_list['artist'].append(metadata['metadata'])
elif metadata['metadata']['type'] == 'album': elif metadata['metadata']['type'] == 'album':