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

refactor ombi and add logging for bad requests so we can see data

Browse source
This commit is contained in:
Nicholas St. Germain 2019-04-19 12:44:07 -05:00
parent c08ce39ed3
commit ed07d69cdc
1 changed files with 25 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
varken/ombi.py
View file

@ -25,27 +25,38 @@ class OmbiAPI(object):
tv_req = self.session.prepare_request(Request('GET', self.server.url + tv_endpoint)) tv_req = self.session.prepare_request(Request('GET', self.server.url + tv_endpoint))
movie_req = self.session.prepare_request(Request('GET', self.server.url + movie_endpoint)) movie_req = self.session.prepare_request(Request('GET', self.server.url + movie_endpoint))
get_tv = connection_handler(self.session, tv_req, self.server.verify_ssl) get_tv = connection_handler(self.session, tv_req, self.server.verify_ssl) or []
get_movie = connection_handler(self.session, movie_req, self.server.verify_ssl) get_movie = connection_handler(self.session, movie_req, self.server.verify_ssl) or []
if not any([get_tv, get_movie]): if not any([get_tv, get_movie]):
self.logger.error('No json replies. Discarding job') self.logger.error('No json replies. Discarding job')
return return
if get_movie:
movie_request_count = len(get_movie) movie_request_count = len(get_movie)
else:
movie_request_count = 0
if get_tv:
tv_request_count = len(get_tv) tv_request_count = len(get_tv)
else:
tv_request_count = 0
tv_show_requests = []
for show in get_tv:
try: try:
tv_show_requests = [OmbiTVRequest(**show) for show in get_tv] tv_show_requests.append(OmbiTVRequest(**show))
except TypeError as e: except TypeError as e:
self.logger.error('TypeError has occurred : %s while creating OmbiTVRequest structure', e) self.logger.error('TypeError has occurred : %s while creating OmbiTVRequest structure for show. '
return 'data attempted is: %s', e, show)
movie_requests = []
for movie in get_movie:
try: try:
movie_requests = [OmbiMovieRequest(**movie) for movie in get_movie] movie_requests.append(OmbiMovieRequest(**movie))
except TypeError as e: except TypeError as e:
self.logger.error('TypeError has occurred : %s while creating OmbiMovieRequest structure', e) self.logger.error('TypeError has occurred : %s while creating OmbiMovieRequest structure for movie. '
return 'data attempted is: %s', e, movie)
influx_payload = [ influx_payload = [
{ {