mirror of
https://github.com/jellyfin/jellyfin-kodi.git
synced 2025-11-11 02:46:34 +00:00
Removed all http calls from connection_manager.py and moved to api.py
This commit is contained in:
parent
e81745f3d1
commit
fd34bab6fb
2 changed files with 21 additions and 59 deletions
|
|
@ -394,7 +394,7 @@ class API(object):
|
||||||
return {}
|
return {}
|
||||||
|
|
||||||
def validate_authentication_token(self, server):
|
def validate_authentication_token(self, server):
|
||||||
url = server['address']+"/System/Info"
|
url = "%s/%s" % (server['address'], "system/info")
|
||||||
authTokenHeader = {
|
authTokenHeader = {
|
||||||
'X-MediaBrowser-Token': server['AccessToken']
|
'X-MediaBrowser-Token': server['AccessToken']
|
||||||
}
|
}
|
||||||
|
|
@ -402,6 +402,11 @@ class API(object):
|
||||||
headers.update(authTokenHeader)
|
headers.update(authTokenHeader)
|
||||||
|
|
||||||
response = requests.get(url, timeout=self.default_timeout, headers=headers, verify=settings('sslverify'))
|
response = requests.get(url, timeout=self.default_timeout, headers=headers, verify=settings('sslverify'))
|
||||||
|
|
||||||
return response.json() if response.status_code == 200 else {}
|
return response.json() if response.status_code == 200 else {}
|
||||||
|
|
||||||
|
def get_public_info(self, server_address):
|
||||||
|
url = "%s/%s" % (server_address, "system/info/public")
|
||||||
|
headers = self.get_default_headers()
|
||||||
|
|
||||||
|
response = requests.get(url, timeout=self.default_timeout, headers=headers, verify=settings('sslverify'))
|
||||||
|
return response.json() if response.status_code == 200 else {}
|
||||||
|
|
@ -7,7 +7,6 @@ import logging
|
||||||
import socket
|
import socket
|
||||||
import time
|
import time
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from distutils.version import LooseVersion
|
|
||||||
|
|
||||||
import urllib3
|
import urllib3
|
||||||
|
|
||||||
|
|
@ -41,10 +40,9 @@ class ConnectionManager(object):
|
||||||
self.config = client.config
|
self.config = client.config
|
||||||
self.credentials = Credentials()
|
self.credentials = Credentials()
|
||||||
|
|
||||||
self.http = HTTP(client)
|
|
||||||
self.API = API(client)
|
self.API = API(client)
|
||||||
|
|
||||||
def clear_data(self): ## Never actually called
|
"""def clear_data(self): ## Never called
|
||||||
|
|
||||||
LOG.info("connection manager clearing data")
|
LOG.info("connection manager clearing data")
|
||||||
|
|
||||||
|
|
@ -53,7 +51,7 @@ class ConnectionManager(object):
|
||||||
credentials['Servers'] = list()
|
credentials['Servers'] = list()
|
||||||
self.credentials.set(credentials)
|
self.credentials.set(credentials)
|
||||||
|
|
||||||
self.config.auth(None, None)
|
self.config.auth(None, None)"""
|
||||||
|
|
||||||
def revoke_token(self): #Called once in http#L130
|
def revoke_token(self): #Called once in http#L130
|
||||||
|
|
||||||
|
|
@ -148,7 +146,7 @@ class ConnectionManager(object):
|
||||||
address = self._normalize_address(address)
|
address = self._normalize_address(address)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
public_info = self._try_connect(address, options=options)
|
public_info = self.API.get_public_info(address)
|
||||||
LOG.info("connectToAddress %s succeeded", address)
|
LOG.info("connectToAddress %s succeeded", address)
|
||||||
server = {
|
server = {
|
||||||
'address': address,
|
'address': address,
|
||||||
|
|
@ -171,7 +169,12 @@ class ConnectionManager(object):
|
||||||
LOG.info("begin connectToServer")
|
LOG.info("begin connectToServer")
|
||||||
|
|
||||||
try:
|
try:
|
||||||
result = self._try_connect(server['address'], self.timeout, options)
|
result = self.API.get_public_info(server.get('address'))
|
||||||
|
|
||||||
|
if not result:
|
||||||
|
LOG.error("Failed to connect to server: %s" % server.get('address'))
|
||||||
|
return { 'State': CONNECTION_STATE['Unavailable'] }
|
||||||
|
|
||||||
LOG.info("calling onSuccessfulConnection with server %s", server.get('Name'))
|
LOG.info("calling onSuccessfulConnection with server %s", server.get('Name'))
|
||||||
|
|
||||||
credentials = self.credentials.get()
|
credentials = self.credentials.get()
|
||||||
|
|
@ -218,52 +221,6 @@ class ConnectionManager(object):
|
||||||
def get_public_users(self): ## Required in connect.py#L213
|
def get_public_users(self): ## Required in connect.py#L213
|
||||||
return self.client.jellyfin.get_public_users()
|
return self.client.jellyfin.get_public_users()
|
||||||
|
|
||||||
def get_jellyfin_url(self, base, handler):
|
|
||||||
return "%s/%s" % (base, handler)
|
|
||||||
|
|
||||||
def _request_url(self, request, headers=True):
|
|
||||||
|
|
||||||
request['timeout'] = request.get('timeout') or self.timeout
|
|
||||||
if headers:
|
|
||||||
self._get_headers(request)
|
|
||||||
|
|
||||||
try:
|
|
||||||
return self.http.request(request)
|
|
||||||
except Exception as error:
|
|
||||||
LOG.exception(error)
|
|
||||||
raise
|
|
||||||
|
|
||||||
def _add_app_info(self):
|
|
||||||
return "%s/%s" % (self.config.data['app.name'], self.config.data['app.version'])
|
|
||||||
|
|
||||||
def _get_headers(self, request):
|
|
||||||
|
|
||||||
headers = request.setdefault('headers', {})
|
|
||||||
|
|
||||||
if request.get('dataType') == "json":
|
|
||||||
headers['Accept'] = "application/json"
|
|
||||||
request.pop('dataType')
|
|
||||||
|
|
||||||
headers['X-Application'] = self._add_app_info()
|
|
||||||
headers['Content-type'] = request.get(
|
|
||||||
'contentType',
|
|
||||||
'application/x-www-form-urlencoded; charset=UTF-8'
|
|
||||||
)
|
|
||||||
|
|
||||||
def _try_connect(self, url, timeout=None, options={}):
|
|
||||||
|
|
||||||
url = self.get_jellyfin_url(url, "system/info/public")
|
|
||||||
LOG.info("tryConnect url: %s", url)
|
|
||||||
|
|
||||||
return self._request_url({
|
|
||||||
'type': "GET",
|
|
||||||
'url': url,
|
|
||||||
'dataType': "json",
|
|
||||||
'timeout': timeout,
|
|
||||||
'verify': options.get('ssl'),
|
|
||||||
'retry': False
|
|
||||||
})
|
|
||||||
|
|
||||||
def _server_discovery(self):
|
def _server_discovery(self):
|
||||||
|
|
||||||
MULTI_GROUP = ("<broadcast>", 7359)
|
MULTI_GROUP = ("<broadcast>", 7359)
|
||||||
|
|
@ -301,7 +258,7 @@ class ConnectionManager(object):
|
||||||
LOG.exception("Error trying to find servers: %s", e)
|
LOG.exception("Error trying to find servers: %s", e)
|
||||||
return servers
|
return servers
|
||||||
|
|
||||||
def _get_last_used_server(self): ## Never used
|
"""def _get_last_used_server(self): ## Never called
|
||||||
|
|
||||||
servers = self.credentials.get()['Servers']
|
servers = self.credentials.get()['Servers']
|
||||||
|
|
||||||
|
|
@ -313,7 +270,7 @@ class ConnectionManager(object):
|
||||||
except TypeError:
|
except TypeError:
|
||||||
servers.sort(key=lambda x: datetime(*(time.strptime(x['DateLastAccessed'], "%Y-%m-%dT%H:%M:%SZ")[0:6])), reverse=True)
|
servers.sort(key=lambda x: datetime(*(time.strptime(x['DateLastAccessed'], "%Y-%m-%dT%H:%M:%SZ")[0:6])), reverse=True)
|
||||||
|
|
||||||
return servers[0]
|
return servers[0]"""
|
||||||
|
|
||||||
def _find_servers(self, found_servers):
|
def _find_servers(self, found_servers):
|
||||||
|
|
||||||
|
|
@ -334,7 +291,7 @@ class ConnectionManager(object):
|
||||||
return servers
|
return servers
|
||||||
|
|
||||||
# TODO: Make IPv6 compatable
|
# TODO: Make IPv6 compatable
|
||||||
def _convert_endpoint_address_to_manual_address(self, info):
|
def _convert_endpoint_address_to_manual_address(self, info): # Called once ^^ right there
|
||||||
|
|
||||||
if info.get('Address') and info.get('EndpointAddress'):
|
if info.get('Address') and info.get('EndpointAddress'):
|
||||||
address = info['EndpointAddress'].split(':')[0]
|
address = info['EndpointAddress'].split(':')[0]
|
||||||
|
|
@ -378,7 +335,7 @@ class ConnectionManager(object):
|
||||||
self.config.data['auth.token'] = server.pop('AccessToken', None)
|
self.config.data['auth.token'] = server.pop('AccessToken', None)
|
||||||
|
|
||||||
elif verify_authentication and server.get('AccessToken'):
|
elif verify_authentication and server.get('AccessToken'):
|
||||||
if self._validate_authentication(server, options):
|
if self._validate_authentication(server):
|
||||||
|
|
||||||
self.config.data['auth.user_id'] = server['UserId']
|
self.config.data['auth.user_id'] = server['UserId']
|
||||||
self.config.data['auth.token'] = server['AccessToken']
|
self.config.data['auth.token'] = server['AccessToken']
|
||||||
|
|
@ -408,7 +365,7 @@ class ConnectionManager(object):
|
||||||
# Connected
|
# Connected
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def _validate_authentication(self, server, options={}):
|
def _validate_authentication(self, server):
|
||||||
|
|
||||||
system_info = self.API.validate_authentication_token(server)
|
system_info = self.API.validate_authentication_token(server)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue