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…
	
	Add table
		Add a link
		
	
		Reference in a new issue