Avoid depending on typing_extensions where not needed

mosquito · Nov 27, 2024 · 1de6e3d · 1de6e3d
1 parent e610621
commit 1de6e3d
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/aio_pika/tools.py b/aio_pika/tools.py
@@ -2,6 +2,7 @@
 
 import asyncio
 import inspect
+import typing
 import warnings
 from functools import wraps
 from itertools import chain
@@ -11,10 +12,16 @@
     List,
     MutableSet, Optional, TypeVar, Union, Generic,
 )
-
-from typing_extensions import ParamSpec, Protocol
 from weakref import ReferenceType, WeakSet, ref
 
+if typing.TYPE_CHECKING:
+    from typing_extensions import ParamSpec, Protocol
+else:
+    try:
+        from typing import ParamSpec, Protocol
+    except ImportError:
+        from typing_extensions import ParamSpec, Protocol
+
 from aio_pika.log import get_logger