Hook up import completion.

marienz · marienz · commit 494a5788e3f5 · 2010-02-10T15:30:24.000+01:00
diff --git a/bpython/urwid.py b/bpython/urwid.py
@@ -45,6 +45,7 @@
 
 from bpython import args as bpargs, repl
 from bpython.formatter import theme_map
+from bpython.importcompletion import find_coroutine
 
 import urwid
 
@@ -479,6 +480,14 @@ def start(main_loop, user_data):
                 repl.write('\n')
             myrepl.start()
 
+            # This bypasses main_loop.set_alarm_in because we must *not*
+            # hit the draw_screen call (it's unnecessary and slow).
+            def run_find_coroutine():
+                if find_coroutine():
+                    main_loop.event_loop.alarm(0, run_find_coroutine)
+
+            run_find_coroutine()
+
         loop.set_alarm_in(0, start)
 
         loop.run()
