# Copyright 2015 Google Inc.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

import json

import os

import logging

from flask import current_app, Flask, redirect, request, session, url_for

import httplib2

from pymemcache.client.base import Client as MemcacheClient

from flask.ext.session import MemcachedSessionInterface

# [START include]

from oauth2client.contrib.flask_util import UserOAuth2

oauth2 = UserOAuth2()

# [END include]

# [START client]

memcache_addr = os.environ.get('MEMCACHE_PORT_11211_TCP_ADDR', 'localhost')

memcache_port = os.environ.get('MEMCACHE_PORT_11211_TCP_PORT', 11211)

memcache_client = MemcacheClient((memcache_addr, int(memcache_port)))

# [END client]

def create_app(config, debug=False, testing=False, config_overrides=None):

app = Flask(__name__)

app.config.from_object(config)

app.session_interface = MemcachedSessionInterface(memcache_client, 'sess')

app.debug = debug

app.testing = testing

if config_overrides:

app.config.update(config_overrides)

# Configure logging

if not app.testing:

logging.basicConfig(level=logging.INFO)

# Setup the data model.

with app.app_context():

model = get_model()

model.init_app(app)

# [START init_app]

# Initalize the OAuth2 helper.

oauth2.init_app(

app,

scopes=['email', 'profile'],

authorize_callback=_request_user_info)

# [END init_app]

# [START logout]

# Add a logout handler.

@app.route('/logout')

def logout():

# Delete the user's profile and the credentials stored by oauth2.

del session['profile']

session.modified = True

oauth2.storage.delete()

return redirect(request.referrer or '/')

# [END logout]

# Register the Bookshelf CRUD blueprint.

from .crud import crud

app.register_blueprint(crud, url_prefix='/books')

# Add a default root route.

@app.route("/")

def index():

return redirect(url_for('crud.list'))

# Add an error handler. This is useful for debugging the live application,

# however, you should disable the output of the exception for production

# applications.

@app.errorhandler(500)

def server_error(e):

return """

An internal error occurred: <pre>{}</pre>

See logs for full stacktrace.

""".format(e), 500

return app

def get_model():

model_backend = current_app.config['DATA_BACKEND']

if model_backend == 'cloudsql':

from . import model_cloudsql

model = model_cloudsql

elif model_backend == 'datastore':

from . import model_datastore

model = model_datastore

elif model_backend == 'mongodb':

from . import model_mongodb

model = model_mongodb

else:

raise ValueError(

"No appropriate databackend configured. "

"Please specify datastore, cloudsql, or mongodb")

return model

# [START request_user_info]

def _request_user_info(credentials):

"""

Makes an HTTP request to the Google+ API to retrieve the user's basic

profile information, including full name and photo, and stores it in the

Flask session.

"""

http = httplib2.Http()

credentials.authorize(http)

resp, content = http.request(

'https://www.googleapis.com/plus/v1/people/me')

if resp.status != 200:

current_app.logger.error(

"Error while obtaining user profile: %s" % resp)

return None

session['profile'] = json.loads(content.decode('utf-8'))

# [END request_user_info]