Update litestar integration

Author: vrslev

sentry_sdk/integrations/litestar.py

@@ -1,6 +1,7 @@
+from collections.abc import Set
 import sentry_sdk
 from sentry_sdk.consts import OP
-from sentry_sdk.integrations import DidNotEnable, Integration
+from sentry_sdk.integrations import _DEFAULT_FAILED_REQUEST_STATUS_CODES, DidNotEnable, Integration
 from sentry_sdk.integrations.asgi import SentryAsgiMiddleware
 from sentry_sdk.integrations.logging import ignore_logger
 from sentry_sdk.scope import should_send_default_pii
@@ -17,6 +18,7 @@
     from litestar.middleware import DefineMiddleware  # type: ignore
     from litestar.routes.http import HTTPRoute  # type: ignore
     from litestar.data_extractors import ConnectionDataExtractor  # type: ignore
+    from litestar.exceptions import HTTPException  # type: ignore
 except ImportError:
     raise DidNotEnable("Litestar is not installed")
 
@@ -45,6 +47,11 @@ class LitestarIntegration(Integration):
     identifier = "litestar"
     origin = f"auto.http.{identifier}"
 
+    def __init__(self,
+    failed_request_status_codes=_DEFAULT_FAILED_REQUEST_STATUS_CODES,  # type: Set[int]
+                 ) -> None:
+        self.failed_request_status_codes = failed_request_status_codes
+
     @staticmethod
     def setup_once():
         # type: () -> None
@@ -277,6 +284,13 @@ def exception_handler(exc, scope):
         sentry_scope = sentry_sdk.get_isolation_scope()
         sentry_scope.set_user(user_info)
 
+    if isinstance(exc, HTTPException):
+        integration = sentry_sdk.get_client().get_integration(
+            LitestarIntegration
+        )
+        if integration is not None and exc.status_code not in integration.failed_request_status_codes:
+                return
+
     event, hint = event_from_exception(
         exc,
         client_options=sentry_sdk.get_client().options,