Refactored connectionmanager to use newer explicit request interface.

This commit is contained in:
Chuddah 2020-02-22 19:40:26 +00:00
commit 2dc040dfea

View file

@ -99,16 +99,13 @@ class ConnectionManager(object):
raise AttributeError("server cannot be empty") raise AttributeError("server cannot be empty")
try: try:
request = { url = self.get_jellyfin_url(server, "Users/AuthenticateByName")
'type': "POST", json = {
'url': self.get_jellyfin_url(server, "Users/AuthenticateByName"),
'json': {
'Username': username, 'Username': username,
'Pw': password or "" 'Pw': password or ""
} }
} result = self._REQUEST_URL(url, "POST", json=json, additional_headers=False)
result = self._request_url(request, False)
except Exception as error: # Failed to login except Exception as error: # Failed to login
LOG.exception(error) LOG.exception(error)
return False return False
@ -155,6 +152,7 @@ class ConnectionManager(object):
return self._after_connect_validated(server, credentials, result, True, options) return self._after_connect_validated(server, credentials, result, True, options)
except Exception as e: except Exception as e:
raise
LOG.info("Failing server connection. ERROR msg: {}".format(e)) LOG.info("Failing server connection. ERROR msg: {}".format(e))
return { 'State': CONNECTION_STATE['Unavailable'] } return { 'State': CONNECTION_STATE['Unavailable'] }
@ -186,14 +184,31 @@ class ConnectionManager(object):
def get_jellyfin_url(self, base, handler): def get_jellyfin_url(self, base, handler):
return "%s/%s" % (base, handler) return "%s/%s" % (base, handler)
def _request_url(self, request, headers=True): def _REQUEST_URL(self, url, type, data_type=None, timeout=None, verify=None, retry=None,
headers={}, json=None, additional_headers=True):
data = {
'type': type,
'url': url,
'headers': headers,
}
if timeout is not None:
data['timeout'] = timeout
if verify is not None:
data['verify'] = verify
if retry is not None:
data['retry'] = retry
if json is not None:
data['json'] = json
return self._request_url(data, data_type, timeout, additional_headers=additional_headers)
request['timeout'] = request.get('timeout') or self.timeout def _request_url(self, data, data_type=None, timeout=None, additional_headers=True):
if headers: data['timeout'] = timeout or self.timeout
self._get_headers(request) if additional_headers:
headers = self._get_headers(data_type)
data.setdefault('headers', {}).update(headers)
try: try:
return self.http.request(request) return self.http.request(data)
except Exception as error: except Exception as error:
LOG.exception(error) LOG.exception(error)
raise raise
@ -201,19 +216,16 @@ class ConnectionManager(object):
def _add_app_info(self): def _add_app_info(self):
return "%s/%s" % (self.config.data['app.name'], self.config.data['app.version']) return "%s/%s" % (self.config.data['app.name'], self.config.data['app.version'])
def _get_headers(self, request): def _get_headers(self, data_type):
headers = {}
headers = request.setdefault('headers', {}) if data_type == "json":
if request.get('dataType') == "json":
headers['Accept'] = "application/json" headers['Accept'] = "application/json"
request.pop('dataType')
headers['X-Application'] = self._add_app_info() headers['X-Application'] = self._add_app_info()
headers['Content-type'] = request.get( headers['Content-type'] = 'application/x-www-form-urlencoded; charset=UTF-8'
'contentType',
'application/x-www-form-urlencoded; charset=UTF-8' return headers
)
def _connect_to_servers(self, servers, options): def _connect_to_servers(self, servers, options):
@ -247,14 +259,8 @@ class ConnectionManager(object):
url = self.get_jellyfin_url(url, "system/info/public") url = self.get_jellyfin_url(url, "system/info/public")
LOG.info("tryConnect url: %s", url) LOG.info("tryConnect url: %s", url)
return self._request_url({ return self._REQUEST_URL(url, "GET", 'json', timeout, \
'type': "GET", options.get('ssl'), retry=False)
'url': url,
'dataType': "json",
'timeout': timeout,
'verify': options.get('ssl'),
'retry': False
})
def _server_discovery(self): def _server_discovery(self):
@ -423,15 +429,14 @@ class ConnectionManager(object):
def _validate_authentication(self, server, options={}): def _validate_authentication(self, server, options={}):
try: try:
system_info = self._request_url({ url = self.get_jellyfin_url(server['address'], "System/Info")
'type': "GET", headers = {
'url': self.get_jellyfin_url(server['address'], "System/Info"),
'verify': options.get('ssl'),
'dataType': "json",
'headers': {
'X-MediaBrowser-Token': server['AccessToken'] 'X-MediaBrowser-Token': server['AccessToken']
} }
})
system_info = self._REQUEST_URL(url, "GET", 'json', \
verify=options.get('ssl'), headers=headers)
self._update_server_info(server, system_info) self._update_server_info(server, system_info)
except Exception as error: except Exception as error:
LOG.exception(error) LOG.exception(error)