added hashing to radarr

This commit is contained in:
Nicholas St. Germain 2018-12-02 22:19:05 -06:00
parent 4f1ec17538
commit 8ef4b7fd0f
3 changed files with 17 additions and 8 deletions

View file

@ -2,7 +2,7 @@ from requests import Session
from datetime import datetime, timezone
from Varken.logger import logging
from Varken.helpers import Movie, Queue
from Varken.helpers import Movie, Queue, hashit
class RadarrAPI(object):
@ -34,6 +34,7 @@ class RadarrAPI(object):
missing.append((movie_name, ma, movie.tmdbId))
for title, ma, mid in missing:
hash_id = hashit('{}{}{}'.format(self.server.id, title, mid))
influx_payload.append(
{
"measurement": "Radarr",
@ -41,11 +42,12 @@ class RadarrAPI(object):
"Missing": True,
"Missing_Available": ma,
"tmdbId": mid,
"server": self.server.id
"server": self.server.id,
"name": title
},
"time": self.now,
"fields": {
"name": title
"hash": hash_id
}
}
)
@ -77,20 +79,22 @@ class RadarrAPI(object):
protocol_id, queue_item.id))
for movie, quality, protocol, protocol_id, qid in queue:
hash_id = hashit('{}{}{}'.format(self.server.id, movie, quality))
influx_payload.append(
{
"measurement": "Radarr",
"tags": {
"type": "Queue",
"tmdbId": qid,
"server": self.server.id
},
"time": self.now,
"fields": {
"server": self.server.id,
"name": movie,
"quality": quality,
"protocol": protocol,
"protocol_id": protocol_id
},
"time": self.now,
"fields": {
"hash": hash_id
}
}
)

View file

@ -77,7 +77,11 @@ class SonarrAPI(object):
for show in tv_shows:
sxe = 'S{:0>2}E{:0>2}'.format(show.seasonNumber, show.episodeNumber)
air_days.append((show.series['title'], show.hasFile, sxe, show.title, show.airDate, show.id))
if show.hasFile:
downloaded = 1
else:
downloaded = 0
air_days.append((show.series['title'], downloaded, sxe, show.title, show.airDate, show.id))
for series_title, dl_status, sxe, episode_title, air_date, sonarr_id in air_days:
hash_id = hashit('{}{}{}'.format(self.server.id, series_title, sxe))

View file

@ -1,6 +1,7 @@
from datetime import datetime, timezone
from geoip2.errors import AddressNotFoundError
from requests import Session
from Varken.helpers import TautulliStream, geo_lookup
from Varken.logger import logging