mirror of
https://github.com/jellyfin/jellyfin-kodi.git
synced 2025-11-10 18:36:33 +00:00
Refactored connectionmanager to use newer explicit request interface.
This commit is contained in:
parent
c6eb82464c
commit
2dc040dfea
1 changed files with 44 additions and 39 deletions
|
|
@ -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)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue