diff --git a/src/manhole/__init__.py b/src/manhole/__init__.py index afdc4ad..a4b8c00 100644 --- a/src/manhole/__init__.py +++ b/src/manhole/__init__.py @@ -40,19 +40,13 @@ class BrokenPipeError(Exception): setinterval = sys.setcheckinterval getinterval = sys.getcheckinterval -try: +if "eventlet" in sys.modules: from eventlet.patcher import original as _original - +elif "gevent" in sys.modules: + from gevent.monkey import get_original as _get_original +else: def _get_original(mod, name): - return getattr(_original(mod), name) - -except ImportError: - try: - from gevent.monkey import get_original as _get_original - except ImportError: - - def _get_original(mod, name): - return getattr(__import__(mod), name) + return getattr(__import__(mod), name) _ORIGINAL_SOCKET = _get_original('socket', 'socket')