d_mcknight/Varken
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update sonarr.py

Browse source 
added ability to track upcoming episodes (https://picload.org/image/dogapgwa/1.png)
slightly modified missing episodes to show additional data (episode name) also added leading 0 to reflect S00E00 format to help my OCD. 

feel free to use what you like!
This commit is contained in:
Will Longo 2018-04-17 14:28:02 -05:00 committed by GitHub
parent 72cdc8baac
commit 5e94e6976b
1 changed files with 37 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
sonarr.py
View file

@ -14,11 +14,11 @@ missing = []
influx_payload = [] influx_payload = []
for show in tv_shows.keys(): for show in tv_shows.keys():
name = '{} - S{}E{}'.format(tv_shows[show]['series']['title'], tv_shows[show]['seasonNumber'], seriesTitle = '{}'.format(tv_shows[show]['series']['title'])
tv_shows[show]['episodeNumber']) sxe = 'S{:0>2} - E{:0>2}'.format(tv_shows[show]['seasonNumber'],tv_shows[show]['episodeNumber'])
missing.append((name, tv_shows[show]['id'])) missing.append((seriesTitle, sxe, tv_shows[show]['id'], tv_shows[show]['title']))
for show, id in missing: for seriesTitle, sxe, id, title in missing:
influx_payload.append( influx_payload.append(
{ {
"measurement": "Sonarr", "measurement": "Sonarr",
@ -28,7 +28,38 @@ for show, id in missing:
}, },
"time": current_time, "time": current_time,
"fields": { "fields": {
"name": show "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']))
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
} }
} }
) )
@ -36,4 +67,4 @@ for show, id in missing:
influx = InfluxDBClient(configuration.grafana_url, configuration.grafana_port, configuration.grafana_username, influx = InfluxDBClient(configuration.grafana_url, configuration.grafana_port, configuration.grafana_username,
configuration.grafana_password, configuration.sonarr_grafana_db_name) configuration.grafana_password, configuration.sonarr_grafana_db_name)
influx.write_points(influx_payload) influx.write_points(influx_payload)
influx.write_points(influx_payload2)