Fix Protocol imports on >=3.10

ariebovenberg · Jan 3, 2023 · c77c27a · c77c27a
1 parent 309adab
commit c77c27a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/slotscheck/cli.py b/src/slotscheck/cli.py
@@ -19,7 +19,6 @@
 )
 
 import click
-from typing_extensions import Protocol
 
 from . import config
 from .checks import (
@@ -52,6 +51,11 @@
     walk_classes,
 )
 
+try:
+    from typing import Protocol
+except ImportError:  # pragma: no cover
+    from typing_extensions import Protocol
+
 
 @click.command("slotscheck")
 @click.argument(

diff --git a/tests/examples/module_not_ok/foo.py b/tests/examples/module_not_ok/foo.py
@@ -1,4 +1,8 @@
-from typing_extensions import Protocol
+
+try:
+    from typing import Protocol
+except ImportError:
+    from typing_extensions import Protocol
 
 
 class A: