6 from eventlet.support import greenlets as greenlet, six
7 from eventlet.hubs.hub import BaseHub, READ, WRITE
10 class event_wrapper(object):
12 def __init__(self, impl=None, seconds=None):
14 self.seconds = seconds
17 if self.impl is not None:
18 return repr(self.impl)
20 return object.__repr__(self)
23 if self.impl is not None:
26 return object.__str__(self)
29 if self.impl is not None:
35 return bool(self.impl and self.impl.pending())
40 SYSTEM_EXCEPTIONS = (KeyboardInterrupt, SystemExit)
43 super(Hub, self).__init__()
46 self.signal_exc_info = None
49 lambda signalnum, frame: self.greenlet.parent.throw(KeyboardInterrupt))
50 self.events_to_add = []
55 for e in self.events_to_add:
56 if e is not None and e.impl is not None and e.seconds is not None:
59 self.events_to_add = []
62 if getattr(event, '__event_exc', None) is not None:
63 # only have to do this because of bug in event.loop
64 t = getattr(event, '__event_exc')
65 setattr(event, '__event_exc', None)
66 assert getattr(event, '__event_exc') is None
67 six.reraise(t[0], t[1], t[2])
76 except greenlet.GreenletExit:
78 except self.SYSTEM_EXCEPTIONS:
81 if self.signal_exc_info is not None:
82 self.schedule_call_global(
83 0, greenlet.getcurrent().parent.throw, *self.signal_exc_info)
84 self.signal_exc_info = None
86 self.squelch_timer_exception(None, sys.exc_info())
88 def abort(self, wait=True):
89 self.schedule_call_global(0, self.greenlet.throw, greenlet.GreenletExit)
91 assert self.greenlet is not greenlet.getcurrent(
92 ), "Can't abort with wait from inside the hub's greenlet."
95 def _getrunning(self):
96 return bool(self.greenlet)
98 def _setrunning(self, value):
99 pass # exists for compatibility with BaseHub
100 running = property(_getrunning, _setrunning)
102 def add(self, evtype, fileno, real_cb, real_tb, mac):
103 # this is stupid: pyevent won't call a callback unless it's a function,
104 # so we have to force it to be one here
105 if isinstance(real_cb, types.BuiltinMethodType):
112 evt = event.read(fileno, cb, fileno)
113 elif evtype is WRITE:
114 evt = event.write(fileno, cb, fileno)
116 return super(Hub, self).add(evtype, fileno, evt, real_tb, mac)
118 def signal(self, signalnum, handler):
121 handler(signalnum, None)
123 self.signal_exc_info = sys.exc_info()
125 return event_wrapper(event.signal(signalnum, wrapper))
127 def remove(self, listener):
128 super(Hub, self).remove(listener)
131 def remove_descriptor(self, fileno):
132 for lcontainer in six.itervalues(self.listeners):
133 listener = lcontainer.pop(fileno, None)
137 except self.SYSTEM_EXCEPTIONS:
140 traceback.print_exc()
142 def schedule_call_local(self, seconds, cb, *args, **kwargs):
143 current = greenlet.getcurrent()
144 if current is self.greenlet:
145 return self.schedule_call_global(seconds, cb, *args, **kwargs)
146 event_impl = event.event(_scheduled_call_local, (cb, args, kwargs, current))
147 wrapper = event_wrapper(event_impl, seconds=seconds)
148 self.events_to_add.append(wrapper)
151 schedule_call = schedule_call_local
153 def schedule_call_global(self, seconds, cb, *args, **kwargs):
154 event_impl = event.event(_scheduled_call, (cb, args, kwargs))
155 wrapper = event_wrapper(event_impl, seconds=seconds)
156 self.events_to_add.append(wrapper)
159 def _version_info(self):
160 baseversion = event.__version__
164 def _scheduled_call(event_impl, handle, evtype, arg):
165 cb, args, kwargs = arg
172 def _scheduled_call_local(event_impl, handle, evtype, arg):
173 cb, args, kwargs, caller_greenlet = arg
175 if not caller_greenlet.dead: