Restore real imports for compatibility with mypy.

Fix #940.

Author: aaugustin

docs/api/client.rst

@@ -1,7 +1,7 @@
 Client
 ======
 
-.. automodule:: websockets.legacy.client
+.. automodule:: websockets.client
 
     Opening a connection
     --------------------

docs/api/index.rst

@@ -46,5 +46,10 @@ both in the client API and server API.
    utilities
 
 All public APIs can be imported from the :mod:`websockets` package, unless
-noted otherwise. Anything that isn't listed in this API documentation is a
-private API, with no guarantees of behavior or backwards-compatibility.
+noted otherwise. This convenience feature is incompatible with static code
+analysis tools such as mypy_, though.
+
+.. _mypy: https://github.com/python/mypy
+
+Anything that isn't listed in this API documentation is a private API. There's
+no guarantees of behavior or backwards-compatibility for private APIs.

docs/api/server.rst

@@ -1,7 +1,7 @@
 Server
 ======
 
-.. automodule:: websockets.legacy.server
+.. automodule:: websockets.server
 
     Starting a server
     -----------------
@@ -90,7 +90,7 @@ Server
 Basic authentication
 --------------------
 
-.. automodule:: websockets.legacy.auth
+.. automodule:: websockets.auth
 
     .. autofunction:: basic_auth_protocol_factory
 

docs/changelog.rst

@@ -25,11 +25,13 @@ fixing regressions shortly after a release.
 Only documented APIs are public. Undocumented APIs are considered private.
 They may change at any time.
 
-9.1
-...
+9.0.2
+.....
 
 *In development*
 
+* Restored compatibility with mypy.
+
 9.0.1
 .....
 

docs/extensions.rst

@@ -14,9 +14,8 @@ specification, WebSocket Per-Message Deflate, specified in :rfc:`7692`.
 Per-Message Deflate
 -------------------
 
-:func:`~websockets.legacy.client.connect` and
-:func:`~websockets.legacy.server.serve` enable the Per-Message Deflate
-extension by default.
+:func:`~websockets.client.connect` and :func:`~websockets.server.serve` enable
+the Per-Message Deflate extension by default.
 
 If you want to disable it, set ``compression=None``::
 

src/websockets/auth.py

@@ -0,0 +1,2 @@
+# See #940 for why lazy_import isn't used here for backwards compatibility.
+from .legacy.auth import *  # noqa

src/websockets/client.py

@@ -24,7 +24,6 @@
 )
 from .http import USER_AGENT, build_host
 from .http11 import Request, Response
-from .imports import lazy_import
 from .typing import (
     ConnectionOption,
     ExtensionHeader,
@@ -35,15 +34,9 @@
 from .uri import parse_uri
 from .utils import accept_key, generate_key
 
+# See #940 for why lazy_import isn't used here for backwards compatibility.
+from .legacy.client import * # noqa
 
-lazy_import(
-    globals(),
-    aliases={
-        "connect": ".legacy.client",
-        "unix_connect": ".legacy.client",
-        "WebSocketClientProtocol": ".legacy.client",
-    },
-)
 
 __all__ = ["ClientConnection"]
 

src/websockets/server.py

@@ -26,7 +26,6 @@
 )
 from .http import USER_AGENT
 from .http11 import Request, Response
-from .imports import lazy_import
 from .typing import (
     ConnectionOption,
     ExtensionHeader,
@@ -36,16 +35,8 @@
 )
 from .utils import accept_key
 
-
-lazy_import(
-    globals(),
-    aliases={
-        "serve": ".legacy.server",
-        "unix_serve": ".legacy.server",
-        "WebSocketServerProtocol": ".legacy.server",
-        "WebSocketServer": ".legacy.server",
-    },
-)
+# See #940 for why lazy_import isn't used here for backwards compatibility.
+from .legacy.server import * # noqa
 
 
 __all__ = ["ServerConnection"]