Use Stackdriver Logging Handlers (GoogleCloudPlatform#91)

waprin · Jon Wayne Parrott · commit e488895dc918 · 2016-09-27T09:25:07.000-07:00
diff --git a/5-logging/bookshelf/__init__.py b/5-logging/bookshelf/__init__.py
@@ -16,6 +16,8 @@
 import logging
 
 from flask import current_app, Flask, redirect, request, session, url_for
+import google.cloud.logging
+from google.cloud.logging.handlers import CloudLoggingHandler, setup_logging
 import httplib2
 from oauth2client.contrib.flask_util import UserOAuth2
 
@@ -34,9 +36,12 @@ def create_app(config, debug=False, testing=False, config_overrides=None):
         app.config.update(config_overrides)
 
     # [START setup_logging]
-    # Configure logging
     if not app.testing:
-        logging.basicConfig(level=logging.INFO)
+        client = google.cloud.logging.Client(app.config['PROJECT_ID'])
+        handler = CloudLoggingHandler(client)
+        # Attaches the handler to the root logger
+        setup_logging(handler)
+        logging.getLogger().setLevel(logging.INFO)
     # [END setup_logging]
 
     # Setup the data model.
diff --git a/6-pubsub/bookshelf/__init__.py b/6-pubsub/bookshelf/__init__.py
@@ -16,6 +16,8 @@
 import logging
 
 from flask import current_app, Flask, redirect, request, session, url_for
+import google.cloud.logging
+from google.cloud.logging.handlers import CloudLoggingHandler, setup_logging
 import httplib2
 from oauth2client.contrib.flask_util import UserOAuth2
 
@@ -35,7 +37,11 @@ def create_app(config, debug=False, testing=False, config_overrides=None):
 
     # Configure logging
     if not app.testing:
-        logging.basicConfig(level=logging.INFO)
+        client = google.cloud.logging.Client(app.config['PROJECT_ID'])
+        handler = CloudLoggingHandler(client)
+        # Attaches the handler to the root logger
+        setup_logging(handler)
+        logging.getLogger().setLevel(logging.INFO)
 
     # Setup the data model.
     with app.app_context():
diff --git a/7-gce/bookshelf/__init__.py b/7-gce/bookshelf/__init__.py
@@ -16,6 +16,8 @@
 import logging
 
 from flask import current_app, Flask, redirect, request, session, url_for
+import google.cloud.logging
+from google.cloud.logging.handlers import CloudLoggingHandler, setup_logging
 import httplib2
 from oauth2client.contrib.flask_util import UserOAuth2
 
@@ -35,7 +37,11 @@ def create_app(config, debug=False, testing=False, config_overrides=None):
 
     # Configure logging
     if not app.testing:
-        logging.basicConfig(level=logging.INFO)
+        client = google.cloud.logging.Client(app.config['PROJECT_ID'])
+        handler = CloudLoggingHandler(client)
+        # Attaches the handler to the root logger
+        setup_logging(handler)
+        logging.getLogger().setLevel(logging.INFO)
 
     # Setup the data model.
     with app.app_context():
