@@ -189,6 +189,14 @@ def prompt_string(self, prompt, default):
189
189
: param default: The default value to use if the user does not provide input.
190
190
:return:
191
191
"""
192
+ prompt = prompt .strip ()
193
+ if prompt .endswith ('.' ):
194
+ prompt = prompt [:- 1 ]
195
+ prompt = prompt .strip ()
196
+ BLUE = "\033 [94m"
197
+ RESET = "\033 [0m"
198
+ if self .supports_ansi ():
199
+ prompt = f"{ BLUE } { prompt } { RESET } "
192
200
inp = input (f"{ prompt } ({ default } ): " )
193
201
result = inp .strip () or default
194
202
return result
@@ -201,6 +209,14 @@ def prompt_boolean(self, prompt, default = None):
201
209
:param default: The default value to return if the user does not provide input.
202
210
:return: bool: True if the user answers yes, False otherwise.
203
211
"""
212
+ prompt = prompt .strip ()
213
+ if prompt .endswith ('.' ):
214
+ prompt = prompt [:- 1 ]
215
+ prompt = prompt .strip ()
216
+ BLUE = "\033 [94m"
217
+ RESET = "\033 [0m"
218
+ if self .supports_ansi ():
219
+ prompt = f"{ BLUE } { prompt } { RESET } "
204
220
while True :
205
221
answer = input (f"{ prompt } ({ 'y/n' if default is None else 'yes' if default else 'no' } ): " ).strip ().lower ()
206
222
if answer in ('y' , 'yes' ):
@@ -338,17 +354,10 @@ def run_cmd(self, cmd, cwd=None):
338
354
"""
339
355
BLUE = "\033 [94m"
340
356
RESET = "\033 [0m"
341
- def supports_ansi ():
342
- if os .name == "posix" :
343
- return True
344
- if os .name == "nt" :
345
- return "WT_SESSION" in os .environ or sys .stdout .isatty ()
346
- return False
347
-
348
357
if cwd is None :
349
358
cwd = os .getcwd ()
350
- color = BLUE if supports_ansi () else ""
351
- reset = RESET if supports_ansi () else ""
359
+ color = BLUE if self . supports_ansi () else ""
360
+ reset = RESET if self . supports_ansi () else ""
352
361
if isinstance (cmd , list ):
353
362
print (f"{ color } { cwd } > { ' ' .join (cmd )} { reset } " )
354
363
else :
0 commit comments