Reorder functions logically in ClientConnection.

Define callers after callees.

Author: aaugustin

src/websockets/asyncio/client.py

@@ -295,19 +295,6 @@ def factory() -> ClientConnection:
 
         self._open_timeout = open_timeout
 
-    # async with connect(...) as ...: ...
-
-    async def __aenter__(self) -> ClientConnection:
-        return await self
-
-    async def __aexit__(
-        self,
-        exc_type: type[BaseException] | None,
-        exc_value: BaseException | None,
-        traceback: TracebackType | None,
-    ) -> None:
-        await self.connection.close()
-
     # ... = await connect(...)
 
     def __await__(self) -> Generator[Any, None, ClientConnection]:
@@ -333,6 +320,19 @@ async def __await_impl__(self) -> ClientConnection:
 
     __iter__ = __await__
 
+    # async with connect(...) as ...: ...
+
+    async def __aenter__(self) -> ClientConnection:
+        return await self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None:
+        await self.connection.close()
+
 
 def unix_connect(
     path: str | None = None,