85 lines
1.8 KiB
Python
85 lines
1.8 KiB
Python
|
"""
|
||
|
|
Support for using stackless python. Broken and riddled with print statements
|
||
|
|
at the moment. Please fix it!
|
||
|
|
"""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import types
|
||
|
|
|
||
|
|
import stackless
|
||
|
|
|
||
|
|
caller = None
|
||
|
|
coro_args = {}
|
||
|
|
tasklet_to_greenlet = {}
|
||
|
|
|
||
|
|
|
||
|
|
def getcurrent():
|
||
|
|
return tasklet_to_greenlet[stackless.getcurrent()]
|
||
|
|
|
||
|
|
|
||
|
|
class FirstSwitch:
|
||
|
|
def __init__(self, gr):
|
||
|
|
self.gr = gr
|
||
|
|
|
||
|
|
def __call__(self, *args, **kw):
|
||
|
|
# print("first call", args, kw)
|
||
|
|
gr = self.gr
|
||
|
|
del gr.switch
|
||
|
|
run, gr.run = gr.run, None
|
||
|
|
t = stackless.tasklet(run)
|
||
|
|
gr.t = t
|
||
|
|
tasklet_to_greenlet[t] = gr
|
||
|
|
t.setup(*args, **kw)
|
||
|
|
t.run()
|
||
|
|
|
||
|
|
|
||
|
|
class greenlet:
|
||
|
|
def __init__(self, run=None, parent=None):
|
||
|
|
self.dead = False
|
||
|
|
if parent is None:
|
||
|
|
parent = getcurrent()
|
||
|
|
|
||
|
|
self.parent = parent
|
||
|
|
if run is not None:
|
||
|
|
self.run = run
|
||
|
|
|
||
|
|
self.switch = FirstSwitch(self)
|
||
|
|
|
||
|
|
def switch(self, *args):
|
||
|
|
# print("switch", args)
|
||
|
|
global caller
|
||
|
|
caller = stackless.getcurrent()
|
||
|
|
coro_args[self] = args
|
||
|
|
self.t.insert()
|
||
|
|
stackless.schedule()
|
||
|
|
if caller is not self.t:
|
||
|
|
caller.remove()
|
||
|
|
rval = coro_args[self]
|
||
|
|
return rval
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def __bool__(self):
|
||
|
|
return self.run is None and not self.dead
|
||
|
|
|
||
|
|
|
||
|
|
class GreenletExit(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def emulate():
|
||
|
|
module = types.ModuleType('greenlet')
|
||
|
|
sys.modules['greenlet'] = module
|
||
|
|
module.greenlet = greenlet
|
||
|
|
module.getcurrent = getcurrent
|
||
|
|
module.GreenletExit = GreenletExit
|
||
|
|
|
||
|
|
caller = stackless.getcurrent()
|
||
|
|
tasklet_to_greenlet[caller] = None
|
||
|
|
main_coro = greenlet()
|
||
|
|
tasklet_to_greenlet[caller] = main_coro
|
||
|
|
main_coro.t = caller
|
||
|
|
del main_coro.switch # It's already running
|
||
|
|
coro_args[main_coro] = None
|