Блог

Я заметил, что запрос с использованием click принимает входные данные с конечными пробелами

 ftp_server = click.prompt("FTP Server")
  

Есть ли способ использовать пользовательскую функцию возврата, подобную этой, чтобы отклонить пробелы в конце?

 def custom_prompt(value):
    if value.strip():
        return True
    else:
        return False

ftp_server = click.prompt("FTP Server", custom_prompt)
  

Я уже использовал это:

 while not ftp_server.strip():
    ftp_server = click.prompt("FTP Server")
  

Но я ищу способ получше, потому что я не хочу использовать цикл while каждый раз, когда я использую приглашение.

Чтобы отклонить неверный пользовательский ввод, вы можете использовать value_proc параметр to click.prompt . Средство проверки для отклонения ввода с конечными пробелами может выглядеть следующим образом:

Средство проверки запросов

 import click

def validate_no_trailing_space(value):
    if value != value.rstrip():
        raise click.UsageError("Trailing space is invalid!")
    return value

ftp_server = click.prompt("FTP Server",
                          value_proc=validate_no_trailing_space)
  

Обрезать пробелы

Вы могли бы также рассмотреть средство проверки, которое обрезает начальные и конечные пробелы, но отклоняет пробелы в имени:

 def validate_no_internal_space(value):
    value = value.strip()
    if ' ' in value:
        raise click.UsageError("Spaces are not valid here!")
    return value