1 from __future__ import unicode_literals
2 from oauthlib.oauth2 import WebApplicationClient, InsecureTransportError
4 from .utils import is_secure_transport
8 """Adds proof of authorization (OAuth2 token) to the request."""
10 def __init__(self, client_id=None, client=None, token=None):
11 """Construct a new OAuth 2 authorization object.
13 :param client_id: Client id obtained during registration
14 :param client: :class:`oauthlib.oauth2.Client` to be used. Default is
15 WebApplicationClient which is useful for any
16 hosted application but not mobile or desktop.
17 :param token: Token dictionary, must include access_token
20 self._client = client or WebApplicationClient(client_id, token=token)
22 for k, v in token.items():
23 setattr(self._client, k, v)
25 def __call__(self, r):
26 """Append an OAuth 2 token to the request.
28 Note that currently HTTPS is required for all requests. There may be
29 a token type that allows for plain HTTP in the future and then this
30 should be updated to allow plain HTTP on a white list basis.
32 if not is_secure_transport(r.url):
33 raise InsecureTransportError()
34 r.url, r.headers, r.body = self._client.add_token(r.url,
35 http_method=r.method, body=r.body, headers=r.headers)