Rework script to add options & switches
This commit is contained in:
samwiseg0
parent
addb7c1a6d
commit
4eb6465b64
1 changed files with 304 additions and 60 deletions
364
sonarr.py
364
sonarr.py
|
|
@ -1,70 +1,314 @@
|
||||||
# Do not edit this script. Edit configuration.py
|
# Do not edit this script. Edit configuration.py
|
||||||
|
import sys
|
||||||
import requests
|
import requests
|
||||||
from datetime import datetime, timezone
|
from datetime import datetime, timezone, date, timedelta
|
||||||
from influxdb import InfluxDBClient
|
from influxdb import InfluxDBClient
|
||||||
|
import argparse
|
||||||
|
from argparse import RawTextHelpFormatter
|
||||||
import configuration
|
import configuration
|
||||||
|
|
||||||
current_time = datetime.now(timezone.utc).astimezone().isoformat()
|
|
||||||
headers = {'X-Api-Key': configuration.sonarr_api_key}
|
|
||||||
get_tv_shows = requests.get('{}/api/wanted/missing/?pageSize=1000'.format(configuration.sonarr_url),
|
|
||||||
headers=headers).json()['records']
|
|
||||||
tv_shows = {d['id']: d for d in get_tv_shows}
|
|
||||||
missing = []
|
|
||||||
influx_payload = []
|
|
||||||
|
|
||||||
for show in tv_shows.keys():
|
def now_iso():
|
||||||
seriesTitle = '{}'.format(tv_shows[show]['series']['title'])
|
now_iso = datetime.now(timezone.utc).astimezone().isoformat()
|
||||||
sxe = 'S{:0>2}E{:0>2}'.format(tv_shows[show]['seasonNumber'],tv_shows[show]['episodeNumber'])
|
return now_iso
|
||||||
missing.append((seriesTitle, sxe, tv_shows[show]['id'], tv_shows[show]['title']))
|
|
||||||
|
|
||||||
for seriesTitle, sxe, id, title in missing:
|
|
||||||
influx_payload.append(
|
|
||||||
{
|
|
||||||
"measurement": "Sonarr",
|
|
||||||
"tags": {
|
|
||||||
"type": "Missing",
|
|
||||||
"sonarrId": id
|
|
||||||
},
|
|
||||||
"time": current_time,
|
|
||||||
"fields": {
|
|
||||||
"name": seriesTitle,
|
|
||||||
"epname": title,
|
|
||||||
"sxe": sxe
|
|
||||||
}
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
get_upcoming_shows = requests.get('{}/api/calendar/'.format(configuration.sonarr_url),
|
|
||||||
headers=headers).json()
|
|
||||||
upcoming_shows = {d['id']: d for d in get_upcoming_shows}
|
|
||||||
upcoming = []
|
|
||||||
influx_payload2 = []
|
|
||||||
|
|
||||||
for show in upcoming_shows.keys():
|
def influx_sender(influx_payload):
|
||||||
seriesTitle = '{}'.format(upcoming_shows[show]['series']['title'])
|
influx = InfluxDBClient(configuration.influxdb_url, configuration.influxdb_port, configuration.influxdb_username,
|
||||||
sxe = 'S{:0>2}E{:0>2}'.format(upcoming_shows[show]['seasonNumber'],upcoming_shows[show]['episodeNumber'])
|
configuration.influxdb_password, configuration.sonarr_influxdb_db_name)
|
||||||
upcoming.append((seriesTitle, sxe, upcoming_shows[show]['id'], upcoming_shows[show]['title'], upcoming_shows[show]['airDate']))
|
influx.write_points(influx_payload)
|
||||||
|
|
||||||
for seriesTitle, sxe, id, title, airDate in upcoming:
|
|
||||||
influx_payload2.append(
|
|
||||||
{
|
|
||||||
"measurement": "Sonarr",
|
|
||||||
"tags": {
|
|
||||||
"type": "Soon",
|
|
||||||
"sonarrId": id
|
|
||||||
},
|
|
||||||
"time": current_time,
|
|
||||||
"fields": {
|
|
||||||
"name": seriesTitle,
|
|
||||||
"epname": title,
|
|
||||||
"sxe": sxe,
|
|
||||||
"airs": airDate
|
|
||||||
}
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
influx = InfluxDBClient(configuration.grafana_url, configuration.grafana_port, configuration.grafana_username,
|
def get_all_missing_shows():
|
||||||
configuration.grafana_password, configuration.sonarr_grafana_db_name)
|
# Set the time here so we have one timestamp to work with
|
||||||
influx.write_points(influx_payload)
|
now = now_iso()
|
||||||
influx.write_points(influx_payload2)
|
|
||||||
|
missing = []
|
||||||
|
|
||||||
|
influx_payload = []
|
||||||
|
|
||||||
|
for sonarr_url, sonarr_api_key, server_id in configuration.sonarr_server_list:
|
||||||
|
|
||||||
|
headers = {'X-Api-Key': sonarr_api_key}
|
||||||
|
|
||||||
|
get_tv_shows = requests.get('{}/api/wanted/missing/?pageSize=1000'.format(sonarr_url),
|
||||||
|
headers=headers).json()['records']
|
||||||
|
|
||||||
|
tv_shows = {d['id']: d for d in get_tv_shows}
|
||||||
|
|
||||||
|
|
||||||
|
for show in tv_shows.keys():
|
||||||
|
series_title = '{}'.format(tv_shows[show]['series']['title'])
|
||||||
|
sxe = 'S{:0>2}E{:0>2}'.format(tv_shows[show]['seasonNumber'],tv_shows[show]['episodeNumber'])
|
||||||
|
missing.append((series_title, sxe, tv_shows[show]['id'], tv_shows[show]['title']))
|
||||||
|
|
||||||
|
for series_title, sxe, id, title in missing:
|
||||||
|
influx_payload.append(
|
||||||
|
{
|
||||||
|
"measurement": "Sonarr",
|
||||||
|
"tags": {
|
||||||
|
"type": "Missing",
|
||||||
|
"sonarrId": id,
|
||||||
|
"server": server_id
|
||||||
|
},
|
||||||
|
"time": now,
|
||||||
|
"fields": {
|
||||||
|
"name": series_title,
|
||||||
|
"epname": title,
|
||||||
|
"sxe": sxe
|
||||||
|
}
|
||||||
|
}
|
||||||
|
)
|
||||||
|
# Empty missing or else things get foo bared
|
||||||
|
missing = []
|
||||||
|
|
||||||
|
return influx_payload
|
||||||
|
|
||||||
|
def get_missing_shows(days_past):
|
||||||
|
# Set the time here so we have one timestamp to work with
|
||||||
|
now = now_iso()
|
||||||
|
|
||||||
|
last_days = str(date.today()+timedelta(days=-days_past))
|
||||||
|
|
||||||
|
today = str(date.today())
|
||||||
|
|
||||||
|
missing = []
|
||||||
|
|
||||||
|
influx_payload = []
|
||||||
|
|
||||||
|
for sonarr_url, sonarr_api_key, server_id in configuration.sonarr_server_list:
|
||||||
|
|
||||||
|
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()
|
||||||
|
|
||||||
|
tv_shows = {d['id']: d for d in get_tv_shows}
|
||||||
|
|
||||||
|
for show in tv_shows.keys():
|
||||||
|
if not (tv_shows[show]['hasFile']):
|
||||||
|
series_title = '{}'.format(tv_shows[show]['series']['title'])
|
||||||
|
sxe = 'S{:0>2}E{:0>2}'.format(tv_shows[show]['seasonNumber'], tv_shows[show]['episodeNumber'])
|
||||||
|
air_date = (tv_shows[show]['airDate'])
|
||||||
|
missing.append((series_title, sxe, air_date, tv_shows[show]['id']))
|
||||||
|
|
||||||
|
for series_title, sxe, air_date, id in missing:
|
||||||
|
influx_payload.append(
|
||||||
|
{
|
||||||
|
"measurement": "Sonarr",
|
||||||
|
"tags": {
|
||||||
|
"type": "Missing_Days",
|
||||||
|
"sonarrId": id,
|
||||||
|
"server": server_id
|
||||||
|
},
|
||||||
|
"time": now,
|
||||||
|
"fields": {
|
||||||
|
"name": series_title,
|
||||||
|
"sxe": sxe,
|
||||||
|
"airs": air_date
|
||||||
|
}
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
# Empty missing or else things get foo bared
|
||||||
|
missing = []
|
||||||
|
|
||||||
|
return influx_payload
|
||||||
|
|
||||||
|
|
||||||
|
def get_upcoming_shows():
|
||||||
|
# Set the time here so we have one timestamp to work with
|
||||||
|
now = now_iso()
|
||||||
|
|
||||||
|
upcoming = []
|
||||||
|
|
||||||
|
influx_payload = []
|
||||||
|
|
||||||
|
for sonarr_url, sonarr_api_key, server_id in configuration.sonarr_server_list:
|
||||||
|
|
||||||
|
headers = {'X-Api-Key': sonarr_api_key}
|
||||||
|
|
||||||
|
get_upcoming_shows = requests.get('{}/api/calendar/'.format(sonarr_url),
|
||||||
|
headers=headers).json()
|
||||||
|
|
||||||
|
upcoming_shows = {d['id']: d for d in get_upcoming_shows}
|
||||||
|
|
||||||
|
for show in upcoming_shows.keys():
|
||||||
|
series_title = '{}'.format(upcoming_shows[show]['series']['title'])
|
||||||
|
sxe = 'S{:0>2}E{:0>2}'.format(upcoming_shows[show]['seasonNumber'],upcoming_shows[show]['episodeNumber'])
|
||||||
|
upcoming.append((series_title, sxe, upcoming_shows[show]['id'], upcoming_shows[show]['title'], upcoming_shows[show]['airDate']))
|
||||||
|
|
||||||
|
for series_title, sxe, id, title, air_date in upcoming:
|
||||||
|
influx_payload.append(
|
||||||
|
{
|
||||||
|
"measurement": "Sonarr",
|
||||||
|
"tags": {
|
||||||
|
"type": "Soon",
|
||||||
|
"sonarrId": id,
|
||||||
|
"server": server_id
|
||||||
|
},
|
||||||
|
"time": now,
|
||||||
|
"fields": {
|
||||||
|
"name": series_title,
|
||||||
|
"epname": title,
|
||||||
|
"sxe": sxe,
|
||||||
|
"airs": air_date
|
||||||
|
}
|
||||||
|
}
|
||||||
|
)
|
||||||
|
# Empty upcoming or else things get foo bared
|
||||||
|
upcoming = []
|
||||||
|
|
||||||
|
return influx_payload
|
||||||
|
|
||||||
|
|
||||||
|
def get_today_shows():
|
||||||
|
# Set the time here so we have one timestamp to work with
|
||||||
|
now = now_iso()
|
||||||
|
|
||||||
|
today = str(date.today())
|
||||||
|
|
||||||
|
tomorrow = str(date.today()+timedelta(days=1))
|
||||||
|
|
||||||
|
air_today = []
|
||||||
|
|
||||||
|
downloaded = []
|
||||||
|
|
||||||
|
influx_payload = []
|
||||||
|
|
||||||
|
|
||||||
|
for sonarr_url, sonarr_api_key, server_id in configuration.sonarr_server_list:
|
||||||
|
|
||||||
|
headers = {'X-Api-Key': sonarr_api_key}
|
||||||
|
|
||||||
|
get_tv_shows = requests.get('{}/api/calendar/?start={}&end={}&pageSize=50'.format(sonarr_url, today, tomorrow),
|
||||||
|
headers=headers).json()
|
||||||
|
|
||||||
|
tv_shows = {d['id']: d for d in get_tv_shows}
|
||||||
|
|
||||||
|
for show in tv_shows.keys():
|
||||||
|
series_title = '{}'.format(tv_shows[show]['series']['title'])
|
||||||
|
dl_status = int(tv_shows[show]['hasFile'])
|
||||||
|
sxe = 'S{:0>2}E{:0>2}'.format(tv_shows[show]['seasonNumber'], tv_shows[show]['episodeNumber'])
|
||||||
|
air_today.append((series_title, dl_status, sxe, tv_shows[show]['id']))
|
||||||
|
|
||||||
|
for series_title, dl_status, sxe, id in air_today:
|
||||||
|
influx_payload.append(
|
||||||
|
{
|
||||||
|
"measurement": "Sonarr",
|
||||||
|
"tags": {
|
||||||
|
"type": "Today",
|
||||||
|
"sonarrId": id,
|
||||||
|
"server": server_id
|
||||||
|
},
|
||||||
|
"time": now,
|
||||||
|
"fields": {
|
||||||
|
"name": series_title,
|
||||||
|
"sxe": sxe,
|
||||||
|
"downloaded": dl_status
|
||||||
|
}
|
||||||
|
}
|
||||||
|
)
|
||||||
|
# Empty today and downloaded or else things get foo bared
|
||||||
|
air_today = []
|
||||||
|
|
||||||
|
return influx_payload
|
||||||
|
|
||||||
|
|
||||||
|
def get_queue_shows():
|
||||||
|
# Set the time here so we have one timestamp to work with
|
||||||
|
now = now_iso()
|
||||||
|
|
||||||
|
queue = []
|
||||||
|
|
||||||
|
downloaded = []
|
||||||
|
|
||||||
|
influx_payload = []
|
||||||
|
|
||||||
|
|
||||||
|
for sonarr_url, sonarr_api_key, server_id in configuration.sonarr_server_list:
|
||||||
|
|
||||||
|
headers = {'X-Api-Key': sonarr_api_key}
|
||||||
|
|
||||||
|
get_tv_shows = requests.get('{}/api/queue'.format(sonarr_url),
|
||||||
|
headers=headers).json()
|
||||||
|
|
||||||
|
tv_shows = {d['id']: d for d in get_tv_shows}
|
||||||
|
|
||||||
|
for show in tv_shows.keys():
|
||||||
|
series_title = '{}'.format(tv_shows[show]['series']['title'])
|
||||||
|
protocol = (tv_shows[show]['protocol'].upper())
|
||||||
|
sxe = 'S{:0>2}E{:0>2}'.format(tv_shows[show]['episode']['seasonNumber'], tv_shows[show]['episode']['episodeNumber'])
|
||||||
|
if protocol == 'USENET':
|
||||||
|
protocol_id = 1
|
||||||
|
else:
|
||||||
|
protocol_id = 0
|
||||||
|
|
||||||
|
queue.append((series_title, protocol, protocol_id, sxe, tv_shows[show]['id']))
|
||||||
|
|
||||||
|
for series_title, protocol, protocol_id, sxe, id in queue:
|
||||||
|
influx_payload.append(
|
||||||
|
{
|
||||||
|
"measurement": "Sonarr",
|
||||||
|
"tags": {
|
||||||
|
"type": "Queue",
|
||||||
|
"sonarrId": id,
|
||||||
|
"server": server_id
|
||||||
|
|
||||||
|
},
|
||||||
|
"time": now,
|
||||||
|
"fields": {
|
||||||
|
"name": show,
|
||||||
|
"name": series_title,
|
||||||
|
"sxe": sxe,
|
||||||
|
"protocol": protocol,
|
||||||
|
"protocol_id": protocol_id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
# Empty missing or else things get foo bared
|
||||||
|
queue = []
|
||||||
|
|
||||||
|
return influx_payload
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
parser = argparse.ArgumentParser(prog='Sonarr stats operations',
|
||||||
|
description='Script to aid in data gathering from Sonarr', formatter_class=RawTextHelpFormatter)
|
||||||
|
|
||||||
|
parser.add_argument("--missing", action='store_true',
|
||||||
|
help='Get all missing TV shows')
|
||||||
|
|
||||||
|
parser.add_argument("--missing_days", type=int,
|
||||||
|
help='Get missing TV shows in X pass days')
|
||||||
|
|
||||||
|
parser.add_argument("--upcoming", action='store_true',
|
||||||
|
help='Get upcoming TV shows')
|
||||||
|
|
||||||
|
parser.add_argument("--today", action='store_true',
|
||||||
|
help='Get TV shows on today')
|
||||||
|
|
||||||
|
parser.add_argument("--queue", action='store_true',
|
||||||
|
help='Get movies in queue')
|
||||||
|
|
||||||
|
opts = parser.parse_args()
|
||||||
|
|
||||||
|
if opts.missing:
|
||||||
|
influx_sender(get_all_missing_shows())
|
||||||
|
|
||||||
|
elif opts.missing_days:
|
||||||
|
influx_sender(get_missing_shows(opts.missing_days))
|
||||||
|
|
||||||
|
elif opts.upcoming:
|
||||||
|
influx_sender(get_upcoming_shows())
|
||||||
|
|
||||||
|
elif opts.today:
|
||||||
|
influx_sender(get_today_shows())
|
||||||
|
|
||||||
|
elif opts.queue:
|
||||||
|
influx_sender(get_queue_shows())
|
||||||
|
|
||||||
|
elif len(sys.argv) == 1:
|
||||||
|
parser.print_help(sys.stderr)
|
||||||
|
sys.exit(1)
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue