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

support for searching through episodes

Browse Source 
using Tautulli data for episode gathering
This commit is contained in:
blacktwin 2020-05-19 18:08:22 -04:00
parent edfa7315a0
commit f38f1ee7af
1 changed files with 40 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
utility/media_manager.py
View File

@ -130,12 +130,20 @@ class Metadata(object):
if self.media_type == 'show': if self.media_type == 'show':
show = plex.fetchItem(int(self.rating_key)) show = plex.fetchItem(int(self.rating_key))
# todo only using the first library location for show types
self.file = show.locations[0] self.file = show.locations[0]
show = tautulli_server.get_new_rating_keys(self.rating_key, self.media_type)
seasons = show['0']['children']
episodes = []
show_size = [] show_size = []
self.episodes = show.episodes() for season in seasons.values():
for episode in self.episodes: for _episode in season['children'].values():
show_size.append(episode.media[0].parts[0].size) metadata = tautulli_server.get_metadata(_episode['rating_key'])
episode = Metadata(metadata)
show_size.append(int(episode.file_size))
episodes.append(episode)
self.file_size = sum(show_size) self.file_size = sum(show_size)
self.episodes = episodes
class User(object): class User(object):
@ -234,6 +242,11 @@ class Tautulli:
else: else:
return [d for d in library_stats['data']] return [d for d in library_stats['data']]
def get_new_rating_keys(self, rating_key, media_type):
"""Call Tautulli's get_new_rating_keys api endpoint."""
payload = {"rating_key": rating_key, "media_type": media_type}
return self._call_api('get_new_rating_keys', payload)
def sizeof_fmt(num, suffix='B'): def sizeof_fmt(num, suffix='B'):
# Function found https://stackoverflow.com/a/1094933 # Function found https://stackoverflow.com/a/1094933
@ -342,25 +355,28 @@ def size_work(sectionID, operator, value, episodes):
tt_size = tautulli_server.get_library_media_info(section_id=sectionID, tt_size = tautulli_server.get_library_media_info(section_id=sectionID,
start=start, length=count, start=start, length=count,
order_column="file_size") order_column="file_size")
if all([tt_size]):
start += count start += count
for item in tt_size: for item in tt_size:
_meta = tautulli_server.get_metadata(item['rating_key']) _meta = tautulli_server.get_metadata(item['rating_key'])
metadata = Metadata(_meta) metadata = Metadata(_meta)
try:
if episodes: if episodes:
for _episode in metadata.episodes: for _episode in metadata.episodes:
file_size = _episode.media[0].parts[0].size file_size = int(_episode.file_size)
if operator(file_size, value): if operator(file_size, value):
size_lst.append(_episode) size_lst.append(_episode)
else: else:
file_size = int(metadata.file_size) file_size = int(metadata.file_size)
if operator(file_size, value): if operator(file_size, value):
size_lst.append(metadata) size_lst.append(metadata)
else: except AttributeError:
return size_lst print("Metadata error found with rating_key: ({})".format(item['rating_key']))
continue continue
elif not all([tt_size]):
break
start += count
return size_lst
def watched_work(user, sectionID=None, ratingKey=None): def watched_work(user, sectionID=None, ratingKey=None):
@ -446,6 +462,8 @@ if __name__ == '__main__':
'">_5G" ie. items greater than 5 gigabytes.\n' '">_5G" ie. items greater than 5 gigabytes.\n'
'">_3" ie. items greater than 3 stars.\n' '">_3" ie. items greater than 3 stars.\n'
'">_3" ie. items played transcoded more than 3 times.') '">_3" ie. items played transcoded more than 3 times.')
parser.add_argument('--episodes', action='store_true',
help='Enable Plex to scan episodes if Show library is selected.')
opts = parser.parse_args() opts = parser.parse_args()
# todo find: watched by list of users[x], unwatched based on time[x], based on size, most transcoded, star rating # todo find: watched by list of users[x], unwatched based on time[x], based on size, most transcoded, star rating
@ -573,7 +591,7 @@ if __name__ == '__main__':
sizes = [] sizes = []
for item in size_lst: for item in size_lst:
added_at = datetime.datetime.utcfromtimestamp(float(item.added_at)).strftime("%Y-%m-%d") added_at = datetime.datetime.utcfromtimestamp(float(item.added_at)).strftime("%Y-%m-%d")
size = int(item.file_size) if item.file_size else '' size = int(item.file_size) if item.file_size else 0
sizes.append(size) sizes.append(size)
print(u"\t{} added {}\tSize: {}\n\t\tFile: {}".format( print(u"\t{} added {}\tSize: {}\n\t\tFile: {}".format(
item.title, added_at, sizeof_fmt(size), item.file)) item.title, added_at, sizeof_fmt(size), item.file))