parent
dce0dcfd23
commit
82c7704fab
@ -48,7 +48,7 @@ Taultulli > Settings > Notification Agents > New Script > Script Arguments:
|
|||||||
|
|
||||||
import requests
|
import requests
|
||||||
import argparse
|
import argparse
|
||||||
from datetime import datetime
|
from datetime import datetime, timedelta
|
||||||
import sys
|
import sys
|
||||||
import os
|
import os
|
||||||
from plexapi.server import PlexServer
|
from plexapi.server import PlexServer
|
||||||
@ -96,7 +96,7 @@ lib_dict = {x.title: x.key for x in plex.library.sections()}
|
|||||||
|
|
||||||
|
|
||||||
SELECTOR = ['watch', 'plays', 'time', 'limit']
|
SELECTOR = ['watch', 'plays', 'time', 'limit']
|
||||||
TODAY = datetime.today().strftime('%Y-%m-%d')
|
TODAY = datetime.now()
|
||||||
unix_time = int(ttime())
|
unix_time = int(ttime())
|
||||||
|
|
||||||
|
|
||||||
@ -171,7 +171,7 @@ def get_history(username, start_date=None, section_id=None):
|
|||||||
|
|
||||||
Optional
|
Optional
|
||||||
----------
|
----------
|
||||||
start_date : str "YYYY-MM-DD"
|
start_date : list ["YYYY-MM-DD", ...]
|
||||||
The date in history to search.
|
The date in history to search.
|
||||||
section_id : int
|
section_id : int
|
||||||
The libraries numeric identifier
|
The libraries numeric identifier
|
||||||
@ -186,7 +186,7 @@ def get_history(username, start_date=None, section_id=None):
|
|||||||
'user': username}
|
'user': username}
|
||||||
|
|
||||||
if start_date:
|
if start_date:
|
||||||
payload['start_date'] = TODAY
|
payload['start_date'] = start_date
|
||||||
|
|
||||||
if section_id:
|
if section_id:
|
||||||
payload['section_id '] = section_id
|
payload['section_id '] = section_id
|
||||||
@ -270,17 +270,24 @@ if __name__ == "__main__":
|
|||||||
parser.add_argument('--section', default=False, choices=lib_dict.keys(), metavar='',
|
parser.add_argument('--section', default=False, choices=lib_dict.keys(), metavar='',
|
||||||
help='Space separated list of case sensitive names to process. Allowed names are: \n'
|
help='Space separated list of case sensitive names to process. Allowed names are: \n'
|
||||||
'(choices: %(choices)s)')
|
'(choices: %(choices)s)')
|
||||||
parser.add_argument('--today', default=False, action='store_true',
|
parser.add_argument('--days', type=int, default=0, nargs='?',
|
||||||
help='Search history only for today. \n'
|
help='Search history limit. \n'
|
||||||
'Default: %(default)s')
|
'Default: %(default)s day(s) (today).')
|
||||||
parser.add_argument('--duration', type=int, default=0,
|
parser.add_argument('--duration', type=int, default=0,
|
||||||
help='Duration of item that triggered script agent.')
|
help='Duration of item that triggered script agent.')
|
||||||
|
|
||||||
opts = parser.parse_args()
|
opts = parser.parse_args()
|
||||||
|
|
||||||
|
history_lst = []
|
||||||
total_limit = 0
|
total_limit = 0
|
||||||
total_jbop = 0
|
total_jbop = 0
|
||||||
duration = 0
|
duration = 0
|
||||||
|
dates = []
|
||||||
|
delta = timedelta(days=opts.days)
|
||||||
|
|
||||||
|
for i in range(delta.days + 1):
|
||||||
|
day = TODAY + timedelta(days=-i)
|
||||||
|
dates.append(day.strftime('%Y-%m-%d'))
|
||||||
|
|
||||||
if opts.limit:
|
if opts.limit:
|
||||||
limit = dict(opts.limit)
|
limit = dict(opts.limit)
|
||||||
@ -308,18 +315,20 @@ if __name__ == "__main__":
|
|||||||
else:
|
else:
|
||||||
message = ''
|
message = ''
|
||||||
|
|
||||||
if opts.section:
|
for date in dates:
|
||||||
section_id = lib_dict[opts.section]
|
if opts.section:
|
||||||
history = get_history(username=opts.username, section_id=section_id, start_date=opts.today)
|
section_id = lib_dict[opts.section]
|
||||||
else:
|
history = get_history(username=opts.username, section_id=section_id, start_date=date)
|
||||||
history = get_history(username=opts.username, start_date=opts.today)
|
else:
|
||||||
|
history = get_history(username=opts.username, start_date=date)
|
||||||
|
history_lst.append(history)
|
||||||
|
|
||||||
if opts.jbop == 'watch':
|
if opts.jbop == 'watch':
|
||||||
total_jbop = sum([data['watched_status'] for data in history['data']])
|
total_jbop = sum([data['watched_status'] for history in history_lst for data in history['data']])
|
||||||
if opts.jbop == 'time':
|
if opts.jbop == 'time':
|
||||||
total_jbop = sum([data['duration'] for data in history['data']])
|
total_jbop = sum([data['duration'] for history in history_lst for data in history['data']])
|
||||||
if opts.jbop == 'plays':
|
if opts.jbop == 'plays':
|
||||||
total_jbop = history['recordsFiltered']
|
total_jbop = sum([history['recordsFiltered'] for history in history_lst])
|
||||||
|
|
||||||
if total_jbop:
|
if total_jbop:
|
||||||
if total_jbop > total_limit:
|
if total_jbop > total_limit:
|
||||||
|
Loading…
Reference in New Issue
Block a user