mirror of
https://github.com/robweber/xbmcbackup.git
synced 2024-12-23 06:27:55 +01:00
520 lines
18 KiB
Python
520 lines
18 KiB
Python
__all__ = [
|
|
'BadRequestException',
|
|
'BadStateException',
|
|
'CsrfException',
|
|
'DropboxOAuth2Flow',
|
|
'DropboxOAuth2FlowNoRedirect',
|
|
'NotApprovedException',
|
|
'OAuth2FlowNoRedirectResult',
|
|
'OAuth2FlowResult',
|
|
'ProviderException',
|
|
]
|
|
|
|
import base64
|
|
import os
|
|
import six
|
|
import urllib
|
|
|
|
from .session import (
|
|
API_HOST,
|
|
WEB_HOST,
|
|
pinned_session,
|
|
)
|
|
|
|
if six.PY3:
|
|
url_path_quote = urllib.parse.quote # pylint: disable=no-member,useless-suppression
|
|
url_encode = urllib.parse.urlencode # pylint: disable=no-member,useless-suppression
|
|
else:
|
|
url_path_quote = urllib.quote # pylint: disable=no-member,useless-suppression
|
|
url_encode = urllib.urlencode # pylint: disable=no-member,useless-suppression
|
|
|
|
|
|
class OAuth2FlowNoRedirectResult(object):
|
|
"""
|
|
Authorization information for an OAuth2Flow performed with no redirect.
|
|
"""
|
|
|
|
def __init__(self, access_token, account_id, user_id):
|
|
"""
|
|
Args:
|
|
access_token (str): Token to be used to authenticate later
|
|
requests.
|
|
account_id (str): The Dropbox user's account ID.
|
|
user_id (str): Deprecated (use account_id instead).
|
|
"""
|
|
self.access_token = access_token
|
|
self.account_id = account_id
|
|
self.user_id = user_id
|
|
|
|
def __repr__(self):
|
|
return 'OAuth2FlowNoRedirectResult(%r, %r, %r)' % (
|
|
self.access_token,
|
|
self.account_id,
|
|
self.user_id,
|
|
)
|
|
|
|
|
|
class OAuth2FlowResult(OAuth2FlowNoRedirectResult):
|
|
"""
|
|
Authorization information for an OAuth2Flow with redirect.
|
|
"""
|
|
|
|
def __init__(self, access_token, account_id, user_id, url_state):
|
|
"""
|
|
Same as OAuth2FlowNoRedirectResult but with url_state.
|
|
|
|
Args:
|
|
url_state (str): The url state that was set by
|
|
:meth:`DropboxOAuth2Flow.start`.
|
|
"""
|
|
super(OAuth2FlowResult, self).__init__(
|
|
access_token, account_id, user_id)
|
|
self.url_state = url_state
|
|
|
|
@classmethod
|
|
def from_no_redirect_result(cls, result, url_state):
|
|
assert isinstance(result, OAuth2FlowNoRedirectResult)
|
|
return cls(
|
|
result.access_token, result.account_id, result.user_id, url_state)
|
|
|
|
def __repr__(self):
|
|
return 'OAuth2FlowResult(%r, %r, %r, %r)' % (
|
|
self.access_token,
|
|
self.account_id,
|
|
self.user_id,
|
|
self.url_state,
|
|
)
|
|
|
|
|
|
class DropboxOAuth2FlowBase(object):
|
|
|
|
def __init__(self, consumer_key, consumer_secret, locale=None):
|
|
self.consumer_key = consumer_key
|
|
self.consumer_secret = consumer_secret
|
|
self.locale = locale
|
|
self.requests_session = pinned_session()
|
|
|
|
def _get_authorize_url(self, redirect_uri, state):
|
|
params = dict(response_type='code',
|
|
client_id=self.consumer_key)
|
|
if redirect_uri is not None:
|
|
params['redirect_uri'] = redirect_uri
|
|
if state is not None:
|
|
params['state'] = state
|
|
|
|
return self.build_url('/oauth2/authorize', params, WEB_HOST)
|
|
|
|
def _finish(self, code, redirect_uri):
|
|
url = self.build_url('/oauth2/token')
|
|
params = {'grant_type': 'authorization_code',
|
|
'code': code,
|
|
'client_id': self.consumer_key,
|
|
'client_secret': self.consumer_secret,
|
|
}
|
|
if self.locale is not None:
|
|
params['locale'] = self.locale
|
|
if redirect_uri is not None:
|
|
params['redirect_uri'] = redirect_uri
|
|
|
|
resp = self.requests_session.post(url, data=params)
|
|
resp.raise_for_status()
|
|
|
|
d = resp.json()
|
|
|
|
if 'team_id' in d:
|
|
account_id = d['team_id']
|
|
else:
|
|
account_id = d['account_id']
|
|
|
|
access_token = d['access_token']
|
|
uid = d['uid']
|
|
|
|
return OAuth2FlowNoRedirectResult(
|
|
access_token,
|
|
account_id,
|
|
uid)
|
|
|
|
def build_path(self, target, params=None):
|
|
"""Build the path component for an API URL.
|
|
|
|
This method urlencodes the parameters, adds them
|
|
to the end of the target url, and puts a marker for the API
|
|
version in front.
|
|
|
|
:param str target: A target url (e.g. '/files') to build upon.
|
|
:param dict params: Optional dictionary of parameters (name to value).
|
|
:return: The path and parameters components of an API URL.
|
|
:rtype: str
|
|
"""
|
|
if six.PY2 and isinstance(target, six.text_type):
|
|
target = target.encode('utf8')
|
|
|
|
target_path = url_path_quote(target)
|
|
|
|
params = params or {}
|
|
params = params.copy()
|
|
|
|
if self.locale:
|
|
params['locale'] = self.locale
|
|
|
|
if params:
|
|
query_string = _params_to_urlencoded(params)
|
|
return "%s?%s" % (target_path, query_string)
|
|
else:
|
|
return target_path
|
|
|
|
def build_url(self, target, params=None, host=API_HOST):
|
|
"""Build an API URL.
|
|
|
|
This method adds scheme and hostname to the path
|
|
returned from build_path.
|
|
|
|
:param str target: A target url (e.g. '/files') to build upon.
|
|
:param dict params: Optional dictionary of parameters (name to value).
|
|
:return: The full API URL.
|
|
:rtype: str
|
|
"""
|
|
return "https://%s%s" % (host, self.build_path(target, params))
|
|
|
|
|
|
class DropboxOAuth2FlowNoRedirect(DropboxOAuth2FlowBase):
|
|
"""
|
|
OAuth 2 authorization helper for apps that can't provide a redirect URI
|
|
(such as the command-line example apps).
|
|
|
|
Example::
|
|
|
|
from dropbox import DropboxOAuth2FlowNoRedirect
|
|
|
|
auth_flow = DropboxOAuth2FlowNoRedirect(APP_KEY, APP_SECRET)
|
|
|
|
authorize_url = auth_flow.start()
|
|
print "1. Go to: " + authorize_url
|
|
print "2. Click \\"Allow\\" (you might have to log in first)."
|
|
print "3. Copy the authorization code."
|
|
auth_code = raw_input("Enter the authorization code here: ").strip()
|
|
|
|
try:
|
|
oauth_result = auth_flow.finish(auth_code)
|
|
except Exception, e:
|
|
print('Error: %s' % (e,))
|
|
return
|
|
|
|
dbx = Dropbox(oauth_result.access_token)
|
|
"""
|
|
|
|
def __init__(self, consumer_key, consumer_secret, locale=None): # noqa: E501; pylint: disable=useless-super-delegation
|
|
"""
|
|
Construct an instance.
|
|
|
|
Parameters
|
|
:param str consumer_key: Your API app's "app key".
|
|
:param str consumer_secret: Your API app's "app secret".
|
|
:param str locale: The locale of the user of your application. For
|
|
example "en" or "en_US". Some API calls return localized data and
|
|
error messages; this setting tells the server which locale to use.
|
|
By default, the server uses "en_US".
|
|
"""
|
|
# pylint: disable=useless-super-delegation
|
|
super(DropboxOAuth2FlowNoRedirect, self).__init__(
|
|
consumer_key,
|
|
consumer_secret,
|
|
locale,
|
|
)
|
|
|
|
def start(self):
|
|
"""
|
|
Starts the OAuth 2 authorization process.
|
|
|
|
:return: The URL for a page on Dropbox's website. This page will let
|
|
the user "approve" your app, which gives your app permission to
|
|
access the user's Dropbox account. Tell the user to visit this URL
|
|
and approve your app.
|
|
"""
|
|
return self._get_authorize_url(None, None)
|
|
|
|
def finish(self, code):
|
|
"""
|
|
If the user approves your app, they will be presented with an
|
|
"authorization code". Have the user copy/paste that authorization code
|
|
into your app and then call this method to get an access token.
|
|
|
|
:param str code: The authorization code shown to the user when they
|
|
approved your app.
|
|
:rtype: OAuth2FlowNoRedirectResult
|
|
:raises: The same exceptions as :meth:`DropboxOAuth2Flow.finish()`.
|
|
"""
|
|
return self._finish(code, None)
|
|
|
|
|
|
class DropboxOAuth2Flow(DropboxOAuth2FlowBase):
|
|
"""
|
|
OAuth 2 authorization helper. Use this for web apps.
|
|
|
|
OAuth 2 has a two-step authorization process. The first step is having the
|
|
user authorize your app. The second involves getting an OAuth 2 access
|
|
token from Dropbox.
|
|
|
|
Example::
|
|
|
|
from dropbox import DropboxOAuth2Flow
|
|
|
|
def get_dropbox_auth_flow(web_app_session):
|
|
redirect_uri = "https://my-web-server.org/dropbox-auth-finish"
|
|
return DropboxOAuth2Flow(
|
|
APP_KEY, APP_SECRET, redirect_uri, web_app_session,
|
|
"dropbox-auth-csrf-token")
|
|
|
|
# URL handler for /dropbox-auth-start
|
|
def dropbox_auth_start(web_app_session, request):
|
|
authorize_url = get_dropbox_auth_flow(web_app_session).start()
|
|
redirect_to(authorize_url)
|
|
|
|
# URL handler for /dropbox-auth-finish
|
|
def dropbox_auth_finish(web_app_session, request):
|
|
try:
|
|
oauth_result = \\
|
|
get_dropbox_auth_flow(web_app_session).finish(
|
|
request.query_params)
|
|
except BadRequestException, e:
|
|
http_status(400)
|
|
except BadStateException, e:
|
|
# Start the auth flow again.
|
|
redirect_to("/dropbox-auth-start")
|
|
except CsrfException, e:
|
|
http_status(403)
|
|
except NotApprovedException, e:
|
|
flash('Not approved? Why not?')
|
|
return redirect_to("/home")
|
|
except ProviderException, e:
|
|
logger.log("Auth error: %s" % (e,))
|
|
http_status(403)
|
|
|
|
"""
|
|
|
|
def __init__(self, consumer_key, consumer_secret, redirect_uri, session,
|
|
csrf_token_session_key, locale=None):
|
|
"""
|
|
Construct an instance.
|
|
|
|
:param str consumer_key: Your API app's "app key".
|
|
:param str consumer_secret: Your API app's "app secret".
|
|
:param str redirect_uri: The URI that the Dropbox server will redirect
|
|
the user to after the user finishes authorizing your app. This URI
|
|
must be HTTPS-based and pre-registered with the Dropbox servers,
|
|
though localhost URIs are allowed without pre-registration and can
|
|
be either HTTP or HTTPS.
|
|
:param dict session: A dict-like object that represents the current
|
|
user's web session (will be used to save the CSRF token).
|
|
:param str csrf_token_session_key: The key to use when storing the CSRF
|
|
token in the session (for example: "dropbox-auth-csrf-token").
|
|
:param str locale: The locale of the user of your application. For
|
|
example "en" or "en_US". Some API calls return localized data and
|
|
error messages; this setting tells the server which locale to use.
|
|
By default, the server uses "en_US".
|
|
"""
|
|
super(DropboxOAuth2Flow, self).__init__(consumer_key, consumer_secret, locale)
|
|
self.redirect_uri = redirect_uri
|
|
self.session = session
|
|
self.csrf_token_session_key = csrf_token_session_key
|
|
|
|
def start(self, url_state=None):
|
|
"""
|
|
Starts the OAuth 2 authorization process.
|
|
|
|
This function builds an "authorization URL". You should redirect your
|
|
user's browser to this URL, which will give them an opportunity to
|
|
grant your app access to their Dropbox account. When the user
|
|
completes this process, they will be automatically redirected to the
|
|
``redirect_uri`` you passed in to the constructor.
|
|
|
|
This function will also save a CSRF token to
|
|
``session[csrf_token_session_key]`` (as provided to the constructor).
|
|
This CSRF token will be checked on :meth:`finish()` to prevent request
|
|
forgery.
|
|
|
|
:param str url_state: Any data that you would like to keep in the URL
|
|
through the authorization process. This exact value will be
|
|
returned to you by :meth:`finish()`.
|
|
:return: The URL for a page on Dropbox's website. This page will let
|
|
the user "approve" your app, which gives your app permission to
|
|
access the user's Dropbox account. Tell the user to visit this URL
|
|
and approve your app.
|
|
"""
|
|
csrf_token = base64.urlsafe_b64encode(os.urandom(16)).decode('ascii')
|
|
state = csrf_token
|
|
if url_state is not None:
|
|
state += "|" + url_state
|
|
self.session[self.csrf_token_session_key] = csrf_token
|
|
|
|
return self._get_authorize_url(self.redirect_uri, state)
|
|
|
|
def finish(self, query_params):
|
|
"""
|
|
Call this after the user has visited the authorize URL (see
|
|
:meth:`start()`), approved your app and was redirected to your redirect
|
|
URI.
|
|
|
|
:param dict query_params: The query parameters on the GET request to
|
|
your redirect URI.
|
|
:rtype: OAuth2FlowResult
|
|
:raises: :class:`BadRequestException` If the redirect URL was missing
|
|
parameters or if the given parameters were not valid.
|
|
:raises: :class:`BadStateException` If there's no CSRF token in the
|
|
session.
|
|
:raises: :class:`CsrfException` If the ``state`` query parameter
|
|
doesn't contain the CSRF token from the user's session.
|
|
:raises: :class:`NotApprovedException` If the user chose not to
|
|
approve your app.
|
|
:raises: :class:`ProviderException` If Dropbox redirected to your
|
|
redirect URI with some unexpected error identifier and error message.
|
|
"""
|
|
# Check well-formedness of request.
|
|
|
|
state = query_params.get('state')
|
|
if state is None:
|
|
raise BadRequestException("Missing query parameter 'state'.")
|
|
|
|
error = query_params.get('error')
|
|
error_description = query_params.get('error_description')
|
|
code = query_params.get('code')
|
|
|
|
if error is not None and code is not None:
|
|
raise BadRequestException(
|
|
"Query parameters 'code' and 'error' are both set; "
|
|
"only one must be set.")
|
|
if error is None and code is None:
|
|
raise BadRequestException(
|
|
"Neither query parameter 'code' or 'error' is set.")
|
|
|
|
# Check CSRF token
|
|
|
|
if self.csrf_token_session_key not in self.session:
|
|
raise BadStateException('Missing CSRF token in session.')
|
|
csrf_token_from_session = self.session[self.csrf_token_session_key]
|
|
if len(csrf_token_from_session) <= 20:
|
|
raise AssertionError('CSRF token unexpectedly short: %r' %
|
|
csrf_token_from_session)
|
|
|
|
split_pos = state.find('|')
|
|
if split_pos < 0:
|
|
given_csrf_token = state
|
|
url_state = None
|
|
else:
|
|
given_csrf_token = state[0:split_pos]
|
|
url_state = state[split_pos + 1:]
|
|
|
|
if not _safe_equals(csrf_token_from_session, given_csrf_token):
|
|
raise CsrfException('expected %r, got %r' %
|
|
(csrf_token_from_session, given_csrf_token))
|
|
|
|
del self.session[self.csrf_token_session_key]
|
|
|
|
# Check for error identifier
|
|
|
|
if error is not None:
|
|
if error == 'access_denied':
|
|
# The user clicked "Deny"
|
|
if error_description is None:
|
|
raise NotApprovedException(
|
|
'No additional description from Dropbox')
|
|
else:
|
|
raise NotApprovedException(
|
|
'Additional description from Dropbox: %s' %
|
|
error_description)
|
|
else:
|
|
# All other errors
|
|
full_message = error
|
|
if error_description is not None:
|
|
full_message += ": " + error_description
|
|
raise ProviderException(full_message)
|
|
|
|
# If everything went ok, make the network call to get an access token.
|
|
|
|
no_redirect_result = self._finish(code, self.redirect_uri)
|
|
return OAuth2FlowResult.from_no_redirect_result(
|
|
no_redirect_result, url_state)
|
|
|
|
|
|
class BadRequestException(Exception):
|
|
"""
|
|
Thrown if the redirect URL was missing parameters or if the
|
|
given parameters were not valid.
|
|
|
|
The recommended action is to show an HTTP 400 error page.
|
|
"""
|
|
pass
|
|
|
|
|
|
class BadStateException(Exception):
|
|
"""
|
|
Thrown if all the parameters are correct, but there's no CSRF token in the
|
|
session. This probably means that the session expired.
|
|
|
|
The recommended action is to redirect the user's browser to try the
|
|
approval process again.
|
|
"""
|
|
pass
|
|
|
|
|
|
class CsrfException(Exception):
|
|
"""
|
|
Thrown if the given 'state' parameter doesn't contain the CSRF token from
|
|
the user's session. This is blocked to prevent CSRF attacks.
|
|
|
|
The recommended action is to respond with an HTTP 403 error page.
|
|
"""
|
|
pass
|
|
|
|
|
|
class NotApprovedException(Exception):
|
|
"""
|
|
The user chose not to approve your app.
|
|
"""
|
|
pass
|
|
|
|
|
|
class ProviderException(Exception):
|
|
"""
|
|
Dropbox redirected to your redirect URI with some unexpected error
|
|
identifier and error message.
|
|
|
|
The recommended action is to log the error, tell the user something went
|
|
wrong, and let them try again.
|
|
"""
|
|
pass
|
|
|
|
|
|
def _safe_equals(a, b):
|
|
if len(a) != len(b):
|
|
return False
|
|
res = 0
|
|
for ca, cb in zip(a, b):
|
|
res |= ord(ca) ^ ord(cb)
|
|
return res == 0
|
|
|
|
|
|
def _params_to_urlencoded(params):
|
|
"""
|
|
Returns a application/x-www-form-urlencoded ``str`` representing the
|
|
key/value pairs in ``params``.
|
|
|
|
Keys are values are ``str()``'d before calling ``urllib.urlencode``, with
|
|
the exception of unicode objects which are utf8-encoded.
|
|
"""
|
|
def encode(o):
|
|
if isinstance(o, six.binary_type):
|
|
return o
|
|
else:
|
|
if isinstance(o, six.text_type):
|
|
return o.encode('utf-8')
|
|
else:
|
|
return str(o).encode('utf-8')
|
|
|
|
#fix for python 2.6
|
|
utf8_params = {}
|
|
for k,v in six.iteritems(params):
|
|
utf8_params[encode(k)] = encode(v)
|
|
|
|
return url_encode(utf8_params)
|