complete redo
use removeFromContinueWatching action instead of hack job
This commit is contained in:
blacktwin
parent
bf8bd61e17
commit
08f872b35c
@ -1,56 +1,36 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
"""Removes Shows from On Deck.
|
||||
"""Removes Shows from Continue Watching.
|
||||
|
||||
Author: Blacktwin
|
||||
Requires: requests, plexapi
|
||||
|
||||
Example:
|
||||
python off_deck.py --action deck --user Steve
|
||||
- Display what shows are on Steve's On Deck
|
||||
|
||||
python off_deck.py --action deck --shows "The Simpsons" Seinfeld
|
||||
- The Simpsons and Seinfeld will be removed from On Deck
|
||||
|
||||
python off_deck.py --action deck --shows "The Simpsons" Seinfeld --user Steve
|
||||
- The Simpsons and Seinfeld will be removed from Steve's On Deck
|
||||
|
||||
python off_deck.py --action deck --playlist "Favorite Shows!"
|
||||
- Any Show found in Favorite Shows playlist will be remove
|
||||
from On Deck
|
||||
|
||||
python off_deck.py --action watch --user Steve
|
||||
python off_deck.py
|
||||
- Display what shows are on admin's Continue Watching
|
||||
|
||||
python off_deck.py --user Steve
|
||||
- Display what shows are on Steve's Continue Watching
|
||||
|
||||
python off_deck.py --action watch --shows "The Simpsons" Seinfeld
|
||||
- The Simpsons and Seinfeld will be removed from Continue Watching
|
||||
python off_deck.py --shows "The Simpsons" Seinfeld
|
||||
- The Simpsons and Seinfeld Episodes will be removed from admin's Continue Watching
|
||||
|
||||
python off_deck.py --action watch --shows "The Simpsons" Seinfeld --user Steve
|
||||
- The Simpsons and Seinfeld will be removed from Steve's Continue Watching
|
||||
python off_deck.py --user Steve --shows "The Simpsons" Seinfeld
|
||||
- The Simpsons and Seinfeld Episodes will be removed from Steve's Continue Watching
|
||||
|
||||
python off_deck.py --action watch --playlist "Favorite Shows!"
|
||||
- Any Show found in Favorite Shows playlist will be remove
|
||||
from Continue Watching
|
||||
|
||||
!!!NOTICE!!!
|
||||
|
||||
* This script should be used for Shows that you have already watched and
|
||||
don't want showing up on your On Deck.
|
||||
|
||||
* For episodes of show already watched the view count will be reset back to it
|
||||
original.
|
||||
|
||||
* For episodes of show not watched the view count will be set to 1.
|
||||
python off_deck.py --playlists "Favorite Shows!"
|
||||
- Any Episode found in admin's "Favorite Shows" playlist will be remove from Continue Watching
|
||||
|
||||
python off_deck.py --user Steve --playlists "Favorite Shows!" SleepMix
|
||||
- Any Episode found in Steve's "Favorite Shows" or SleepMix playlist will be remove from Continue Watching
|
||||
|
||||
"""
|
||||
from __future__ import print_function
|
||||
from __future__ import unicode_literals
|
||||
|
||||
from builtins import range
|
||||
import requests
|
||||
import argparse
|
||||
import datetime
|
||||
from plexapi.server import PlexServer, CONFIG
|
||||
|
||||
PLEX_URL = ''
|
||||
@ -77,106 +57,25 @@ plex = PlexServer(PLEX_URL, PLEX_TOKEN, session=sess)
|
||||
account = plex.myPlexAccount()
|
||||
|
||||
|
||||
def actions():
|
||||
"""
|
||||
deck - Items that are On Deck
|
||||
watch - Items that are Continue Watching
|
||||
"""
|
||||
return ['deck', 'watch']
|
||||
|
||||
|
||||
def get_con_watch(server, off_deck=None):
|
||||
"""
|
||||
|
||||
Parameters
|
||||
----------
|
||||
server : class
|
||||
User's server to pull On Deck list
|
||||
Returns
|
||||
-------
|
||||
dict
|
||||
Show, episodes, and episodes view count
|
||||
|
||||
"""
|
||||
con_watch = []
|
||||
for item in server.library.onDeck():
|
||||
if off_deck and item.type == 'episode' and item.viewOffset > 0:
|
||||
if item.grandparentTitle in off_deck:
|
||||
print('Removing {}: S{:02}E{:02} {} from Continue Watching '
|
||||
'by marking watched.'.format(
|
||||
item.grandparentTitle.encode('UTF-8'),
|
||||
int(item.parentIndex), int(item.index),
|
||||
item.title.encode('UTF-8')))
|
||||
item.markWatched()
|
||||
else:
|
||||
if item.type == 'episode' and item.viewOffset > 0:
|
||||
con_watch.append(item)
|
||||
|
||||
if con_watch:
|
||||
print('The following shows are marked Continue Watching:')
|
||||
for item in con_watch:
|
||||
offset = datetime.timedelta(milliseconds=item.viewOffset)
|
||||
print('{}: S{:02}E{:02} {} ({})'.format(
|
||||
item.grandparentTitle.encode('UTF-8'),
|
||||
int(item.parentIndex), int(item.index),
|
||||
item.title.encode('UTF-8'), offset))
|
||||
|
||||
|
||||
def get_on_deck(server, off_deck=None):
|
||||
"""
|
||||
|
||||
Parameters
|
||||
----------
|
||||
server : class
|
||||
User's server to pull On Deck list
|
||||
off_deck : list
|
||||
List of Shows to remove from On Deck
|
||||
|
||||
Returns
|
||||
-------
|
||||
dict
|
||||
Show, episodes, and episodes view count
|
||||
|
||||
"""
|
||||
watched_statuses = {}
|
||||
on_deck = []
|
||||
for item in server.library.onDeck():
|
||||
if off_deck and item.type == 'episode' and item.viewOffset == 0:
|
||||
if item.grandparentTitle in off_deck:
|
||||
grandparent = server.fetchItem(item.grandparentRatingKey)
|
||||
watched_statuses['grandparent'] = grandparent
|
||||
watched_statuses['episodes'] = []
|
||||
for episode in grandparent.episodes():
|
||||
watched_statuses['episodes'].append({
|
||||
'object': episode,
|
||||
'viewCount': episode.viewCount})
|
||||
else:
|
||||
if item.type == 'episode':
|
||||
on_deck.append(item)
|
||||
|
||||
if on_deck:
|
||||
watched_statuses['on_deck'] = on_deck
|
||||
return watched_statuses
|
||||
def remove_from_cw(server, ratingKey):
|
||||
key = '/actions/removeFromContinueWatching?ratingKey=%s&' % ratingKey
|
||||
server.query(key, method=server._session.put)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
parser = argparse.ArgumentParser(description="Remove items from On Deck.",
|
||||
parser = argparse.ArgumentParser(description="Remove items from Continue Watching.",
|
||||
formatter_class=argparse.RawTextHelpFormatter)
|
||||
parser.add_argument('--shows', nargs='+',
|
||||
help='Shows to be removed from On Deck.')
|
||||
help='Shows to be removed from Continue Watching.')
|
||||
parser.add_argument('--user', nargs='?',
|
||||
help='User whose On Deck will be modified.')
|
||||
parser.add_argument('--playlist', nargs='?',
|
||||
help='Shows in playlist to be removed from On Deck')
|
||||
parser.add_argument('--action', required=True, choices=actions(),
|
||||
help='Action selector.'
|
||||
'{}'.format(actions.__doc__))
|
||||
help='User whose Continue Watching will be modified.')
|
||||
parser.add_argument('--playlists', nargs='+',
|
||||
help='Shows in playlist to be removed from Continue Watching')
|
||||
|
||||
opts = parser.parse_args()
|
||||
|
||||
ep_list = []
|
||||
to_remove = ''
|
||||
to_remove = []
|
||||
|
||||
if opts.user:
|
||||
user_acct = account.user(opts.user)
|
||||
@ -184,43 +83,25 @@ if __name__ == '__main__':
|
||||
else:
|
||||
plex_server = plex
|
||||
|
||||
if opts.shows and not opts.playlist:
|
||||
to_remove = opts.shows
|
||||
elif not opts.shows and opts.playlist:
|
||||
to_remove = [x.grandparentTitle for x in plex_server.playlist(opts.playlist).items()]
|
||||
onDeck = [item for item in plex_server.library.onDeck() if item.type == 'episode']
|
||||
|
||||
if opts.action == 'deck':
|
||||
if not to_remove:
|
||||
print('The following shows are On Deck...')
|
||||
on_deck = get_on_deck(plex_server)['on_deck']
|
||||
for item in on_deck:
|
||||
print('{}: S{:02}E{:02} {}'.format(
|
||||
item.grandparentTitle.encode('UTF-8'),
|
||||
int(item.parentIndex), int(item.index),
|
||||
item.title.encode('UTF-8')))
|
||||
if opts.shows and not opts.playlists:
|
||||
for show in opts.shows:
|
||||
searched_show = plex_server.search(show, mediatype='show')[0]
|
||||
if searched_show.title == show:
|
||||
to_remove += searched_show.episodes()
|
||||
elif not opts.shows and opts.playlists:
|
||||
for pl in plex_server.playlists():
|
||||
if pl.title in opts.playlists:
|
||||
to_remove += pl.items()
|
||||
else:
|
||||
for item in onDeck:
|
||||
print('{}: S{:02}E{:02} {}'.format(item.grandparentTitle, int(item.parentIndex),
|
||||
int(item.index), item.title))
|
||||
|
||||
else:
|
||||
print('Finding listed shows On Deck...')
|
||||
while True:
|
||||
off_deck = get_on_deck(plex_server, to_remove)
|
||||
if off_deck:
|
||||
ep_list += off_deck['episodes']
|
||||
print('Marking {} Unwatched'.format(off_deck['grandparent']
|
||||
.title.encode('UTF-8')))
|
||||
off_deck['grandparent'].markUnwatched()
|
||||
else:
|
||||
break
|
||||
|
||||
print('Resetting watch counts...')
|
||||
for item in ep_list:
|
||||
print('Resetting view count for {}: S{:02}E{:02} {}'.format(
|
||||
item['object'].grandparentTitle.encode('UTF-8'),
|
||||
int(item['object'].parentIndex), int(item['object'].index),
|
||||
item['object'].title.encode('UTF-8')))
|
||||
# if viewCount was 0 then make 1 so as not to return to On Deck.
|
||||
for _ in range(item['viewCount'] if item['viewCount'] != 0 else 1):
|
||||
item['object'].markWatched()
|
||||
|
||||
elif opts.action == 'watch':
|
||||
print('Finding shows marked Continue Watching...')
|
||||
get_con_watch(plex_server, to_remove)
|
||||
for item in onDeck:
|
||||
if item in to_remove:
|
||||
print('Removing {}: S{:02}E{:02} {} from Continue Watching'.format(
|
||||
item.grandparentTitle, int(item.parentIndex), int(item.index), item.title))
|
||||
# item.removeFromContinueWatching()
|
||||
remove_from_cw(plex_server, item.ratingKey)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user