This repository was archived by the owner on Aug 27, 2021. It is now read-only.
forked from bpython/bpython
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathwizard.py
More file actions
119 lines (95 loc) · 4.15 KB
/
wizard.py
File metadata and controls
119 lines (95 loc) · 4.15 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#!/usr/bin/env python
#
# The MIT License
#
# Copyright (c) 2008 Bob Farrell
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#
import os
import sys
from ConfigParser import ConfigParser
# questions asked by the wizard (keys are the config values, followed by text
# and answers and defaults
positive_answers = ('Y', 'y', 'yes', 'ja',)
negative_answers = ('N', 'n', 'no', 'nein',)
# Create default answer
positive_negative = {positive_answers: 'True',
negative_answers: 'False'}
# Create dict for storing everythin
questions = {}
# Create sections
questions['general'] = {}
questions['keyboard'] = {}
questions['general']['auto_display_list'] = {'question': 'Do you want to show the autocomplete list as you type (y/n):',
'answers': positive_negative}
questions['general']['arg_spec'] = {'question': 'Do you want to show the argspec (y/n):',
'answers': positive_negative}
questions['general']['syntax'] = {'question': 'Do you want syntax highlighting as you type (y/n):',
'answers': positive_negative}
filename = os.path.expanduser('~/.bpython/config')
dirname = os.path.expanduser('~/.bpython/')
def is_first_run():
return not os.path.isdir(dirname)
def create_empty_file():
os.mkdir(dirname)
f = open(filename, 'w')
f.write('')
f.close()
def run_wizard():
"""Run an interactive wizard asking a few questions about the users'
enviroment and write the answers to the configuration file."""
print """Hi there. It seems this is the first time you run bpython. I can
tell because you do not have a configuration file yet. There are a few
options I am going to give you.
I am ready to run the wizard for you now. If you do not want to run the
wizard and just have me create an empty configuration file for you you can
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 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
for config_value, q in section.iteritems():
while 1:
print q['question'],
answer = raw_input()
if answer in positive_answers:
config.set(section_name, config_value, q['answers'][positive_answers])
break
elif answer in negative_answers:
config.set(section_name, config_value, q['answers'][negative_answers])
break
else:
print
print "I couldn't understand the answer you provided, please try again"
# Write the config file
config.write(open(filename, 'w'))
def main():
if is_first_run():
run_wizard()
if __name__ == '__main__':
main()