mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
skshetry
b647475a64
* drop Python 3.8 Note that this PR only makes typing related changes. Other changes can be done once this gets merged, they are relatively few in numbers. * bump build python version * add typing_extensions to lint group
58 lines
1.3 KiB
Python
58 lines
1.3 KiB
Python
"""Manages user prompts."""
|
|
|
|
from collections.abc import Collection
|
|
from getpass import getpass
|
|
from typing import Optional
|
|
|
|
from dvc.log import logger
|
|
|
|
logger = logger.getChild(__name__)
|
|
|
|
|
|
def ask(prompt: str, limited_to: Optional[Collection[str]] = None):
|
|
from dvc.ui import Console
|
|
|
|
if not Console.isatty():
|
|
return None
|
|
|
|
while True:
|
|
try:
|
|
answer = input(prompt + " ").lower()
|
|
except EOFError:
|
|
return None
|
|
|
|
if not limited_to:
|
|
return answer
|
|
|
|
if answer in limited_to:
|
|
return answer
|
|
|
|
logger.info("Your response must be one of: %s. Please try again.", limited_to)
|
|
|
|
|
|
def confirm(statement: str) -> bool:
|
|
"""Ask the user for confirmation about the specified statement.
|
|
|
|
Args:
|
|
statement (unicode): statement to ask the user confirmation about.
|
|
|
|
Returns:
|
|
bool: whether or not specified statement was confirmed.
|
|
"""
|
|
prompt = f"{statement} [y/n]"
|
|
answer = ask(prompt, limited_to=["yes", "no", "y", "n"])
|
|
return answer and answer.startswith("y")
|
|
|
|
|
|
def password(statement: str) -> str:
|
|
"""Ask the user for a password.
|
|
|
|
Args:
|
|
statement (str): string to prompt the user with.
|
|
|
|
Returns:
|
|
str: password entered by the user.
|
|
"""
|
|
logger.info("%s: ", statement)
|
|
return getpass("")
|