uasyncio: Implement run_until_complete().

pfalcon · pfalcon · commit 67b713535b2e · 2014-10-18T04:57:27.000+03:00
diff --git a/uasyncio/uasyncio.py b/uasyncio/uasyncio.py
@@ -77,6 +77,8 @@ def run_forever(self):
                             self.remove_reader(arg.fileno())
                         elif isinstance(ret, IOWriteDone):
                             self.remove_writer(arg.fileno())
+                        elif isinstance(ret, StopLoop):
+                            return arg
                     elif isinstance(ret, type_gen):
                         self.call_soon(ret)
                     elif ret is None:
@@ -90,16 +92,11 @@ def run_forever(self):
                 self.call_later(delay, cb, *args)
 
     def run_until_complete(self, coro):
-        val = None
-        while True:
-            try:
-                ret = coro.send(val)
-            except StopIteration as e:
-                print(e)
-                break
-            print("ret:", ret)
-            if isinstance(ret, SysCall):
-                ret.handle()
+        def _run_and_stop():
+            yield from coro
+            yield StopLoop(0)
+        self.call_soon(_run_and_stop())
+        self.run_forever()
 
     def close(self):
         pass
@@ -152,6 +149,9 @@ def handle(self):
 class Sleep(SysCall):
     pass
 
+class StopLoop(SysCall):
+    pass
+
 class IORead(SysCall):
     pass
 
