from __future__ import unicode_literals
from flask_dance.consumer import OAuth2ConsumerBlueprint
from functools import partial
from flask.globals import LocalProxy, _lookup_app_object
try:
from flask import _app_ctx_stack as stack
except ImportError:
from flask import _request_ctx_stack as stack
__maintainer__ = "David Baumgold <david@davidbaumgold.com>"
[docs]def make_google_blueprint(
client_id=None, client_secret=None, scope=None,
offline=False, reprompt_consent=False,
redirect_url=None, redirect_to=None, login_url=None, authorized_url=None,
session_class=None, backend=None, hosted_domain=None):
"""
Make a blueprint for authenticating with Google using OAuth 2. This requires
a client ID and client secret from Google. You should either pass them to
this constructor, or make sure that your Flask application config defines
them, using the variables GOOGLE_OAUTH_CLIENT_ID and GOOGLE_OAUTH_CLIENT_SECRET.
Args:
client_id (str): The client ID for your application on Google
client_secret (str): The client secret for your application on Google
scope (str, optional): comma-separated list of scopes for the OAuth token.
Defaults to the "https://www.googleapis.com/auth/userinfo.profile" scope.
offline (bool): Whether to request `offline access
<https://developers.google.com/accounts/docs/OAuth2WebServer#offline>`_
for the OAuth token. Defaults to False
reprompt_consent (bool): If True, force Google to re-prompt the user
for their consent, even if the user has already given their
consent. Defaults to False
redirect_url (str): the URL to redirect to after the authentication
dance is complete
redirect_to (str): if ``redirect_url`` is not defined, the name of the
view to redirect to after the authentication dance is complete.
The actual URL will be determined by :func:`flask.url_for`
login_url (str, optional): the URL path for the ``login`` view.
Defaults to ``/google``
authorized_url (str, optional): the URL path for the ``authorized`` view.
Defaults to ``/google/authorized``.
session_class (class, optional): The class to use for creating a
Requests session. Defaults to
:class:`~flask_dance.consumer.requests.OAuth2Session`.
backend: A storage backend class, or an instance of a storage
backend class, to use for this blueprint. Defaults to
:class:`~flask_dance.consumer.backend.session.SessionBackend`.
hosted_domain (str, optional): The domain of the G Suite user. Used to indicate that the account selection UI
should be optimized for accounts at this domain. Note that this only provides UI optimization, and requires
response validation (see warning).
.. _google_hosted_domain_warning:
.. warning::
The ``hosted_domain`` argument **only provides UI optimization**. Don't rely on this argument to control
who can access your application. You must verify that the ``hd`` claim of the response ID token matches the
``hosted_domain`` argument passed to ``make_google_blueprint``. For example:
.. code-block:: python
from flask import session, abort
from flask_dance.consumer import oauth_authorized
from flask_dance.contrib.google import make_google_blueprint, google
import requests
google_bp = make_google_blueprint(
client_id="foo",
client_secret="bar",
scope=["profile", "email"],
hosted_domain="example.com"
)
@oauth_authorized.connect_via(google_bp)
def logged_in(blueprint, token):
resp_json = google.get("/oauth2/v2/userinfo").json()
if resp_json["hd"] != blueprint.authorization_url_params["hd"]:
requests.post(
"https://accounts.google.com/o/oauth2/revoke",
params={"token": token["access_token"]}
)
session.clear()
abort(403)
:rtype: :class:`~flask_dance.consumer.OAuth2ConsumerBlueprint`
:returns: A :ref:`blueprint <flask:blueprints>` to attach to your Flask app.
"""
scope = scope or ["https://www.googleapis.com/auth/userinfo.profile"]
authorization_url_params = {}
auto_refresh_url = None
if offline:
authorization_url_params["access_type"] = "offline"
auto_refresh_url = "https://accounts.google.com/o/oauth2/token"
if reprompt_consent:
authorization_url_params["approval_prompt"] = "force"
if hosted_domain:
authorization_url_params["hd"] = hosted_domain
google_bp = OAuth2ConsumerBlueprint("google", __name__,
client_id=client_id,
client_secret=client_secret,
scope=scope,
base_url="https://www.googleapis.com/",
authorization_url="https://accounts.google.com/o/oauth2/auth",
token_url="https://accounts.google.com/o/oauth2/token",
auto_refresh_url=auto_refresh_url,
redirect_url=redirect_url,
redirect_to=redirect_to,
login_url=login_url,
authorized_url=authorized_url,
authorization_url_params=authorization_url_params,
session_class=session_class,
backend=backend,
)
google_bp.from_config["client_id"] = "GOOGLE_OAUTH_CLIENT_ID"
google_bp.from_config["client_secret"] = "GOOGLE_OAUTH_CLIENT_SECRET"
@google_bp.before_app_request
def set_applocal_session():
ctx = stack.top
ctx.google_oauth = google_bp.session
return google_bp
google = LocalProxy(partial(_lookup_app_object, "google_oauth"))