From bc7dea7ad6ed64f998ffde9fe705f5ebddee81df Mon Sep 17 00:00:00 2001 From: samwiseg0 Date: Thu, 1 Nov 2018 16:01:53 -0400 Subject: [PATCH] Add option to toggle SSL verification for other scripts --- configuration.example.py | 3 +++ ombi.py | 13 ++++++++++--- radarr.py | 17 ++++++++++++++--- sonarr.py | 21 ++++++++++++++------- 4 files changed, 41 insertions(+), 13 deletions(-) diff --git a/configuration.example.py b/configuration.example.py index 492d500..89e6097 100644 --- a/configuration.example.py +++ b/configuration.example.py @@ -22,6 +22,7 @@ sonarr_server_list = [ #('https://sonarr3.domain.tld', 'xxxxxxxxxxxxxxx', '3') ] sonarr_influxdb_db_name = 'plex' +sonarr_verify_ssl = True ############################ RADARR CONFIG ############################ radarr_server_list = [ @@ -30,11 +31,13 @@ radarr_server_list = [ #('https://radarr3.domain.tld', 'xxxxxxxxxxxxxxx', '3') ] radarr_influxdb_db_name = 'plex' +radarr_verify_ssl = True ############################ OMBI CONFIG ############################## ombi_url = 'https://ombi.domain.tld' ombi_api_key = 'xxxxxxxxxxxxxxx' ombi_influxdb_db_name = 'plex' +ombi_verify_ssl = True ########################## TAUTULLI CONFIG ############################ tautulli_url = 'https://tautulli.domain.tld' diff --git a/ombi.py b/ombi.py index 5aa1812..f7a1f91 100644 --- a/ombi.py +++ b/ombi.py @@ -19,8 +19,13 @@ def influx_sender(influx_payload): influx.write_points(influx_payload) def get_total_requests(): - get_tv_requests = requests.get('{}/api/v1/Request/tv'.format(configuration.ombi_url), headers=headers).json() - get_movie_requests = requests.get('{}/api/v1/Request/movie'.format(configuration.ombi_url), headers=headers).json() + get_tv_requests = requests.get('{}/api/v1/Request/tv'.format( + configuration.ombi_url), headers=headers, + verify=configuration.ombi_verify_ssl).json() + + get_movie_requests = requests.get('{}/api/v1/Request/movie'.format( + configuration.ombi_url), headers=headers, + verify=configuration.ombi_verify_ssl).json() count_movie_requests = 0 count_tv_requests = 0 @@ -46,7 +51,9 @@ def get_total_requests(): return influx_payload def get_request_counts(): - get_request_counts = requests.get('{}/api/v1/Request/count'.format(configuration.ombi_url), headers=headers).json() + get_request_counts = requests.get('{}/api/v1/Request/count'.format( + configuration.ombi_url), headers=headers, + verify=configuration.ombi_verify_ssl).json() influx_payload = [ { diff --git a/radarr.py b/radarr.py index 2358a73..0f95d0b 100644 --- a/radarr.py +++ b/radarr.py @@ -27,7 +27,11 @@ def get_missing_movies(): for radarr_url, radarr_api_key, server_id in configuration.radarr_server_list: headers = {'X-Api-Key': radarr_api_key} - get_movies = requests.get('{}/api/movie'.format(radarr_url), headers=headers).json() + + get_movies = requests.get('{}/api/movie'.format(radarr_url), + headers=headers, + verify=configuration.radarr_verify_ssl).json() + movies = {d['tmdbId']: d for d in get_movies} for movie in movies.keys(): @@ -64,7 +68,11 @@ def get_missing_avl(): for radarr_url, radarr_api_key, server_id in configuration.radarr_server_list: headers = {'X-Api-Key': radarr_api_key} - get_movies = requests.get('{}/api/movie'.format(radarr_url), headers=headers).json() + + get_movies = requests.get('{}/api/movie'.format(radarr_url), + headers=headers, + verify=configuration.radarr_verify_ssl).json() + movies = {d['tmdbId']: d for d in get_movies} for movie in movies.keys(): @@ -103,7 +111,10 @@ def get_queue_movies(): for radarr_url, radarr_api_key, server_id in configuration.radarr_server_list: headers = {'X-Api-Key': radarr_api_key} - get_movies = requests.get('{}/api/queue'.format(radarr_url), headers=headers).json() + get_movies = requests.get('{}/api/queue'.format(radarr_url), + headers=headers, + verify=configuration.radarr_verify_ssl).json() + queue_movies = {d['id']: d for d in get_movies} for movie in queue_movies.keys(): diff --git a/sonarr.py b/sonarr.py index 8be5f6c..6b463a3 100644 --- a/sonarr.py +++ b/sonarr.py @@ -32,7 +32,8 @@ def get_all_missing_shows(): headers = {'X-Api-Key': sonarr_api_key} get_tv_shows = requests.get('{}/api/wanted/missing/?pageSize=1000'.format(sonarr_url), - headers=headers).json()['records'] + headers=headers, + verify=configuration.sonarr_verify_ssl).json()['records'] tv_shows = {d['id']: d for d in get_tv_shows} @@ -81,8 +82,10 @@ def get_missing_shows(days_past): headers = {'X-Api-Key': sonarr_api_key} - get_tv_shows = requests.get('{}/api/calendar/?start={}&end={}&pageSize=1000'.format(sonarr_url, last_days, today), - headers=headers).json() + get_tv_shows = requests.get('{}/api/calendar/?start={}&end={}&pageSize=1000' + .format(sonarr_url, last_days, today), + headers=headers, + verify=configuration.sonarr_verify_ssl).json() tv_shows = {d['id']: d for d in get_tv_shows} @@ -130,7 +133,8 @@ def get_upcoming_shows(): headers = {'X-Api-Key': sonarr_api_key} get_upcoming_shows = requests.get('{}/api/calendar/'.format(sonarr_url), - headers=headers).json() + headers=headers, + verify=configuration.sonarr_verify_ssl).json() upcoming_shows = {d['id']: d for d in get_upcoming_shows} @@ -181,8 +185,10 @@ def get_future_shows(future_days): headers = {'X-Api-Key': sonarr_api_key} - get_tv_shows = requests.get('{}/api/calendar/?start={}&end={}&pageSize=200'.format(sonarr_url, today, future), - headers=headers).json() + get_tv_shows = requests.get('{}/api/calendar/?start={}&end={}&pageSize=200' + .format(sonarr_url, today, future), + headers=headers, + verify=configuration.sonarr_verify_ssl).json() tv_shows = {d['id']: d for d in get_tv_shows} @@ -232,7 +238,8 @@ def get_queue_shows(): headers = {'X-Api-Key': sonarr_api_key} get_tv_shows = requests.get('{}/api/queue'.format(sonarr_url), - headers=headers).json() + headers=headers, + verify=configuration.sonarr_verify_ssl).json() tv_shows = {d['id']: d for d in get_tv_shows}