rework logic, add more informative output, and update.

This commit is contained in:
dirtycajunrice 2018-12-19 18:21:54 -06:00
parent aa6017374e
commit 6d418e54f7

View File

@ -2,13 +2,14 @@
""" """
Description: Purge Tautulli users that no longer exist as a friend in Plex Description: Purge Tautulli users that no longer exist as a friend in Plex
Author: DirtyCajunRice Author: DirtyCajunRice
Requires: requests, plexapi Requires: requests, plexapi, python3.6+
""" """
import requests from requests import Session
from json.decoder import JSONDecodeError
from plexapi.myplex import MyPlexAccount from plexapi.myplex import MyPlexAccount
TAUTULLI_BASE_URL = '' TAUTULLI_URL = ''
TAUTULLI_API_KEY = '' TAUTULLI_API_KEY = ''
PLEX_USERNAME = '' PLEX_USERNAME = ''
@ -20,22 +21,27 @@ BACKUP_DB = True
# Do not edit past this line # # Do not edit past this line #
account = MyPlexAccount(PLEX_USERNAME, PLEX_PASSWORD) account = MyPlexAccount(PLEX_USERNAME, PLEX_PASSWORD)
payload = {'apikey': TAUTULLI_API_KEY, 'cmd': 'get_user_names'} session = Session()
tautulli_users = requests.get('http://{}/api/v2' session.params = {'apikey': TAUTULLI_API_KEY}
.format(TAUTULLI_BASE_URL), params=payload).json()['response']['data'] formatted_url = f'{TAUTULLI_URL}/api/v2'
request = session.get(formatted_url, params={'cmd': 'get_user_names'})
tautulli_users = None
try:
tautulli_users = request.json()['response']['data']
except JSONDecodeError:
exit("Error talking to Tautulli API, please check your TAUTULLI_URL")
plex_friend_ids = [friend.id for friend in account.users()] plex_friend_ids = [friend.id for friend in account.users()]
tautulli_user_ids = [user['user_id'] for user in tautulli_users] removed_users = [user for user in tautulli_users if user['user_id'] not in plex_friend_ids]
removed_user_ids = [user_id for user_id in tautulli_user_ids if user_id not in plex_friend_ids]
if BACKUP_DB: if BACKUP_DB:
payload['cmd'] = 'backup_db' backup = session.get(formatted_url, params={'cmd': 'backup_db'})
backup = requests.get('http://{}/api/v2'.format(TAUTULLI_BASE_URL), params=payload)
if removed_user_ids: if removed_users:
payload['cmd'] = 'delete_user' for user in removed_users:
removed_user = session.get(formatted_url, params={'cmd': 'delete_user', 'user_id': user['user_id']})
for user_id in removed_user_ids: print(f"Removed {user['friendly_name']} from Tautulli")
payload['user_id'] = user_id else:
remove_user = requests.get('http://{}/api/v2'.format(TAUTULLI_BASE_URL), params=payload) print('No users to remove')