since init uses only one server, moved headers/params to init
This commit is contained in:
parent
76427e582e
commit
d7d4535dc3
5 changed files with 11 additions and 16 deletions
|
@ -1,5 +1,6 @@
|
||||||
from influxdb import InfluxDBClient
|
from influxdb import InfluxDBClient
|
||||||
|
|
||||||
|
|
||||||
class DBManager(object):
|
class DBManager(object):
|
||||||
def __init__(self, server):
|
def __init__(self, server):
|
||||||
self.server = server
|
self.server = server
|
||||||
|
|
|
@ -4,6 +4,7 @@ from datetime import datetime, timezone
|
||||||
from Varken.helpers import OmbiRequestCounts
|
from Varken.helpers import OmbiRequestCounts
|
||||||
from Varken.logger import logging
|
from Varken.logger import logging
|
||||||
|
|
||||||
|
|
||||||
class OmbiAPI(object):
|
class OmbiAPI(object):
|
||||||
def __init__(self, server, dbmanager):
|
def __init__(self, server, dbmanager):
|
||||||
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
||||||
|
|
|
@ -12,16 +12,16 @@ class RadarrAPI(object):
|
||||||
self.server = server
|
self.server = server
|
||||||
# Create session to reduce server web thread load, and globally define pageSize for all requests
|
# Create session to reduce server web thread load, and globally define pageSize for all requests
|
||||||
self.session = Session()
|
self.session = Session()
|
||||||
|
self.session.headers = {'X-Api-Key': self.server.api_key}
|
||||||
|
|
||||||
@logging
|
@logging
|
||||||
def get_missing(self):
|
def get_missing(self):
|
||||||
endpoint = '/api/movie'
|
endpoint = '/api/movie'
|
||||||
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
||||||
influx_payload = []
|
influx_payload = []
|
||||||
|
|
||||||
missing = []
|
missing = []
|
||||||
headers = {'X-Api-Key': self.server.api_key}
|
|
||||||
get = self.session.get(self.server.url + endpoint, headers=headers, verify=self.server.verify_ssl).json()
|
get = self.session.get(self.server.url + endpoint, verify=self.server.verify_ssl).json()
|
||||||
movies = [Movie(**movie) for movie in get]
|
movies = [Movie(**movie) for movie in get]
|
||||||
|
|
||||||
for movie in movies:
|
for movie in movies:
|
||||||
|
@ -59,10 +59,8 @@ class RadarrAPI(object):
|
||||||
endpoint = '/api/queue'
|
endpoint = '/api/queue'
|
||||||
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
||||||
influx_payload = []
|
influx_payload = []
|
||||||
|
|
||||||
queue = []
|
queue = []
|
||||||
headers = {'X-Api-Key': self.server.api_key}
|
get = self.session.get(self.server.url + endpoint, verify=self.server.verify_ssl).json()
|
||||||
get = self.session.get(self.server.url + endpoint, headers=headers, verify=self.server.verify_ssl).json()
|
|
||||||
for movie in get:
|
for movie in get:
|
||||||
movie['movie'] = Movie(**movie['movie'])
|
movie['movie'] = Movie(**movie['movie'])
|
||||||
download_queue = [Queue(**movie) for movie in get]
|
download_queue = [Queue(**movie) for movie in get]
|
||||||
|
|
|
@ -14,6 +14,7 @@ class SonarrAPI(object):
|
||||||
self.server = server
|
self.server = server
|
||||||
# Create session to reduce server web thread load, and globally define pageSize for all requests
|
# Create session to reduce server web thread load, and globally define pageSize for all requests
|
||||||
self.session = Session()
|
self.session = Session()
|
||||||
|
self.session.headers = {'X-Api-Key': self.server.api_key}
|
||||||
self.session.params = {'pageSize': 1000}
|
self.session.params = {'pageSize': 1000}
|
||||||
|
|
||||||
@logging
|
@logging
|
||||||
|
@ -24,10 +25,8 @@ class SonarrAPI(object):
|
||||||
params = {'start': last_days, 'end': self.today}
|
params = {'start': last_days, 'end': self.today}
|
||||||
influx_payload = []
|
influx_payload = []
|
||||||
missing = []
|
missing = []
|
||||||
headers = {'X-Api-Key': self.server.api_key}
|
|
||||||
|
|
||||||
get = self.session.get(self.server.url + endpoint, params=params, headers=headers,
|
get = self.session.get(self.server.url + endpoint, params=params, verify=self.server.verify_ssl).json()
|
||||||
verify=self.server.verify_ssl).json()
|
|
||||||
# Iteratively create a list of TVShow Objects from response json
|
# Iteratively create a list of TVShow Objects from response json
|
||||||
tv_shows = [TVShow(**show) for show in get]
|
tv_shows = [TVShow(**show) for show in get]
|
||||||
|
|
||||||
|
@ -68,11 +67,9 @@ class SonarrAPI(object):
|
||||||
future = str(date.today() + timedelta(days=self.server.future_days))
|
future = str(date.today() + timedelta(days=self.server.future_days))
|
||||||
influx_payload = []
|
influx_payload = []
|
||||||
air_days = []
|
air_days = []
|
||||||
headers = {'X-Api-Key': self.server.api_key}
|
|
||||||
params = {'start': self.today, 'end': future}
|
params = {'start': self.today, 'end': future}
|
||||||
|
|
||||||
get = self.session.get(self.server.url + endpoint, params=params, headers=headers,
|
get = self.session.get(self.server.url + endpoint, params=params, verify=self.server.verify_ssl).json()
|
||||||
verify=self.server.verify_ssl).json()
|
|
||||||
tv_shows = [TVShow(**show) for show in get]
|
tv_shows = [TVShow(**show) for show in get]
|
||||||
|
|
||||||
for show in tv_shows:
|
for show in tv_shows:
|
||||||
|
@ -113,9 +110,8 @@ class SonarrAPI(object):
|
||||||
endpoint = '/api/queue'
|
endpoint = '/api/queue'
|
||||||
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
||||||
queue = []
|
queue = []
|
||||||
headers = {'X-Api-Key': self.server.api_key}
|
|
||||||
|
|
||||||
get = self.session.get(self.server.url + endpoint, headers=headers, verify=self.server.verify_ssl).json()
|
get = self.session.get(self.server.url + endpoint, verify=self.server.verify_ssl).json()
|
||||||
download_queue = [Queue(**show) for show in get]
|
download_queue = [Queue(**show) for show in get]
|
||||||
|
|
||||||
for show in download_queue:
|
for show in download_queue:
|
||||||
|
|
|
@ -13,6 +13,7 @@ class TautulliAPI(object):
|
||||||
self.dbmanager = dbmanager
|
self.dbmanager = dbmanager
|
||||||
self.server = server
|
self.server = server
|
||||||
self.session = Session()
|
self.session = Session()
|
||||||
|
self.session.params['apikey'] = self.server.api_key
|
||||||
self.endpoint = '/api/v2'
|
self.endpoint = '/api/v2'
|
||||||
|
|
||||||
@logging
|
@logging
|
||||||
|
@ -20,7 +21,6 @@ class TautulliAPI(object):
|
||||||
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
||||||
params = {'cmd': 'get_activity'}
|
params = {'cmd': 'get_activity'}
|
||||||
influx_payload = []
|
influx_payload = []
|
||||||
params['apikey'] = self.server.api_key
|
|
||||||
g = self.session.get(self.server.url + self.endpoint, params=params, verify=self.server.verify_ssl)
|
g = self.session.get(self.server.url + self.endpoint, params=params, verify=self.server.verify_ssl)
|
||||||
get = g.json()['response']['data']
|
get = g.json()['response']['data']
|
||||||
|
|
||||||
|
@ -51,7 +51,6 @@ class TautulliAPI(object):
|
||||||
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
self.now = datetime.now(timezone.utc).astimezone().isoformat()
|
||||||
params = {'cmd': 'get_activity'}
|
params = {'cmd': 'get_activity'}
|
||||||
influx_payload = []
|
influx_payload = []
|
||||||
params['apikey'] = self.server.api_key
|
|
||||||
g = self.session.get(self.server.url + self.endpoint, params=params, verify=self.server.verify_ssl)
|
g = self.session.get(self.server.url + self.endpoint, params=params, verify=self.server.verify_ssl)
|
||||||
get = g.json()['response']['data']['sessions']
|
get = g.json()['response']['data']['sessions']
|
||||||
sessions = [TautulliStream(**session) for session in get]
|
sessions = [TautulliStream(**session) for session in get]
|
||||||
|
|
Loading…
Reference in a new issue