Asking the user for input until they give a valid response

User Input

Valid Response

Input Validation

Programming Tips

User Interaction

Asking the user for input until they give a valid response

Interview Questions practice on Codemia

Over 8,000 real interview questions from top companies, searchable by company and role.

Browse interview questions

Introduction

Repeatedly prompting the user for input until they provide a valid response is a fundamental pattern in interactive programs. In Python, this is typically done with a while True loop that uses break when the input passes validation, or a while loop with a condition that becomes True once valid input is received. The key is to validate the input inside the loop, display clear error messages, and exit only when the response meets all criteria.

Basic Pattern: while True with break

python

1while True:
2    age = input("Enter your age: ")
3    if age.isdigit() and 0 < int(age) < 150:
4        age = int(age)
5        break
6    print("Please enter a valid age (1-149).")
7
8print(f"Your age is {age}")

The while True loop runs indefinitely until break is called. This pattern is clean because there is only one exit point — the break statement after validation succeeds.

Pattern: while with Sentinel Value

python

1age = None
2while age is None:
3    user_input = input("Enter your age: ")
4    if user_input.isdigit() and 0 < int(user_input) < 150:
5        age = int(user_input)
6    else:
7        print("Please enter a valid age (1-149).")
8
9print(f"Your age is {age}")

This variation uses None as a sentinel. The loop continues while age remains None, and only assigns a value when input is valid.

Validating Numeric Input

python

1def get_integer(prompt, min_val=None, max_val=None):
2    while True:
3        user_input = input(prompt)
4        try:
5            value = int(user_input)
6        except ValueError:
7            print(f"'{user_input}' is not a valid integer.")
8            continue
9
10        if min_val is not None and value < min_val:
11            print(f"Value must be at least {min_val}.")
12            continue
13        if max_val is not None and value > max_val:
14            print(f"Value must be at most {max_val}.")
15            continue
16
17        return value
18
19# Usage
20age = get_integer("Enter your age: ", min_val=1, max_val=149)
21quantity = get_integer("How many items? ", min_val=1)

Using try/except around int() handles non-numeric input gracefully. The continue statement skips back to the top of the loop on validation failure.

python

1def get_choice(prompt, valid_options):
2    options_str = ", ".join(valid_options)
3    while True:
4        choice = input(f"{prompt} ({options_str}): ").strip().lower()
5        if choice in valid_options:
6            return choice
7        print(f"Invalid choice. Please enter one of: {options_str}")
8
9# Usage
10color = get_choice("Pick a color", ["red", "green", "blue"])
11confirm = get_choice("Continue?", ["yes", "no"])

Validating with Regular Expressions

python

1import re
2
3def get_matching_input(prompt, pattern, error_msg="Invalid format."):
4    compiled = re.compile(pattern)
5    while True:
6        user_input = input(prompt).strip()
7        if compiled.fullmatch(user_input):
8            return user_input
9        print(error_msg)
10
11# Email (basic pattern)
12email = get_matching_input(
13    "Enter email: ",
14    r'[^@\s]+@[^@\s]+\.[^@\s]+',
15    "Please enter a valid email address."
16)
17
18# Phone number (US format)
19phone = get_matching_input(
20    "Enter phone (XXX-XXX-XXXX): ",
21    r'\d{3}-\d{3}-\d{4}',
22    "Please use format: XXX-XXX-XXXX"
23)

Limiting Retries

python

1def get_input_with_retries(prompt, validator, max_attempts=3):
2    for attempt in range(1, max_attempts + 1):
3        user_input = input(prompt).strip()
4        if validator(user_input):
5            return user_input
6        remaining = max_attempts - attempt
7        if remaining > 0:
8            print(f"Invalid input. {remaining} attempt(s) remaining.")
9        else:
10            print("Too many invalid attempts.")
11            return None
12
13# Usage
14result = get_input_with_retries(
15    "Enter a 4-digit PIN: ",
16    lambda s: s.isdigit() and len(s) == 4,
17    max_attempts=3
18)
19
20if result is None:
21    print("Access denied.")
22else:
23    print(f"PIN accepted: {result}")

Validating Multiple Fields

python

1def get_user_info():
2    while True:
3        name = input("Name: ").strip()
4        if not name:
5            print("Name cannot be empty.")
6            continue
7
8        age_str = input("Age: ").strip()
9        try:
10            age = int(age_str)
11            if age < 1 or age > 149:
12                raise ValueError
13        except ValueError:
14            print("Please enter a valid age (1-149).")
15            continue
16
17        email = input("Email: ").strip()
18        if "@" not in email or "." not in email:
19            print("Please enter a valid email.")
20            continue
21
22        return {"name": name, "age": age, "email": email}
23
24user = get_user_info()
25print(f"Registered: {user}")

Yes/No Confirmation Pattern

python

1def confirm(prompt="Are you sure?"):
2    while True:
3        response = input(f"{prompt} (y/n): ").strip().lower()
4        if response in ("y", "yes"):
5            return True
6        if response in ("n", "no"):
7            return False
8        print("Please enter 'y' or 'n'.")
9
10if confirm("Delete all files?"):
11    print("Deleting...")
12else:
13    print("Cancelled.")

Common Pitfalls

Forgetting to handle ValueError for numeric conversion: Calling int(input(...)) without try/except crashes the program when the user enters non-numeric text. Always wrap conversions in a try/except block.
Using input() return value without .strip(): Trailing whitespace or newlines from copy-pasted input can cause validation to fail unexpectedly. Call .strip() on user input before validating.
Not normalizing case for string comparisons: Checking choice == "Yes" fails when the user types "yes" or "YES". Use .lower() or .casefold() before comparing against lowercase options.
Infinite loop without clear error messages: If the loop only re-prompts without explaining what went wrong, users cannot correct their input. Always print a specific error message before the next iteration.
Validating only format, not semantics: An input like "2024-02-30" passes a regex date format check but is not a valid date. Use domain-specific validation (like datetime.strptime) in addition to format checks.

Summary

Use while True with break for the cleanest input validation loop pattern
Wrap numeric conversions in try/except ValueError to handle non-numeric input
Create reusable validation functions with configurable constraints (min/max, regex, allowed values)
Always call .strip() and normalize case before comparing string input
Limit retries with a for loop when unlimited attempts are undesirable
Print specific error messages so users know how to correct their input

Free course

Beginner

7 lessons

2 hours

Tackling System Design Interview Problems

A short course that equips you with the skills to approach system design interviews methodically.

Start the free course

Track what you have practised

A free account saves your progress, solutions and study plan across every problem on Codemia.

Interview Questions practice on Codemia

Over 8,000 real interview questions from top companies, searchable by company and role.

Browse interview questions

Asking the user for input until they give a valid response

Interview Questions practice on Codemia

Introduction

Basic Pattern: while True with break

Pattern: while with Sentinel Value

Validating Numeric Input

Validating String Input (Menu Selection)

Validating with Regular Expressions

Limiting Retries

Validating Multiple Fields

Yes/No Confirmation Pattern

Common Pitfalls

Summary

Tackling System Design Interview Problems

Track what you have practised

Interview Questions practice on Codemia