1
|
|
|
"""Event loop implementation that uses the `asyncio` standard module. |
2
|
|
|
|
3
|
|
|
The `asyncio` module was added to python standard library on 3.4, and it |
4
|
|
|
provides a pure python implementation of an event loop library. It is used |
5
|
|
|
as a fallback in case pyuv is not available(on python implementations other |
6
|
|
|
than CPython). |
7
|
|
|
|
8
|
|
|
Earlier python versions are supported through the `trollius` package, which |
9
|
|
|
is a backport of `asyncio` that works on Python 2.6+. |
10
|
|
|
""" |
11
|
5 |
|
from __future__ import absolute_import |
12
|
|
|
|
13
|
5 |
|
import logging |
14
|
5 |
|
import os |
15
|
5 |
|
import sys |
16
|
|
|
from collections import deque |
17
|
5 |
|
import threading |
18
|
|
|
|
19
|
5 |
|
try: |
20
|
3 |
|
# For python 3.4+, use the standard library module |
21
|
|
|
import asyncio |
22
|
3 |
|
except (ImportError, SyntaxError): |
23
|
|
|
# Fallback to trollius |
24
|
5 |
|
import trollius as asyncio |
|
|
|
|
25
|
|
|
|
26
|
|
|
from .base import BaseEventLoop |
27
|
5 |
|
|
28
|
5 |
|
logger = logging.getLogger(__name__) |
29
|
|
|
debug, info, warn = (logger.debug, logger.info, logger.warning,) |
30
|
|
|
|
31
|
|
|
loop_cls = asyncio.SelectorEventLoop |
32
|
|
|
if os.name == 'nt': |
33
|
|
|
# On windows use ProactorEventLoop which support pipes and is backed by the |
34
|
5 |
|
# more powerful IOCP facility |
35
|
|
|
# NOTE: we override in the stdio case, because it doesn't work. |
36
|
|
|
loop_cls = asyncio.ProactorEventLoop |
37
|
|
|
|
38
|
|
|
import msvcrt |
|
|
|
|
39
|
5 |
|
from ctypes import windll, byref, wintypes, GetLastError, WinError, POINTER |
|
|
|
|
40
|
|
|
from ctypes.wintypes import HANDLE, DWORD, BOOL |
41
|
5 |
|
|
42
|
5 |
|
LPDWORD = POINTER(DWORD) |
43
|
5 |
|
|
44
|
|
|
PIPE_NOWAIT = wintypes.DWORD(0x00000001) |
45
|
5 |
|
|
46
|
|
|
ERROR_NO_DATA = 232 |
47
|
|
|
|
48
|
|
|
def pipe_no_wait(pipefd): |
49
|
5 |
|
SetNamedPipeHandleState = windll.kernel32.SetNamedPipeHandleState |
50
|
|
|
SetNamedPipeHandleState.argtypes = [HANDLE, LPDWORD, LPDWORD, LPDWORD] |
51
|
|
|
SetNamedPipeHandleState.restype = BOOL |
52
|
|
|
|
53
|
|
|
h = msvcrt.get_osfhandle(pipefd) |
54
|
|
|
|
55
|
|
|
res = windll.kernel32.SetNamedPipeHandleState(h, byref(PIPE_NOWAIT), None, None) |
56
|
5 |
|
if res == 0: |
57
|
|
|
print(WinError()) |
58
|
4 |
|
return False |
59
|
|
|
return True |
60
|
5 |
|
|
61
|
|
|
class AsyncioEventLoop(BaseEventLoop, asyncio.Protocol, |
62
|
5 |
|
asyncio.SubprocessProtocol): |
63
|
4 |
|
|
64
|
5 |
|
"""`BaseEventLoop` subclass that uses `asyncio` as a backend.""" |
65
|
5 |
|
|
66
|
|
|
def connection_made(self, transport): |
67
|
|
|
"""Used to signal `asyncio.Protocol` of a successful connection.""" |
68
|
|
|
self._transport = transport |
|
|
|
|
69
|
5 |
|
self._raw_transport = transport |
|
|
|
|
70
|
|
|
if isinstance(transport, asyncio.SubprocessTransport): |
71
|
|
|
self._transport = transport.get_pipe_transport(0) |
|
|
|
|
72
|
|
|
|
73
|
5 |
|
def connection_lost(self, exc): |
74
|
5 |
|
"""Used to signal `asyncio.Protocol` of a lost connection.""" |
75
|
5 |
|
self._on_error(exc.args[0] if exc else 'EOF') |
76
|
5 |
|
|
77
|
|
|
def data_received(self, data): |
78
|
5 |
|
"""Used to signal `asyncio.Protocol` of incoming data.""" |
79
|
|
|
if self._on_data: |
80
|
|
|
self._on_data(data) |
81
|
|
|
return |
82
|
5 |
|
self._queued_data.append(data) |
83
|
|
|
|
84
|
|
|
def pipe_connection_lost(self, fd, exc): |
85
|
|
|
"""Used to signal `asyncio.SubprocessProtocol` of a lost connection.""" |
86
|
|
|
self._on_error(exc.args[0] if exc else 'EOF') |
87
|
|
|
|
88
|
|
|
def pipe_data_received(self, fd, data): |
89
|
5 |
|
"""Used to signal `asyncio.SubprocessProtocol` of incoming data.""" |
90
|
|
|
if fd == 2: # stderr fd number |
91
|
|
|
self._on_stderr(data) |
|
|
|
|
92
|
|
|
elif self._on_data: |
93
|
1 |
|
self._on_data(data) |
94
|
|
|
else: |
95
|
5 |
|
self._queued_data.append(data) |
96
|
5 |
|
|
97
|
5 |
|
def process_exited(self): |
98
|
|
|
"""Used to signal `asyncio.SubprocessProtocol` when the child exits.""" |
99
|
5 |
|
self._on_error('EOF') |
100
|
5 |
|
|
101
|
|
|
def _init(self): |
102
|
5 |
|
self._loop = loop_cls() |
|
|
|
|
103
|
5 |
|
self._queued_data = deque() |
|
|
|
|
104
|
|
|
self._fact = lambda: self |
|
|
|
|
105
|
5 |
|
self._raw_transport = None |
|
|
|
|
106
|
5 |
|
self._raw_stdout = False |
|
|
|
|
107
|
|
|
self._active = False |
|
|
|
|
108
|
5 |
|
|
109
|
|
|
def _connect_tcp(self, address, port): |
110
|
5 |
|
coroutine = self._loop.create_connection(self._fact, address, port) |
111
|
5 |
|
self._loop.run_until_complete(coroutine) |
112
|
|
|
|
113
|
5 |
|
def _connect_socket(self, path): |
114
|
5 |
|
if os.name == 'nt': |
115
|
|
|
coroutine = self._loop.create_pipe_connection(self._fact, path) |
116
|
5 |
|
else: |
117
|
5 |
|
coroutine = self._loop.create_unix_connection(self._fact, path) |
118
|
|
|
self._loop.run_until_complete(coroutine) |
119
|
|
|
|
120
|
|
|
def _on_stdin(self): |
121
|
|
|
while True: |
122
|
5 |
|
data = sys.stdin.buffer.read(1) |
123
|
5 |
|
if len(data) == 0: |
124
|
5 |
|
break |
125
|
|
|
self.data_received(data) |
126
|
5 |
|
|
127
|
5 |
|
def _stdin_reader(self): |
128
|
5 |
|
debug("started reader thread") |
129
|
|
|
while True: #self._active |
130
|
|
|
data = self._stdin.read(1) |
131
|
|
|
debug("reader thread read %d", len(data)) |
132
|
|
|
self._loop.call_soon_threadsafe(self.data_received, data) |
133
|
|
|
|
134
|
|
|
def _connect_stdio(self): |
135
|
|
|
try: |
136
|
|
|
coroutine = self._loop.connect_read_pipe(self._fact, sys.stdin) |
137
|
|
|
self._loop.run_until_complete(coroutine) |
138
|
|
|
debug("native stdin connection successful") |
139
|
|
|
except OSError: |
140
|
|
|
debug("native stdin connection failed, using reader thread") |
141
|
|
|
if os.name == "nt": |
142
|
|
|
pass |
143
|
|
|
#pipe_no_wait(sys.stdin.fileno()) |
144
|
|
|
else: |
145
|
|
|
import fcntl |
|
|
|
|
146
|
|
|
#orig_fl = fcntl.fcntl(sys.stdin.fileno(), fcntl.F_GETFL) |
147
|
|
|
#fcntl.fcntl(sys.stdin.fileno(), fcntl.F_SETFL, orig_fl | os.O_NONBLOCK) |
148
|
|
|
#coro = self._loop.run_in_executor(None, self._stdin_reader) |
149
|
|
|
#self._loop.create_task(coro) |
150
|
|
|
self._stdin = sys.stdin.buffer |
|
|
|
|
151
|
|
|
self._active = True |
|
|
|
|
152
|
|
|
threading.Thread(target=self._stdin_reader).start() |
153
|
|
|
|
154
|
|
|
|
155
|
|
|
try: |
156
|
|
|
coroutine = self._loop.connect_write_pipe(self._fact, sys.stdout) |
157
|
|
|
self._loop.run_until_complete(coroutine) |
158
|
|
|
debug("native stdout connection successful") |
159
|
|
|
except OSError: |
160
|
|
|
debug("native stdin connection failed, using reader thread") |
161
|
|
|
|
162
|
|
|
self._stdout = sys.stdout.buffer |
|
|
|
|
163
|
|
|
self._raw_stdout = True |
|
|
|
|
164
|
|
|
|
165
|
|
|
def _connect_child(self, argv): |
166
|
|
|
self._child_watcher = asyncio.get_child_watcher() |
|
|
|
|
167
|
|
|
self._child_watcher.attach_loop(self._loop) |
168
|
|
|
coroutine = self._loop.subprocess_exec(self._fact, *argv) |
169
|
|
|
self._loop.run_until_complete(coroutine) |
170
|
|
|
|
171
|
|
|
def _start_reading(self): |
172
|
|
|
pass |
173
|
|
|
|
174
|
|
|
def _send(self, data): |
175
|
|
|
debug("sent %s %s", repr(data), str(self._raw_stdout)) |
176
|
|
|
if self._raw_stdout: |
177
|
|
|
self._stdout.write(data) |
178
|
|
|
self._stdout.flush() |
179
|
|
|
else: |
180
|
|
|
self._transport.write(data) |
181
|
|
|
|
182
|
|
|
def _run(self): |
183
|
|
|
while self._queued_data: |
184
|
|
|
self._on_data(self._queued_data.popleft()) |
185
|
|
|
self._loop.run_forever() |
186
|
|
|
|
187
|
|
|
def _stop(self): |
188
|
|
|
self._loop.stop() |
189
|
|
|
|
190
|
|
|
def _close(self): |
191
|
|
|
if self._raw_transport is not None: |
192
|
|
|
self._raw_transport.close() |
193
|
|
|
# FIXME: this is racy and stuff |
194
|
|
|
self._active = False |
|
|
|
|
195
|
|
|
self._loop.close() |
196
|
|
|
|
197
|
|
|
def _threadsafe_call(self, fn): |
198
|
|
|
self._loop.call_soon_threadsafe(fn) |
199
|
|
|
|
200
|
|
|
def _setup_signals(self, signals): |
201
|
|
|
if os.name == 'nt': |
202
|
|
|
# add_signal_handler is not supported in win32 |
203
|
|
|
self._signals = [] |
|
|
|
|
204
|
|
|
return |
205
|
|
|
|
206
|
|
|
self._signals = list(signals) |
|
|
|
|
207
|
|
|
for signum in self._signals: |
208
|
|
|
self._loop.add_signal_handler(signum, self._on_signal, signum) |
209
|
|
|
|
210
|
|
|
def _teardown_signals(self): |
211
|
|
|
for signum in self._signals: |
212
|
|
|
self._loop.remove_signal_handler(signum) |
213
|
|
|
|
This can be caused by one of the following:
1. Missing Dependencies
This error could indicate a configuration issue of Pylint. Make sure that your libraries are available by adding the necessary commands.
2. Missing __init__.py files
This error could also result from missing
__init__.py
files in your module folders. Make sure that you place one file in each sub-folder.