Add fastapi contrib

rollbar/contrib/fastapi/__init__.py

@@ -0,0 +1,23 @@
+"""Integration with fastapi.
+
+See: https://fastapi.tiangolo.com/
+"""
+
+import rollbar
+
+from fastapi import Request
+
+
+def report_exception(request: Request):
+    rollbar.report_exc_info(request=request)
+
+
+def _hook(request, data):
+    data["framework"] = "fastapi"
+
+    if request:
+        endpoint = request.scope["endpoint"]
+        data["context"] = f"{endpoint.__module__}.{endpoint.__name__}"
+
+
+rollbar.BASE_DATA_HOOK = _hook

rollbar/examples/fastapi/app.py

@@ -0,0 +1,37 @@
+import rollbar
+import rollbar.contrib.fastapi
+import uvicorn
+
+from fastapi import FastAPI, Request, status
+from fastapi.responses import JSONResponse, Response
+
+app = FastAPI()
+
+
+rollbar.init("ACCESS_TOKEN", environment="development")
+
+
+@app.exception_handler(Exception)
+async def handle_unexpected_exceptions(request: Request, exc: Exception):
+    """This won't capture HTTPException."""
+    try:
+        raise exc
+    except Exception:
+        rollbar.contrib.fastapi.report_exception(request=request)
+
+    return Response(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+
+@app.get("/")
+async def root(raise_exception: bool = False):
+    """Hello world api endpoint.
+
+    Use `?raise_exception=1` to raise an exception.
+    """
+    if raise_exception:
+        raise Exception("Testing exceptions")
+    return JSONResponse({"message": "Hello World"})
+
+
+if __name__ == "__main__":
+    uvicorn.run(app)

setup.py

@@ -1,7 +1,8 @@
-import re
 import os.path
+import re
 import sys
-from setuptools import setup, find_packages
+
+from setuptools import find_packages, setup
 
 HERE = os.path.abspath(os.path.dirname(__file__))
 
@@ -65,6 +66,7 @@
         "Framework :: Bottle",
         "Framework :: Django",
         "Framework :: Flask",
+        "Framework :: Fastapi",
         "Framework :: Pylons",
         "Framework :: Pyramid",
         "Framework :: Twisted",
@@ -81,5 +83,11 @@
         'requests>=0.12.1',
         'six>=1.9.0'
     ],
+    extra_requires={
+        "fastapi": [
+            "fastapi",
+            "uvicorn",
+        ],
+    },
     tests_require=tests_require,
     )