1 import _pyio as _original_pyio
3 import os as _original_os
4 import socket as _original_socket
6 BufferedRandom as _OriginalBufferedRandom,
7 BufferedReader as _OriginalBufferedReader,
8 BufferedWriter as _OriginalBufferedWriter,
10 TextIOWrapper as _OriginalTextIOWrapper,
11 IOBase as _OriginalIOBase,
13 from types import FunctionType
15 from eventlet.greenio.base import (
16 _operation_on_closed_file,
21 from eventlet.hubs import notify_close, notify_opened, IOClosed, trampoline
22 from eventlet.support import get_errno, six
24 __all__ = ['_fileobject', 'GreenPipe']
26 # TODO get rid of this, it only seems like the original _fileobject
27 _fileobject = _original_socket.SocketIO
29 # Large part of the following code is copied from the original
30 # eventlet.greenio module
33 class GreenFileIO(_OriginalIOBase):
34 def __init__(self, name, mode='r', closefd=True, opener=None):
35 if isinstance(name, int):
37 self._name = "<fd:%d>" % fileno
39 assert isinstance(name, six.string_types)
40 with open(name, mode) as fd:
42 fileno = _original_os.dup(fd.fileno())
56 if self._seekable is None:
58 _original_os.lseek(self._fileno, 0, _original_os.SEEK_CUR)
60 if get_errno(e) == errno.ESPIPE:
61 self._seekable = False
70 return 'r' in self._mode or '+' in self._mode
73 return 'w' in self._mode or '+' in self._mode
78 def read(self, buflen):
81 return _original_os.read(self._fileno, buflen)
83 if get_errno(e) not in SOCKET_BLOCKING:
84 raise IOError(*e.args)
85 self._trampoline(self, read=True)
87 def readinto(self, b):
89 data = self.read(up_to)
90 bytes_read = len(data)
96 return _original_os.isatty(self.fileno())
98 raise IOError(*e.args)
100 def _trampoline(self, fd, read=False, write=False, timeout=None, timeout_exc=None):
102 # Don't trampoline if we're already closed.
105 return trampoline(fd, read=read, write=write, timeout=timeout,
106 timeout_exc=timeout_exc,
107 mark_as_closed=self._mark_as_closed)
109 # Our fileno has been obsoleted. Defang ourselves to
110 # prevent spurious closes.
111 self._mark_as_closed()
114 def _mark_as_closed(self):
115 """ Mark this socket as being closed """
118 def write(self, data):
121 return _original_os.write(self._fileno, data)
123 if get_errno(e) not in SOCKET_BLOCKING:
124 raise IOError(*e.args)
126 trampoline(self, write=True)
131 _original_os.close(self._fileno)
132 notify_close(self._fileno)
134 'fileno', 'flush', 'isatty', 'next', 'read', 'readinto',
135 'readline', 'readlines', 'seek', 'tell', 'truncate',
136 'write', 'xreadlines', '__iter__', '__next__', 'writelines']:
137 setattr(self, method, _operation_on_closed_file)
139 def truncate(self, size=-1):
143 rv = _original_os.ftruncate(self._fileno, size)
145 raise IOError(*e.args)
147 self.seek(size) # move position&clear buffer
150 def seek(self, offset, whence=_original_os.SEEK_SET):
152 return _original_os.lseek(self._fileno, offset, whence)
154 raise IOError(*e.args)
159 def __exit__(self, *args):
163 _open_environment = dict(globals())
164 _open_environment.update(dict(
165 BufferedRandom=_OriginalBufferedRandom,
166 BufferedWriter=_OriginalBufferedWriter,
167 BufferedReader=_OriginalBufferedReader,
168 TextIOWrapper=_OriginalTextIOWrapper,
173 _open = FunctionType(
174 six.get_function_code(_original_pyio.open),
179 def GreenPipe(name, mode="r", buffering=-1, encoding=None, errors=None,
180 newline=None, closefd=True, opener=None):
182 fileno = name.fileno()
183 except AttributeError:
186 fileno = _original_os.dup(fileno)
190 return _open(name, mode, buffering, encoding, errors, newline, closefd, opener)
192 GreenPipe.__doc__ = greenpipe_doc