jellyfin-kodi/jellyfin_kodi/jellyfin/http.py

241 lines
7.9 KiB
Python
Raw Normal View History

2018-09-06 08:36:32 +00:00
# -*- coding: utf-8 -*-
from __future__ import division, absolute_import, print_function, unicode_literals
2018-09-06 08:36:32 +00:00
#################################################################################################
import json
import logging
import time
2019-01-30 12:42:06 +00:00
import requests
2020-01-04 04:17:51 +00:00
from six import string_types
from .exceptions import HTTPException
from helper import debug
2018-09-06 08:36:32 +00:00
#################################################################################################
2019-10-03 02:14:54 +00:00
LOG = logging.getLogger('Jellyfin.' + __name__)
2018-09-06 08:36:32 +00:00
#################################################################################################
2019-07-09 20:05:28 +00:00
2018-09-06 08:36:32 +00:00
class HTTP(object):
session = None
keep_alive = False
def __init__(self, client):
self.client = client
self.config = client.config
2018-09-06 08:36:32 +00:00
def start_session(self):
2019-07-09 20:05:28 +00:00
2018-09-06 08:36:32 +00:00
self.session = requests.Session()
max_retries = self.config.data['http.max_retries']
2018-09-06 08:36:32 +00:00
self.session.mount("http://", requests.adapters.HTTPAdapter(max_retries=max_retries))
self.session.mount("https://", requests.adapters.HTTPAdapter(max_retries=max_retries))
def stop_session(self):
2019-07-09 20:05:28 +00:00
2018-09-06 08:36:32 +00:00
if self.session is None:
return
try:
LOG.info("--<[ session/%s ]", id(self.session))
2018-09-06 08:36:32 +00:00
self.session.close()
except Exception as error:
LOG.warning("The requests session could not be terminated: %s", error)
2018-09-06 08:36:32 +00:00
def _replace_user_info(self, string):
if '{server}' in string:
if self.config.data.get('auth.server', None):
string = string.replace("{server}", self.config.data['auth.server'])
else:
LOG.debug("Server address not set")
if '{UserId}'in string:
if self.config.data.get('auth.user_id', None):
string = string.replace("{UserId}", self.config.data['auth.user_id'])
else:
LOG.debug("UserId is not set.")
2018-09-06 08:36:32 +00:00
return string
def request(self, data, session=None):
''' Give a chance to retry the connection. Jellyfin sometimes can be slow to answer back
2018-09-06 08:36:32 +00:00
data dictionary can contain:
type: GET, POST, etc.
url: (optional)
handler: not considered when url is provided (optional)
params: request parameters (optional)
json: request body (optional)
headers: (optional),
verify: ssl certificate, True (verify using device built-in library) or False
'''
if not data:
raise AttributeError("Request cannot be empty")
data = self._request(data)
LOG.debug("--->[ http ] %s", debug.JsonDebugPrinter(data))
2018-09-06 08:36:32 +00:00
retry = data.pop('retry', 5)
while True:
try:
r = self._requests(session or self.session or requests, data.pop('type', "GET"), **data)
2019-07-09 20:05:28 +00:00
r.content # release the connection
2018-09-06 08:36:32 +00:00
if not self.keep_alive and self.session is not None:
self.stop_session()
r.raise_for_status()
except requests.exceptions.ConnectionError as error:
if retry:
retry -= 1
time.sleep(1)
continue
LOG.error(error)
self.client.callback("ServerUnreachable", {'ServerId': self.config.data['auth.server-id']})
2018-09-06 08:36:32 +00:00
raise HTTPException("ServerUnreachable", error)
except requests.exceptions.ReadTimeout as error:
if retry:
retry -= 1
time.sleep(1)
continue
LOG.error(error)
raise HTTPException("ReadTimeout", error)
except requests.exceptions.HTTPError as error:
LOG.error(error)
if r.status_code == 401:
if 'X-Application-Error-Code' in r.headers:
self.client.callback("AccessRestricted", {'ServerId': self.config.data['auth.server-id']})
2018-09-06 08:36:32 +00:00
raise HTTPException("AccessRestricted", error)
else:
self.client.callback("Unauthorized", {'ServerId': self.config.data['auth.server-id']})
2019-10-03 02:14:54 +00:00
self.client.auth.revoke_token()
2018-09-06 08:36:32 +00:00
raise HTTPException("Unauthorized", error)
2019-07-09 20:05:28 +00:00
elif r.status_code == 500: # log and ignore.
2018-09-06 08:36:32 +00:00
LOG.error("--[ 500 response ] %s", error)
return
elif r.status_code == 502:
if retry:
retry -= 1
time.sleep(1)
continue
raise HTTPException(r.status_code, error)
except requests.exceptions.MissingSchema as error:
LOG.error("Request missing Schema. " + str(error))
raise HTTPException("MissingSchema", {'Id': self.config.data.get('auth.server', "None")})
2018-09-06 08:36:32 +00:00
except Exception as error:
raise
else:
try:
self.config.data['server-time'] = r.headers['Date']
2018-09-06 08:36:32 +00:00
elapsed = int(r.elapsed.total_seconds() * 1000)
response = r.json()
LOG.debug("---<[ http ][%s ms]", elapsed)
LOG.debug(debug.JsonDebugPrinter(response))
2018-09-06 08:36:32 +00:00
return response
except ValueError:
return
def _request(self, data):
if 'url' not in data:
data['url'] = "%s/%s" % (self.config.data.get("auth.server", ""), data.pop('handler', ""))
2018-09-06 08:36:32 +00:00
self._get_header(data)
data['timeout'] = data.get('timeout') or self.config.data['http.timeout']
data['verify'] = data.get('verify') or self.config.data.get('auth.ssl', False)
2018-09-06 08:36:32 +00:00
data['url'] = self._replace_user_info(data['url'])
self._process_params(data.get('params') or {})
self._process_params(data.get('json') or {})
return data
def _process_params(self, params):
for key in params:
value = params[key]
if isinstance(value, dict):
self._process_params(value)
2020-01-04 04:17:51 +00:00
if isinstance(value, string_types):
2018-09-06 08:36:32 +00:00
params[key] = self._replace_user_info(value)
def _get_header(self, data):
data['headers'] = data.setdefault('headers', {})
if not data['headers']:
data['headers'].update({
'Content-type': "application/json",
'Accept-Charset': "UTF-8,*",
'Accept-encoding': "gzip",
'User-Agent': self.config.data['http.user_agent'] or "%s/%s" % (self.config.data.get('app.name', 'Jellyfin for Kodi'), self.config.data.get('app.version', "0.0.0"))
2018-09-06 08:36:32 +00:00
})
if 'x-emby-authorization' not in data['headers']:
2018-09-06 08:36:32 +00:00
self._authorization(data)
return data
def _authorization(self, data):
2019-07-09 20:05:28 +00:00
auth = "MediaBrowser "
auth += "Client=%s, " % self.config.data.get('app.name', "Jellyfin for Kodi")
auth += "Device=%s, " % self.config.data.get('app.device_name', 'Unknown Device')
auth += "DeviceId=%s, " % self.config.data.get('app.device_id', 'Unknown Device id')
auth += "Version=%s" % self.config.data.get('app.version', '0.0.0')
2018-09-06 08:36:32 +00:00
data['headers'].update({'x-emby-authorization': auth})
2018-09-06 08:36:32 +00:00
if self.config.data.get('auth.token') and self.config.data.get('auth.user_id'):
auth += ', UserId=%s' % self.config.data.get('auth.user_id')
data['headers'].update({'x-emby-authorization': auth, 'X-MediaBrowser-Token': self.config.data.get('auth.token')})
2018-09-06 08:36:32 +00:00
return data
def _requests(self, session, action, **kwargs):
if action == "GET":
return session.get(**kwargs)
elif action == "POST":
return session.post(**kwargs)
elif action == "HEAD":
return session.head(**kwargs)
elif action == "DELETE":
return session.delete(**kwargs)