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
|
||||
import sys
|
||||
import requests
|
||||
from datetime import datetime, timezone
|
||||
from datetime import datetime, timezone, date, timedelta
|
||||
from influxdb import InfluxDBClient
|
||||
|
||||
import argparse
|
||||
from argparse import RawTextHelpFormatter
|
||||
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():
|
||||
seriesTitle = '{}'.format(tv_shows[show]['series']['title'])
|
||||
sxe = 'S{:0>2}E{:0>2}'.format(tv_shows[show]['seasonNumber'],tv_shows[show]['episodeNumber'])
|
||||
missing.append((seriesTitle, sxe, tv_shows[show]['id'], tv_shows[show]['title']))
|
||||
def now_iso():
|
||||
now_iso = datetime.now(timezone.utc).astimezone().isoformat()
|
||||
return now_iso
|
||||
|
||||
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():
|
||||
seriesTitle = '{}'.format(upcoming_shows[show]['series']['title'])
|
||||
sxe = 'S{:0>2}E{:0>2}'.format(upcoming_shows[show]['seasonNumber'],upcoming_shows[show]['episodeNumber'])
|
||||
upcoming.append((seriesTitle, sxe, upcoming_shows[show]['id'], upcoming_shows[show]['title'], upcoming_shows[show]['airDate']))
|
||||
def influx_sender(influx_payload):
|
||||
influx = InfluxDBClient(configuration.influxdb_url, configuration.influxdb_port, configuration.influxdb_username,
|
||||
configuration.influxdb_password, configuration.sonarr_influxdb_db_name)
|
||||
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,
|
||||
configuration.grafana_password, configuration.sonarr_grafana_db_name)
|
||||
influx.write_points(influx_payload)
|
||||
influx.write_points(influx_payload2)
|
||||
def get_all_missing_shows():
|
||||
# Set the time here so we have one timestamp to work with
|
||||
now = now_iso()
|
||||
|
||||
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