getsentry · untitaker · Mar 29, 2019 · Mar 1, 2019 · Mar 20, 2019 · Mar 26, 2019
@@ -0,0 +1,167 @@
+from __future__ import absolute_import
+
+from sentry_sdk.hub import Hub
+from sentry_sdk.utils import (
+    capture_internal_exceptions,
+    event_from_exception,
+    transaction_from_function,
+)
+from sentry_sdk.integrations import Integration
+from sentry_sdk.integrations.wsgi import SentryWsgiMiddleware
+from sentry_sdk.integrations._wsgi_common import RequestExtractor
+
+if False:
+    from sentry_sdk.integrations.wsgi import _ScopedResponse
+    from typing import Any
+    from typing import Dict
+    from typing import Callable
+    from typing import Optional
+    from bottle import FileUpload, FormsDict, LocalRequest  # type: ignore
+
+from bottle import Bottle, Route, request as bottle_request  # type: ignore
+
+
+class BottleIntegration(Integration):
+    identifier = "bottle"
+
+    transaction_style = None
+
+    def __init__(self, transaction_style="endpoint"):
+        # type: (str) -> None
+        TRANSACTION_STYLE_VALUES = ("endpoint", "url")
+        if transaction_style not in TRANSACTION_STYLE_VALUES:
+            raise ValueError(
+                "Invalid value for transaction_style: %s (must be in %s)"
+                % (transaction_style, TRANSACTION_STYLE_VALUES)
+            )
+        self.transaction_style = transaction_style
+
+    @staticmethod
+    def setup_once():
+        # type: () -> None
+
+        # monkey patch method Bottle.__call__
+        old_app = Bottle.__call__
+
+        def sentry_patched_wsgi_app(self, environ, start_response):
+            # type: (Any, Dict[str, str], Callable) -> _ScopedResponse
+
+            hub = Hub.current
+            integration = hub.get_integration(BottleIntegration)
+            if integration is None:
+                return old_app(self, environ, start_response)
+
+            return SentryWsgiMiddleware(lambda *a, **kw: old_app(self, *a, **kw))(
+                environ, start_response
+            )
+
+        Bottle.__call__ = sentry_patched_wsgi_app  # type: ignore
+
+        # monkey patch method Bottle._handle
+        old_handle = Bottle._handle
+
+        def _patched_handle(self, environ):
+            hub = Hub.current
+            integration = hub.get_integration(BottleIntegration)
+            if integration is None:
+                return old_handle(self, environ)
+
+            # create new scope
+            scope_manager = hub.push_scope()
+
+            with scope_manager:
+                app = self
+                with hub.configure_scope() as scope:
+                    scope._name = "bottle"
+                    scope.add_event_processor(
+                        _make_request_event_processor(app, bottle_request, integration)
+                    )
+                res = old_handle(self, environ)
+
+            # scope cleanup
+            return res
+
+        Bottle._handle = _patched_handle
+
+        # monkey patch method Route._make_callback
+        old_make_callback = Route._make_callback
+
+        def patched_make_callback(self, *args, **kwargs):
+            hub = Hub.current
+            integration = hub.get_integration(BottleIntegration)
+            prepared_callback = old_make_callback(self, *args, **kwargs)
+            if integration is None:
+                return prepared_callback
+
+            def wrapped_callback(*args, **kwargs):
+                try:
+                    res = prepared_callback(*args, **kwargs)
+                except Exception as exception:
+                    hub = Hub.current
+                    event, hint = event_from_exception(
+                        exception,
+                        client_options=hub.client.options,
+                        mechanism={"type": "bottle", "handled": False},
+                    )
+                    hub.capture_event(event, hint=hint)
+                    raise exception
+
+                return res
+
+            return wrapped_callback
+
+        Route._make_callback = patched_make_callback
+
+
+class BottleRequestExtractor(RequestExtractor):
+    def env(self):
+        # type: () -> Dict[str, str]
+        return self.request.environ
+
+    def cookies(self):
+        # type: () -> Dict[str, str]
+        return self.request.cookies
+
+    def raw_data(self):
+        # type: () -> bytes
+        return self.request.body.read()
+
+    def form(self):
+        # type: () -> FormsDict
+        if self.is_json():
+            return None
+        return self.request.forms.decode()
+
+    def files(self):
+        # type: () -> Optional[Dict[str, str]]
+        if self.is_json():
+            return None
+
+        return self.request.files
+
+    def size_of_file(self, file):
+        # type: (FileUpload) -> int
+        return file.content_length
+
+
+def _make_request_event_processor(app, request, integration):
+    # type: (Bottle, LocalRequest, BottleIntegration) -> Callable
+    def inner(event, hint):
+        # type: (Dict[str, Any], Dict[str, Any]) -> Dict[str, Any]
+
+        try:
+            if integration.transaction_style == "endpoint":
+                event["transaction"] = request.route.name or transaction_from_function(
+                    request.route.callback
+                )
+            elif integration.transaction_style == "url":
+                event["transaction"] = request.route.rule  # type: ignore
+        except Exception:
+            pass
+
+        with capture_internal_exceptions():
+            BottleRequestExtractor(request).extract_into_event(event)
+
+        return event
+
+    return inner
@@ -22,7 +22,10 @@
     zip_safe=False,
     license="BSD",
     install_requires=["urllib3", "certifi"],
-    extras_require={"flask": ["flask>=0.8", "blinker>=1.1"]},
+    extras_require={
+        "flask": ["flask>=0.8", "blinker>=1.1"],
+        "bottle": ["bottle>=0.12.13"],
+    },
     classifiers=[
         "Development Status :: 5 - Production/Stable",
         "Environment :: Web Environment",

@@ -4,4 +4,4 @@ pytest-xdist==1.23.0
 tox==3.7.0
 Werkzeug==0.14.1
 pytest-localserver==0.4.1
 pytest-cov==2.6.0
+pytest-cov==2.6.0