Try to run wizard on start of bpython.cli

ikanobori · ikanobori · commit 030c8345ffab · 2009-12-30T15:37:11.000+01:00
diff --git a/bpython/cli.py b/bpython/cli.py
@@ -59,6 +59,9 @@
 # This for keys
 from bpython.keys import key_dispatch
 
+# This for the wizard
+from bpython import wizard
+
 from bpython.pager import page
 from bpython.repl import Interpreter, Repl
 import bpython.args
@@ -1581,6 +1584,8 @@ def main(args=None, locals_=None, banner=None):
 
     config, options, exec_args = bpython.args.parse(args)
 
+    wizard.main()
+
     # Save stdin, stdout and stderr for later restoration
     orig_stdin = sys.stdin
     orig_stdout = sys.stdout
diff --git a/bpython/wizard.py b/bpython/wizard.py
@@ -91,6 +91,6 @@ def run_wizard():
 
         print answers
 
-if __name__ == '__main__':
+def main():
     if is_first_run():
         run_wizard()
