Group: allow adding greenlets without 'rawlink' method (like MAIN)

5fef0b6b · Denis Bilenko · 247ebba3 · 5fef0b6b
Commit 5fef0b6b authored Jul 31, 2011 by Denis Bilenko
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

gevent/pool.py gevent/pool.py +6 -1

No files found.
--- a/gevent/pool.py
+++ b/gevent/pool.py
@@ -56,7 +56,12 @@ class Group(object):
        return iter(self.greenlets)

    def add(self, greenlet):
-        greenlet.rawlink(self.discard)
+        try:
+            rawlink = greenlet.rawlink
+        except AttributeError:
+            pass  # non-Greenlet greenlet, like MAIN
+        else:
+            rawlink(self.discard)
        self.greenlets.add(greenlet)
        self._empty_event.clear()