aboutsummaryrefslogtreecommitdiff
path: root/pydora/configure.py
diff options
context:
space:
mode:
Diffstat (limited to 'pydora/configure.py')
-rw-r--r--pydora/configure.py9
1 files changed, 5 insertions, 4 deletions
diff --git a/pydora/configure.py b/pydora/configure.py
index 01de8df..7d3c754 100644
--- a/pydora/configure.py
+++ b/pydora/configure.py
@@ -108,6 +108,7 @@ class Configurator(object):
108 self.builder = PydoraConfigFileBuilder() 108 self.builder = PydoraConfigFileBuilder()
109 109
110 self.cfg = ConfigParser() 110 self.cfg = ConfigParser()
111 self.screen = Screen()
111 112
112 if self.builder.file_exists: 113 if self.builder.file_exists:
113 self.read_config() 114 self.read_config()
@@ -116,11 +117,11 @@ class Configurator(object):
116 self.cfg.add_section("api") 117 self.cfg.add_section("api")
117 118
118 def fail(self, message): 119 def fail(self, message):
119 print(Screen.print_error(message)) 120 print(self.screen.print_error(message))
120 sys.exit(1) 121 sys.exit(1)
121 122
122 def finished(self, message): 123 def finished(self, message):
123 Screen.print_success(message) 124 self.screen.print_success(message)
124 sys.exit(0) 125 sys.exit(0)
125 126
126 def print_message(self, message): 127 def print_message(self, message):
@@ -133,10 +134,10 @@ class Configurator(object):
133 self.fail("Error loading config file. Unable to continue.") 134 self.fail("Error loading config file. Unable to continue.")
134 135
135 def get_value(self, section, key, prompt): 136 def get_value(self, section, key, prompt):
136 self.cfg.set(section, key, Screen.get_string(prompt)) 137 self.cfg.set(section, key, self.screen.get_string(prompt))
137 138
138 def get_password(self, section, key, prompt): 139 def get_password(self, section, key, prompt):
139 self.cfg.set(section, key, Screen.get_password(prompt)) 140 self.cfg.set(section, key, self.screen.get_password(prompt))
140 141
141 def set_static_value(self, section, key, value): 142 def set_static_value(self, section, key, value):
142 self.cfg.set(section, key, value) 143 self.cfg.set(section, key, value)