Add Flask Helper to Error Reporting

Author: Bill Prin

packages/google-cloud-error-reporting/google/cloud/error_reporting/__init__.py

@@ -16,3 +16,5 @@
 
 
 from google.cloud.error_reporting.client import Client
+from google.cloud.error_reporting.client import HTTPContext
+from google.cloud.error_reporting.util import build_flask_context

packages/google-cloud-error-reporting/google/cloud/error_reporting/util.py

@@ -0,0 +1,44 @@
+# Copyright 2016 Google Inc. All Rights Reserved.
+#
+# 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.
+
+"""Utility functions for Stackdriver Error Reporting."""
+
+from google.cloud.error_reporting.client import HTTPContext
+
+
+def build_flask_context(request):
+    """Builds an HTTP context object from a Flask (Werkzeug) request object.
+
+     This helper method extracts the relevant HTTP context from a Flask request
+     object into an object ready to be sent to Error Reporting.
+
+    .. code-block:: python
+
+        >>> @app.errorhandler(HTTPException)
+        ... def handle_error(exc):
+        ...     client.report_exception(
+        ...         http_context=build_flask_context(request))
+        ...     # rest of error response code here
+
+    :type request: :class:`werkzeug.wrappers.request`
+    :param request: The Flask request object to convert.
+
+    :rtype: :class:`~google.cloud.error_reporting.client.HTTPContext`
+    :returns: An HTTPContext object ready to be sent to the Stackdriver Error
+              Reporting API.
+    """
+    return HTTPContext(url=request.url, method=request.method,
+                       user_agent=request.user_agent.string,
+                       referrer=request.referrer,
+                       remote_ip=request.remote_addr)

packages/google-cloud-error-reporting/tox.ini

@@ -10,6 +10,7 @@ localdeps =
 deps =
     {toxinidir}/../core
     {toxinidir}/../logging
+    mock
     pytest
 covercmd =
     py.test --quiet \

packages/google-cloud-error-reporting/unit_tests/test_util.py

@@ -0,0 +1,48 @@
+# Copyright 2016 Google Inc. All Rights Reserved.
+#
+# 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 unittest
+
+
+class Test_build_flask_context(unittest.TestCase):
+
+    def _call_fut(self, request):
+        from google.cloud.error_reporting.util import build_flask_context
+        return build_flask_context(request)
+
+    def test_flask_helper(self):
+        import mock
+
+        user_agent = mock.Mock(string='Google Cloud Unit Tests Agent')
+        request = _Request('http://google.com', 'GET',
+                           user_agent,
+                           'http://gmail.com',
+                           '127.0.0.1')
+
+        context = self._call_fut(request)
+        self.assertEqual(request.url, context.url)
+        self.assertEqual(request.method, context.method)
+        self.assertEqual(request.user_agent.string, context.userAgent)
+        self.assertEqual(request.referrer, context.referrer)
+        self.assertEqual(request.remote_addr, context.remoteIp)
+
+
+class _Request(object):
+
+    def __init__(self, url, method, user_agent, referrer, remote_addr):
+        self.url = url
+        self.method = method
+        self.user_agent = user_agent
+        self.referrer = referrer
+        self.remote_addr = remote_addr