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

Fix lint issues

Browse source
This commit is contained in:
samwiseg0 2022-01-14 21:49:17 -05:00
parent 2607584eba
commit 533ec1058f
2 changed files with 14 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
varken/radarr.py
View file

@ -18,7 +18,7 @@ class RadarrAPI(object):
def __repr__(self): def __repr__(self):
return f"<radarr-{self.server.id}>" return f"<radarr-{self.server.id}>"
def get_movie(self,id): def get_movie(self, id):
endpoint = '/api/v3/movie/' endpoint = '/api/v3/movie/'
req = self.session.prepare_request(Request('GET', self.server.url + endpoint + str(id))) req = self.session.prepare_request(Request('GET', self.server.url + endpoint + str(id)))
@ -88,19 +88,19 @@ class RadarrAPI(object):
queueResponse = [] queueResponse = []
queue = [] queue = []
req = self.session.prepare_request(Request('GET', self.server.url + endpoint,params=params)) req = self.session.prepare_request(Request('GET', self.server.url + endpoint, params=params))
get = connection_handler(self.session, req, self.server.verify_ssl) get = connection_handler(self.session, req, self.server.verify_ssl)
if not get: if not get:
return return
response = QueuePages(**get) response = QueuePages(**get)
queueResponse.extend(response.records) queueResponse.extend(response.records)
while response.totalRecords > response.page * response.pageSize: while response.totalRecords > response.page * response.pageSize:
page = response.page + 1 page = response.page + 1
params = {'pageSize': pageSize, 'page': page} params = {'pageSize': pageSize, 'page': page}
req = self.session.prepare_request(Request('GET', self.server.url + endpoint,params=params)) req = self.session.prepare_request(Request('GET', self.server.url + endpoint, params=params))
get = connection_handler(self.session, req, self.server.verify_ssl) get = connection_handler(self.session, req, self.server.verify_ssl)
if not get: if not get:
return return
@ -152,4 +152,4 @@ class RadarrAPI(object):
} }
) )
self.dbmanager.write_points(influx_payload) self.dbmanager.write_points(influx_payload)

16
varken/sonarr.py
View file

@ -18,7 +18,7 @@ class SonarrAPI(object):
def __repr__(self): def __repr__(self):
return f"<sonarr-{self.server.id}>" return f"<sonarr-{self.server.id}>"
def get_series(self, id): def get_series(self, id):
endpoint = '/api/v3/series/' endpoint = '/api/v3/series/'
@ -27,14 +27,14 @@ class SonarrAPI(object):
if not get: if not get:
return return
return SonarrTVShow(**get) return SonarrTVShow(**get)
def get_episode(self, id): def get_episode(self, id):
endpoint = '/api/v3/episode' endpoint = '/api/v3/episode'
params = {'episodeIds': id} params = {'episodeIds': id}
req = self.session.prepare_request(Request('GET', self.server.url + endpoint,params = params)) req = self.session.prepare_request(Request('GET', self.server.url + endpoint, params=params))
get = connection_handler(self.session, req, self.server.verify_ssl) get = connection_handler(self.session, req, self.server.verify_ssl)
if not get: if not get:
@ -117,19 +117,19 @@ class SonarrAPI(object):
queueResponse = [] queueResponse = []
queue = [] queue = []
req = self.session.prepare_request(Request('GET', self.server.url + endpoint,params=params)) req = self.session.prepare_request(Request('GET', self.server.url + endpoint, params=params))
get = connection_handler(self.session, req, self.server.verify_ssl) get = connection_handler(self.session, req, self.server.verify_ssl)
if not get: if not get:
return return
response = QueuePages(**get) response = QueuePages(**get)
queueResponse.extend(response.records) queueResponse.extend(response.records)
while response.totalRecords > response.page * response.pageSize: while response.totalRecords > response.page * response.pageSize:
page = response.page + 1 page = response.page + 1
params = {'pageSize': pageSize, 'page': page} params = {'pageSize': pageSize, 'page': page}
req = self.session.prepare_request(Request('GET', self.server.url + endpoint,params=params)) req = self.session.prepare_request(Request('GET', self.server.url + endpoint, params=params))
get = connection_handler(self.session, req, self.server.verify_ssl) get = connection_handler(self.session, req, self.server.verify_ssl)
if not get: if not get:
return return
@ -189,4 +189,4 @@ class SonarrAPI(object):
if influx_payload: if influx_payload:
self.dbmanager.write_points(influx_payload) self.dbmanager.write_points(influx_payload)
else: else:
self.logger.debug("No data to send to influx for sonarr instance, discarding.") self.logger.debug("No data to send to influx for sonarr instance, discarding.")