Add "prompt_yes_no" to utils

This commit is contained in:
I-Al-Istannen 2020-04-20 14:29:28 +02:00
parent a3d6dc7873
commit 5c2ff14839

View File

@ -14,24 +14,56 @@ def move(path: PurePath, from_folders: Tuple[str], to_folders: Tuple[str]) -> Op
return PurePath(*to_folders, *path.parts[l:]) return PurePath(*to_folders, *path.parts[l:])
return None return None
def rename(path: PurePath, to_name: str) -> PurePath: def rename(path: PurePath, to_name: str) -> PurePath:
return PurePath(*path.parts[:-1], to_name) return PurePath(*path.parts[:-1], to_name)
def stream_to_path(response: requests.Response, to_path: Path, chunk_size: int = 1024 ** 2) -> None: def stream_to_path(response: requests.Response, to_path: Path, chunk_size: int = 1024 ** 2) -> None:
with open(to_path, 'wb') as fd: with open(to_path, 'wb') as fd:
for chunk in response.iter_content(chunk_size=chunk_size): for chunk in response.iter_content(chunk_size=chunk_size):
fd.write(chunk) fd.write(chunk)
def prompt_yes_no(question: str, default: Optional[bool] = None) -> bool:
"""Prompts the user and returns their choice."""
if default is True:
prompt = "[Y/n]"
elif default is False:
prompt = "[y/N]"
else:
prompt = "[y/n]"
text = f"{question} {prompt} "
WRONG_REPLY = "Please reply with 'yes'/'y' or 'no'/'n'."
while True:
response = input(text).strip().lower()
if response in {"yes", "ye", "y"}:
return True
elif response in {"no", "n"}:
return False
elif response == "":
if default is None:
print(WRONG_REPLY)
else:
return default
else:
print(WRONG_REPLY)
class PrettyLogger: class PrettyLogger:
def __init__(self, logger: logging.Logger) -> None: def __init__(self, logger: logging.Logger) -> None:
self.logger = logger self.logger = logger
def modified_file(self, file_name: Path) -> None: def modified_file(self, file_name: Path) -> None:
self.logger.info(f"{Fore.MAGENTA}{Style.BRIGHT}Modified {file_name}.{Style.RESET_ALL}") self.logger.info(
f"{Fore.MAGENTA}{Style.BRIGHT}Modified {file_name}.{Style.RESET_ALL}")
def new_file(self, file_name: Path) -> None: def new_file(self, file_name: Path) -> None:
self.logger.info(f"{Fore.GREEN}{Style.BRIGHT}Created {file_name}.{Style.RESET_ALL}") self.logger.info(
f"{Fore.GREEN}{Style.BRIGHT}Created {file_name}.{Style.RESET_ALL}")
def ignored_file(self, file_name: Path) -> None: def ignored_file(self, file_name: Path) -> None:
self.logger.info(f"{Style.DIM}Ignored {file_name}.{Style.RESET_ALL}") self.logger.info(f"{Style.DIM}Ignored {file_name}.{Style.RESET_ALL}")