Move type comments

sentry_sdk/utils.py

@@ -32,7 +32,7 @@
 from sentry_sdk.consts import DEFAULT_MAX_VALUE_LENGTH, EndpointType
 
 if TYPE_CHECKING:
     from collections.abc import Awaitable
 
     from types import FrameType, TracebackType
     from typing import (
@@ -51,12 +51,12 @@
         TypeVar,
         Union,
     )
     from sentry_sdk.integrations import Integration
 
     from sentry_sdk._types import Event, ExcInfo
 
     P = ParamSpec("P")
     R = TypeVar("R")
 
 
 epoch = datetime(1970, 1, 1)
@@ -1635,6 +1635,7 @@
     integration,  # type: type[Integration]
     original_function,  # type: Callable[P, R]
 ):
+    # type: (...) -> Callable[[Callable[P, R]], Callable[P, R]]
     """
     Ensures a given integration is enabled prior to calling a Sentry-patched function.
 
@@ -1657,46 +1658,45 @@
     ```
     """
 
-    # type: (...) -> Callable[[Callable[P, R]], Callable[P, R]]
     def patcher(sentry_patched_function):
         # type: (Callable[P, R]) -> Callable[P, R]
         @wraps(original_function)
         def runner(*args, **kwargs):
             # type: (*object, **object) -> R
             if sentry_sdk.get_client().get_integration(integration) is None:
                 return original_function(*args, **kwargs)
 
             return sentry_patched_function(*args, **kwargs)
 
         return runner
 
     return patcher
 
 
 def ensure_integration_enabled_async(
     integration,  # type: type[Integration]
     original_function,  # type: Callable[P, Awaitable[R]]
 ):
+    # type: (...) -> Callable[[Callable[P, Awaitable[R]]], Callable[P, Awaitable[R]]]
     """
     Version of `ensure_integration_enabled` for decorating async functions.
 
     Please refer to the `ensure_integration_enabled` documentation for more information.
     """
 
-    # type: (...) -> Callable[[Callable[P, Awaitable[R]]], Callable[P, Awaitable[R]]]
     def patcher(sentry_patched_function):
         # type: (Callable[P, Awaitable[R]]) -> Callable[P, Awaitable[R]]
         @wraps(original_function)
         async def runner(*args, **kwargs):
             # type: (*object, **object) -> R
             if sentry_sdk.get_client().get_integration(integration) is None:
                 return await original_function(*args, **kwargs)
 
             return await sentry_patched_function(*args, **kwargs)
 
         return runner
 
     return patcher
 
 
 if PY37: