from apistar import App as BaseApp
from apistar_sentry import SentryComponent, SentryMixin


class App(SentryMixin, BaseApp):
    pass


COMPONENTS = [
    SentryComponent("YOUR_DSN_GOES_HERE", environment="example", version="0.1.0"),
]

ROUTES = [
    # ...
]

app = App(components=COMPONENTS, routes=ROUTES)

class SentryHooks:
  def on_request(self, account: Account, sentry: Sentry = None) -> None:
    if sentry is not None:
      sentry.track_user(account.to_dict())

  def on_response(self, response: http.Response, sentry: Sentry = None) -> http.Response:
    if sentry is not None:
      sentry.clear_user()
    return response

  on_error = on_response