lazy import _overlapped

youknowone · youknowone · commit ede4124f9d07 · 2024-05-08T18:46:51.000+09:00
diff --git a/Lib/asyncio/windows_events.py b/Lib/asyncio/windows_events.py
@@ -5,7 +5,7 @@
 if sys.platform != 'win32':  # pragma: no cover
     raise ImportError('win32 only')
 
-import _overlapped
+# import _overlapped; XXX: RUSTPYTHON
 import _winapi
 import errno
 from functools import partial
@@ -132,6 +132,7 @@ def _unregister_wait_cb(self, fut):
         self._ov = None
 
     def _unregister_wait(self):
+        import _overlapped
         if not self._registered:
             return
         self._registered = False
@@ -194,6 +195,7 @@ def set_exception(self, exception):
 
 class _WaitHandleFuture(_BaseWaitHandleFuture):
     def __init__(self, ov, handle, wait_handle, proactor, *, loop=None):
+        import _overlapped  # XXX: RUSTPYTHON
         super().__init__(ov, handle, wait_handle, loop=loop)
         self._proactor = proactor
         self._unregister_proactor = True
@@ -219,6 +221,7 @@ def _unregister_wait_cb(self, fut):
         super()._unregister_wait_cb(fut)
 
     def _unregister_wait(self):
+        import _overlapped  # XXX: RUSTPYTHON
         if not self._registered:
             return
         self._registered = False
@@ -417,6 +420,7 @@ class IocpProactor:
     """Proactor implementation using IOCP."""
 
     def __init__(self, concurrency=INFINITE):
+        import _overlapped  # XXX: RUSTPYTHON
         self._loop = None
         self._results = []
         self._iocp = _overlapped.CreateIoCompletionPort(
@@ -458,6 +462,7 @@ def _result(self, value):
 
     @staticmethod
     def finish_socket_func(trans, key, ov):
+        import _overlapped  # XXX: RUSTPYTHON
         try:
             return ov.getresult()
         except OSError as exc:
@@ -469,6 +474,7 @@ def finish_socket_func(trans, key, ov):
 
     @classmethod
     def _finish_recvfrom(cls, trans, key, ov, *, empty_result):
+        import _overlapped  # XXX: RUSTPYTHON
         try:
             return cls.finish_socket_func(trans, key, ov)
         except OSError as exc:
@@ -480,6 +486,7 @@ def _finish_recvfrom(cls, trans, key, ov, *, empty_result):
                 raise
 
     def recv(self, conn, nbytes, flags=0):
+        import _overlapped  # XXX: RUSTPYTHON
         self._register_with_iocp(conn)
         ov = _overlapped.Overlapped(NULL)
         try:
@@ -493,6 +500,7 @@ def recv(self, conn, nbytes, flags=0):
         return self._register(ov, conn, self.finish_socket_func)
 
     def recv_into(self, conn, buf, flags=0):
+        import _overlapped  # XXX: RUSTPYTHON
         self._register_with_iocp(conn)
         ov = _overlapped.Overlapped(NULL)
         try:
@@ -506,6 +514,7 @@ def recv_into(self, conn, buf, flags=0):
         return self._register(ov, conn, self.finish_socket_func)
 
     def recvfrom(self, conn, nbytes, flags=0):
+        import _overlapped  # XXX: RUSTPYTHON
         self._register_with_iocp(conn)
         ov = _overlapped.Overlapped(NULL)
         try:
@@ -517,6 +526,7 @@ def recvfrom(self, conn, nbytes, flags=0):
                                                 empty_result=b''))
 
     def recvfrom_into(self, conn, buf, flags=0):
+        import _overlapped  # XXX: RUSTPYTHON
         self._register_with_iocp(conn)
         ov = _overlapped.Overlapped(NULL)
         try:
@@ -528,6 +538,7 @@ def recvfrom_into(self, conn, buf, flags=0):
                                                 empty_result=0))
 
     def sendto(self, conn, buf, flags=0, addr=None):
+        import _overlapped  # XXX: RUSTPYTHON
         self._register_with_iocp(conn)
         ov = _overlapped.Overlapped(NULL)
 
@@ -536,6 +547,7 @@ def sendto(self, conn, buf, flags=0, addr=None):
         return self._register(ov, conn, self.finish_socket_func)
 
     def send(self, conn, buf, flags=0):
+        import _overlapped  # XXX: RUSTPYTHON
         self._register_with_iocp(conn)
         ov = _overlapped.Overlapped(NULL)
         if isinstance(conn, socket.socket):
@@ -546,6 +558,7 @@ def send(self, conn, buf, flags=0):
         return self._register(ov, conn, self.finish_socket_func)
 
     def accept(self, listener):
+        import _overlapped  # XXX: RUSTPYTHON
         self._register_with_iocp(listener)
         conn = self._get_accept_socket(listener.family)
         ov = _overlapped.Overlapped(NULL)
@@ -574,6 +587,7 @@ async def accept_coro(future, conn):
         return future
 
     def connect(self, conn, address):
+        import _overlapped  # XXX: RUSTPYTHON
         if conn.type == socket.SOCK_DGRAM:
             # WSAConnect will complete immediately for UDP sockets so we don't
             # need to register any IOCP operation
@@ -605,6 +619,7 @@ def finish_connect(trans, key, ov):
         return self._register(ov, conn, finish_connect)
 
     def sendfile(self, sock, file, offset, count):
+        import _overlapped  # XXX: RUSTPYTHON
         self._register_with_iocp(sock)
         ov = _overlapped.Overlapped(NULL)
         offset_low = offset & 0xffff_ffff
@@ -617,6 +632,7 @@ def sendfile(self, sock, file, offset, count):
         return self._register(ov, sock, self.finish_socket_func)
 
     def accept_pipe(self, pipe):
+        import _overlapped  # XXX: RUSTPYTHON
         self._register_with_iocp(pipe)
         ov = _overlapped.Overlapped(NULL)
         connected = ov.ConnectNamedPipe(pipe.fileno())
@@ -634,6 +650,7 @@ def finish_accept_pipe(trans, key, ov):
         return self._register(ov, pipe, finish_accept_pipe)
 
     async def connect_pipe(self, address):
+        import _overlapped  # XXX: RUSTPYTHON
         delay = CONNECT_PIPE_INIT_DELAY
         while True:
             # Unfortunately there is no way to do an overlapped connect to
@@ -668,6 +685,7 @@ def _wait_cancel(self, event, done_callback):
         return fut
 
     def _wait_for_handle(self, handle, timeout, _is_cancel):
+        import _overlapped  # XXX: RUSTPYTHON
         self._check_closed()
 
         if timeout is None:
@@ -702,6 +720,7 @@ def finish_wait_for_handle(trans, key, ov):
         return f
 
     def _register_with_iocp(self, obj):
+        import _overlapped  # XXX: RUSTPYTHON
         # To get notifications of finished ops on this objects sent to the
         # completion port, were must register the handle.
         if obj not in self._registered:
@@ -759,6 +778,7 @@ def _get_accept_socket(self, family):
         return s
 
     def _poll(self, timeout=None):
+        import _overlapped  # XXX: RUSTPYTHON
         if timeout is None:
             ms = INFINITE
         elif timeout < 0:
