Marius/JBOPS
1
0
Fork 0
mirror of https://github.com/blacktwin/JBOPS.git synced 2025-12-16 13:05:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

multiple filter value support for shows

Browse Source
This commit is contained in:
Blacktwin 2019-05-28 14:09:09 -04:00
parent 2a2a9f74ff
commit 5ddd981f2f
1 changed files with 24 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
fun/playlist_manager.py
View File

@ -145,7 +145,7 @@ account = plex.myPlexAccount()
user_lst = [x.title for x in plex.myPlexAccount().users() if x.servers] user_lst = [x.title for x in plex.myPlexAccount().users() if x.servers]
sections = plex.library.sections() sections = plex.library.sections()
sections_dict = {x.key: x.title for x in sections} sections_dict = {x.key: x.title for x in sections}
filter_lst = list(set([y for x in sections if x.type != 'photo' for y in x.ALLOWED_FILTERS])) filters_lst = list(set([y for x in sections if x.type != 'photo' for y in x.ALLOWED_FILTERS]))
playlist_lst = [x.title for x in plex.playlists()] playlist_lst = [x.title for x in plex.playlists()]
today = datetime.datetime.now().date() today = datetime.datetime.now().date()
weeknum = datetime.date(today.year, today.month, today.day).isocalendar()[1] weeknum = datetime.date(today.year, today.month, today.day).isocalendar()[1]
@ -296,10 +296,10 @@ def multi_filter_search(keyword_dict, library):
for value in values: for value in values:
search_dict = {} search_dict = {}
search_dict[key] = value search_dict[key] = value
search_lst = [movie.ratingKey for movie in library.all(**search_dict)] search_lst = [item.ratingKey for item in library.all(**search_dict)]
multi_lst += search_lst multi_lst += search_lst
else: else:
multi_lst += [movie.ratingKey for movie in library.all(**{key: values})] multi_lst += [item.ratingKey for item in library.all(**{key: values})]
counts = Counter(multi_lst) counts = Counter(multi_lst)
# Use amount of keywords to check that all keywords were found in results # Use amount of keywords to check that all keywords were found in results
search_lst = [id for id in multi_lst if counts[id] == keyword_count] search_lst = [id for id in multi_lst if counts[id] == keyword_count]
@ -366,10 +366,24 @@ def get_content(libraries, jbop, filters=None, search=None, limit=None):
search_lst += [episode.ratingKey] search_lst += [episode.ratingKey]
child_lst += search_lst child_lst += search_lst
if filters: if filters:
for show in plex_library.search(**filters): # Update filters for tagged filtered keys
for episode in show.episodes(): for key, value in filters.items():
filter_lst += [episode.ratingKey] # Genre needs special handling
child_lst += filter_lst if key == "genre":
del filters[key]
filters[key + tags] = value
for key, value in filters.items():
# Only genre filtering should allow multiple values and allow for AND statement
if key.endswith(tags):
shows_lst = multi_filter_search({key: value}, plex_library)
else:
shows_lst = [show for show in plex_library.search(**{key: value})]
for showkey in shows_lst:
show = plex.fetchItem(showkey)
for episode in show.episodes():
filter_lst += [episode.ratingKey]
child_lst += filter_lst
if keywords and filters: if keywords and filters:
child_lst += list(set(filter_lst) & set(search_lst)) child_lst += list(set(filter_lst) & set(search_lst))
else: else:
@ -678,7 +692,7 @@ if __name__ == "__main__":
help='Limit the amount items to be added to a playlist.') help='Limit the amount items to be added to a playlist.')
parser.add_argument('--filter', action='append', type=lambda kv: kv.split("="), parser.add_argument('--filter', action='append', type=lambda kv: kv.split("="),
help='Search filtered metadata fields.\n' help='Search filtered metadata fields.\n'
'Filters: ({}).'.format(', '.join(filter_lst))) 'Filters: ({}).'.format(', '.join(filters_lst)))
parser.add_argument('--search', action='append', type=lambda kv: kv.split("="), parser.add_argument('--search', action='append', type=lambda kv: kv.split("="),
help='Search non-filtered metadata fields for keywords ' help='Search non-filtered metadata fields for keywords '
'in title, summary, etc.') 'in title, summary, etc.')
@ -713,9 +727,9 @@ if __name__ == "__main__":
if "," in v: if "," in v:
filters[k] = v.split(",") filters[k] = v.split(",")
# Check if provided filter exist, exit if it doesn't exist # Check if provided filter exist, exit if it doesn't exist
if not (set(filters.keys()) & set(filter_lst)): if not (set(filters.keys()) & set(filters_lst)):
print('({}) was not found in filters list: [{}]' print('({}) was not found in filters list: [{}]'
.format(' '.join(filters.keys()), ', '.join(filter_lst))) .format(' '.join(filters.keys()), ', '.join(filters_lst)))
exit() exit()
# Defining users # Defining users