Check for directory instead of config file

ikanobori · ikanobori · commit a119f6173a8d · 2009-12-30T16:39:20.000+01:00
diff --git a/bpython/wizard.py b/bpython/wizard.py
@@ -53,11 +53,13 @@
                                   'answers': positive_negative}
 
 filename = os.path.expanduser('~/.bpython/config')
+dirname  = os.path.expanduser('~/.bpython/')
 
 def is_first_run():
-    return not os.path.isfile(filename)
+    return not os.path.isdir(dirname)
 
 def create_empty_file():
+    os.mkdir(dirname)
     f = open(filename, 'w')
     f.write('')
     f.close()
@@ -76,14 +78,18 @@ def run_wizard():
 answer no to the following question.
 """
 
+    # Always create an empty file to prevent the wizard from running the next
+    # time
+    create_empty_file()
+
     answer = raw_input('Do you want to run the wizard: ')
 
-    if answer.lower() in negative_answers:
-        create_empty_file()
-    else:
+    # If the answer is positive, start wizard
+    if answer.lower() in positive_answers:
         config = ConfigParser()
 
         for section_name, section in questions.iteritems():
+            # Create sections on the fly
             config.add_section(section_name)
 
             print
@@ -102,6 +108,7 @@ def run_wizard():
                         print
                         print "I couldn't understand the answer you provided, please try again"
 
+        # Write the config file
         config.write(open(filename, 'w'))
 
 def main():
