80633ab224
First step to Python 3 compatibility "2to3 -w -f except ." See [1] [1] http://docs.python.org/2/library/2to3.html#fixers
85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
os_orig = __import__("os")
|
|
import errno
|
|
socket = __import__("socket")
|
|
|
|
from eventlet import greenio
|
|
from eventlet.support import get_errno
|
|
from eventlet import greenthread
|
|
from eventlet import hubs
|
|
from eventlet.patcher import slurp_properties
|
|
|
|
__all__ = os_orig.__all__
|
|
__patched__ = ['fdopen', 'read', 'write', 'wait', 'waitpid']
|
|
|
|
slurp_properties(os_orig, globals(),
|
|
ignore=__patched__, srckeys=dir(os_orig))
|
|
|
|
def fdopen(fd, *args, **kw):
|
|
"""fdopen(fd [, mode='r' [, bufsize]]) -> file_object
|
|
|
|
Return an open file object connected to a file descriptor."""
|
|
if not isinstance(fd, int):
|
|
raise TypeError('fd should be int, not %r' % fd)
|
|
try:
|
|
return greenio.GreenPipe(fd, *args, **kw)
|
|
except IOError as e:
|
|
raise OSError(*e.args)
|
|
|
|
__original_read__ = os_orig.read
|
|
def read(fd, n):
|
|
"""read(fd, buffersize) -> string
|
|
|
|
Read a file descriptor."""
|
|
while True:
|
|
try:
|
|
return __original_read__(fd, n)
|
|
except (OSError, IOError) as e:
|
|
if get_errno(e) != errno.EAGAIN:
|
|
raise
|
|
except socket.error as e:
|
|
if get_errno(e) == errno.EPIPE:
|
|
return ''
|
|
raise
|
|
hubs.trampoline(fd, read=True)
|
|
|
|
__original_write__ = os_orig.write
|
|
def write(fd, st):
|
|
"""write(fd, string) -> byteswritten
|
|
|
|
Write a string to a file descriptor.
|
|
"""
|
|
while True:
|
|
try:
|
|
return __original_write__(fd, st)
|
|
except (OSError, IOError) as e:
|
|
if get_errno(e) != errno.EAGAIN:
|
|
raise
|
|
except socket.error as e:
|
|
if get_errno(e) != errno.EPIPE:
|
|
raise
|
|
hubs.trampoline(fd, write=True)
|
|
|
|
def wait():
|
|
"""wait() -> (pid, status)
|
|
|
|
Wait for completion of a child process."""
|
|
return waitpid(0,0)
|
|
|
|
__original_waitpid__ = os_orig.waitpid
|
|
def waitpid(pid, options):
|
|
"""waitpid(...)
|
|
waitpid(pid, options) -> (pid, status)
|
|
|
|
Wait for completion of a given child process."""
|
|
if options & os_orig.WNOHANG != 0:
|
|
return __original_waitpid__(pid, options)
|
|
else:
|
|
new_options = options | os_orig.WNOHANG
|
|
while True:
|
|
rpid, status = __original_waitpid__(pid, new_options)
|
|
if rpid and status >= 0:
|
|
return rpid, status
|
|
greenthread.sleep(0.01)
|
|
|
|
# TODO: open
|