test greenlet link

greenlet-link.py

import mock
import gevent
import random
def do_job(name='noname', sleep=10):
    print "begin...%s" % sleep
    for i in range(sleep):
        gevent.sleep(i)
        print "sleep...%s" % i
        if random.randint(0, 3) == 0:
            print "oops"
            print 10 / 0
    print "done"
    return "oops"


glet = gevent.spawn(do_job, sleep=1)

@glet.link
def callback(glet, *args, **kw):
    """when glet finished, called, even if any exception"""
    print "callback called"
    print glet, args, kw

# or
# glet.link(callback)


@glet.link_value
def callback_with_value(glet, *args, **kw):
    """when glet finished but no exception, called"""
    print "callback_with_value called"
    print glet, args, kw


gevent.sleep(5)
print "done"