Import main from bpython.{cli,gtk_} when the modules are executed directly.

Trundle · Trundle · commit c6ce27021170 · 2010-02-06T23:49:51.000+01:00
Otherwise, "import bpython.cli" will be another module and you can't access
e.g. "bpython.cli.stdscr" (for testing purposes only, of course).
diff --git a/bpython/cli.py b/bpython/cli.py
@@ -1610,6 +1610,7 @@ def main(args=None, locals_=None, banner=None):
 
 
 if __name__ == '__main__':
+    from bpython.cli import main
     main()
 
 # vim: sw=4 ts=4 sts=4 ai et
diff --git a/bpython/gtk_.py b/bpython/gtk_.py
@@ -695,4 +695,5 @@ def main(args=None):
 
 
 if __name__ == '__main__':
+    from bpython.gtk_ import main
     main()
