Fix typing issue with optional websockets import. (#376)

tobiasraabe · web-flow · commit e4f5b5a6f92e · 2025-09-29T16:10:38.000+02:00
diff --git a/ariadne_codegen/client_generators/dependencies/async_base_client.py b/ariadne_codegen/client_generators/dependencies/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/ariadne_codegen/client_generators/dependencies/async_base_client_open_telemetry.py b/ariadne_codegen/client_generators/dependencies/async_base_client_open_telemetry.py
@@ -30,7 +30,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/client_forward_refs/expected_client/async_base_client.py b/tests/main/clients/client_forward_refs/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/client_forward_refs_shorter_results/expected_client/async_base_client.py b/tests/main/clients/client_forward_refs_shorter_results/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/custom_config_file/expected_client/async_base_client.py b/tests/main/clients/custom_config_file/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/custom_files_names/expected_client/async_base_client.py b/tests/main/clients/custom_files_names/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/custom_query_builder/expected_client/async_base_client.py b/tests/main/clients/custom_query_builder/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/custom_scalars/expected_client/async_base_client.py b/tests/main/clients/custom_scalars/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/example/expected_client/async_base_client.py b/tests/main/clients/example/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/extended_models/expected_client/async_base_client.py b/tests/main/clients/extended_models/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/fragments_on_abstract_types/expected_client/async_base_client.py b/tests/main/clients/fragments_on_abstract_types/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/inline_fragments/expected_client/async_base_client.py b/tests/main/clients/inline_fragments/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/interface_as_fragment/expected_client/async_base_client.py b/tests/main/clients/interface_as_fragment/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/multiple_fragments/expected_client/async_base_client.py b/tests/main/clients/multiple_fragments/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/only_used_inputs_and_enums/expected_client/async_base_client.py b/tests/main/clients/only_used_inputs_and_enums/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/operations/expected_client/async_base_client.py b/tests/main/clients/operations/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/remote_schema/expected_client/async_base_client.py b/tests/main/clients/remote_schema/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/clients/shorter_results/expected_client/async_base_client.py b/tests/main/clients/shorter_results/expected_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
diff --git a/tests/main/custom_operation_builder/graphql_client/async_base_client.py b/tests/main/custom_operation_builder/graphql_client/async_base_client.py
@@ -19,7 +19,7 @@
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]

Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,7 @@`
`19`	`19`	`from websockets import ( # type: ignore[import-not-found,unused-ignore]`
`20`	`20`	`ClientConnection,`
`21`	`21`	`)`
`22`		`- from websockets import (`
	`22`	`+ from websockets import ( # type: ignore[import-not-found,unused-ignore]`
`23`	`23`	`connect as ws_connect,`
`24`	`24`	`)`
`25`	`25`	`from websockets.typing import ( # type: ignore[import-not-found,unused-ignore]`
Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@`
`30`	`30`	`from websockets import ( # type: ignore[import-not-found,unused-ignore]`
`31`	`31`	`ClientConnection,`
`32`	`32`	`)`
`33`		`- from websockets import (`
	`33`	`+ from websockets import ( # type: ignore[import-not-found,unused-ignore]`
`34`	`34`	`connect as ws_connect,`
`35`	`35`	`)`
`36`	`36`	`from websockets.typing import ( # type: ignore[import-not-found,unused-ignore]`