sentry.py

from typing import Callable

import raven


_CLIENT = raven.Client()


def track_exceptions(func: Callable) -> Callable:
    def tracked_call(*args, **kwargs):
        try:
            return func(*args, **kwargs)
        except:
            _CLIENT.captureException()
            raise
    return tracked_call


def setup(sentry_dsn: str) -> None:
    _CLIENT.set_dsn(sentry_dsn)