[DO NOT MERGE] Add memcache sessions

Bill Prin · Bill Prin · commit 9aa3055b0eeb · 2016-05-13T15:35:00.000-07:00
diff --git a/4-auth/bookshelf/__init__.py b/4-auth/bookshelf/__init__.py
@@ -13,10 +13,15 @@
 # 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
@@ -25,10 +30,22 @@
 # [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
 
diff --git a/4-auth/requirements.txt b/4-auth/requirements.txt
@@ -7,3 +7,5 @@ PyMySQL==0.7.2
 Flask-PyMongo==0.4.1
 PyMongo==3.2.2
 six==1.10.0
+Flask-Session
+pymemcache==1.3.5
