from itertools import chain, repeat
prompts = chain(["Enter a number: "], repeat("Not a number! Try again: "))
replies = map(input, prompts)
valid_response = next(filter(str.isdigit, replies))
print(valid_response)
Enter a number: a
Not a number! Try again: b
Not a number! Try again: 1
1
або якщо ви хочете, щоб повідомлення про неправильний вхід було відокремлено від вхідного запиту, як і в інших відповідях:
prompt_msg = "Enter a number: "
bad_input_msg = "Sorry, I didn't understand that."
prompts = chain([prompt_msg], repeat('\n'.join([bad_input_msg, prompt_msg])))
replies = map(input, prompts)
valid_response = next(filter(str.isdigit, replies))
print(valid_response)
Enter a number: a
Sorry, I didn't understand that.
Enter a number: b
Sorry, I didn't understand that.
Enter a number: 1
1
Як це працює?
prompts = chain(["Enter a number: "], repeat("Not a number! Try again: "))
Ця комбінація itertools.chain
і itertools.repeat
створить ітератор, який дасть рядки "Enter a number: "
один раз і "Not a number! Try again: "
нескінченну кількість разів:
for prompt in prompts:
print(prompt)
Enter a number:
Not a number! Try again:
Not a number! Try again:
Not a number! Try again:
# ... and so on
replies = map(input, prompts)
- тут map
будуть застосовані всі prompts
рядки з попереднього кроку до input
функції. Наприклад:
for reply in replies:
print(reply)
Enter a number: a
a
Not a number! Try again: 1
1
Not a number! Try again: it doesn't care now
it doesn't care now
# and so on...
- Ми використовуємо
filter
та str.isdigit
фільтруємо ті рядки, які містять лише цифри:
only_digits = filter(str.isdigit, replies)
for reply in only_digits:
print(reply)
Enter a number: a
Not a number! Try again: 1
1
Not a number! Try again: 2
2
Not a number! Try again: b
Not a number! Try again: # and so on...
І щоб отримати лише перші рядки, які використовуються лише для цифр, ми використовуємо next
.
Інші правила перевірки:
Методи струн: Звичайно, ви можете використовувати інші рядкові методи, наприклад, str.isalpha
щоб отримати лише алфавітні рядки або str.isupper
отримати лише великі регістри. Повний список див. У документах .
Тестування членства:
Існує кілька різних способів його виконання. Один з них - за допомогою __contains__
методу:
from itertools import chain, repeat
fruits = {'apple', 'orange', 'peach'}
prompts = chain(["Enter a fruit: "], repeat("I don't know this one! Try again: "))
replies = map(input, prompts)
valid_response = next(filter(fruits.__contains__, replies))
print(valid_response)
Enter a fruit: 1
I don't know this one! Try again: foo
I don't know this one! Try again: apple
apple
Порівняння чисел:
Є корисні методи порівняння, які ми можемо використовувати тут. Наприклад, для __lt__
( <
):
from itertools import chain, repeat
prompts = chain(["Enter a positive number:"], repeat("I need a positive number! Try again:"))
replies = map(input, prompts)
numeric_strings = filter(str.isnumeric, replies)
numbers = map(float, numeric_strings)
is_positive = (0.).__lt__
valid_response = next(filter(is_positive, numbers))
print(valid_response)
Enter a positive number: a
I need a positive number! Try again: -5
I need a positive number! Try again: 0
I need a positive number! Try again: 5
5.0
Або, якщо вам не подобається використання методів dnder (dunder = подвійне підкреслення), ви завжди можете визначити власну функцію або використовувати ті з operator
модуля.
Існування шляху:
тут можна використовувати pathlib
бібліотеку та її Path.exists
метод:
from itertools import chain, repeat
from pathlib import Path
prompts = chain(["Enter a path: "], repeat("This path doesn't exist! Try again: "))
replies = map(input, prompts)
paths = map(Path, replies)
valid_response = next(filter(Path.exists, paths))
print(valid_response)
Enter a path: a b c
This path doesn't exist! Try again: 1
This path doesn't exist! Try again: existing_file.txt
existing_file.txt
Обмежувальна кількість спроб:
Якщо ви не хочете катувати користувача, запитуючи його щось нескінченну кількість разів, ви можете вказати ліміт у виклику itertools.repeat
. Це може поєднуватися із наданням функції за замовчуванням для next
функції:
from itertools import chain, repeat
prompts = chain(["Enter a number:"], repeat("Not a number! Try again:", 2))
replies = map(input, prompts)
valid_response = next(filter(str.isdigit, replies), None)
print("You've failed miserably!" if valid_response is None else 'Well done!')
Enter a number: a
Not a number! Try again: b
Not a number! Try again: c
You've failed miserably!
Попередня обробка вхідних даних:
Іноді ми не хочемо відхиляти введення даних, якщо користувач випадково надав його IN CAPS або з пробілом на початку або в кінці рядка. Для врахування цих простих помилок ми можемо попередньо обробити вхідні дані за допомогою застосувань str.lower
та str.strip
методів. Наприклад, у випадку тестування членства код буде виглядати приблизно так:
from itertools import chain, repeat
fruits = {'apple', 'orange', 'peach'}
prompts = chain(["Enter a fruit: "], repeat("I don't know this one! Try again: "))
replies = map(input, prompts)
lowercased_replies = map(str.lower, replies)
stripped_replies = map(str.strip, lowercased_replies)
valid_response = next(filter(fruits.__contains__, stripped_replies))
print(valid_response)
Enter a fruit: duck
I don't know this one! Try again: Orange
orange
У випадку, коли у вас є багато функцій, які потрібно використовувати для попередньої обробки, може бути простіше використовувати функцію, що виконує композицію функції . Наприклад, використовуючи один звідси :
from itertools import chain, repeat
from lz.functional import compose
fruits = {'apple', 'orange', 'peach'}
prompts = chain(["Enter a fruit: "], repeat("I don't know this one! Try again: "))
replies = map(input, prompts)
process = compose(str.strip, str.lower) # you can add more functions here
processed_replies = map(process, replies)
valid_response = next(filter(fruits.__contains__, processed_replies))
print(valid_response)
Enter a fruit: potato
I don't know this one! Try again: PEACH
peach
Поєднання правил перевірки:
Наприклад, для простого випадку, коли програма просить вік від 1 до 120 років, можна просто додати інший filter
:
from itertools import chain, repeat
prompt_msg = "Enter your age (1-120): "
bad_input_msg = "Wrong input."
prompts = chain([prompt_msg], repeat('\n'.join([bad_input_msg, prompt_msg])))
replies = map(input, prompts)
numeric_replies = filter(str.isdigit, replies)
ages = map(int, numeric_replies)
positive_ages = filter((0).__lt__, ages)
not_too_big_ages = filter((120).__ge__, positive_ages)
valid_response = next(not_too_big_ages)
print(valid_response)
Але у випадку, коли існує багато правил, краще реалізувати функцію, що виконує логічний сполучник . У наступному прикладі я буду використовувати готовий звідси :
from functools import partial
from itertools import chain, repeat
from lz.logical import conjoin
def is_one_letter(string: str) -> bool:
return len(string) == 1
rules = [str.isalpha, str.isupper, is_one_letter, 'C'.__le__, 'P'.__ge__]
prompt_msg = "Enter a letter (C-P): "
bad_input_msg = "Wrong input."
prompts = chain([prompt_msg], repeat('\n'.join([bad_input_msg, prompt_msg])))
replies = map(input, prompts)
valid_response = next(filter(conjoin(*rules), replies))
print(valid_response)
Enter a letter (C-P): 5
Wrong input.
Enter a letter (C-P): f
Wrong input.
Enter a letter (C-P): CDE
Wrong input.
Enter a letter (C-P): Q
Wrong input.
Enter a letter (C-P): N
N
На жаль, якщо комусь потрібне спеціальне повідомлення для кожного невдалого випадку, то, боюся, немає досить функціонального способу. Або, принаймні, я не міг її знайти.