diff options
Diffstat (limited to 'pydora/configure.py')
-rw-r--r-- | pydora/configure.py | 9 |
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) |