main.py

import string
import secrets

def generate_password(length=12, use_special=True, use_numbers=True):
    # Check the minimum length to avoid an infinite loop
    if length < 8:
        raise ValueError("Password length must be at least 8 characters.")
    
    # Define basic character sets
    letters = string.ascii_letters
    digits = string.digits
    special = string.punctuation

    # Build the character pool based on user options
    alphabet = letters
    if use_numbers:
        alphabet += digits
    if use_special:
        alphabet += special

    while True:
        # Generate the password using the secure secrets module
        password = ''.join(secrets.choice(alphabet) for _ in range(length))
        
        # Validate whether the password meets all requirements
        has_upper = any(c.isupper() for c in password)
        has_lower = any(c.islower() for c in password)
        has_digit = any(c.isdigit() for c in password) or not use_numbers
        has_spec = any(c in special for c in password) or not use_special

        # If the password is strong enough, return it
        if has_upper and has_lower and has_digit and has_spec:
            return password


# --- USER INTERFACE ---
print("--- Secure Password Generator ---")

try:
    # Get password length with validation
    while True:
        try:
            length = int(input("Enter password length (min. 8, max. 128 characters): "))
            if length < 8:
                print("Password too short! Setting length to 8.")
                length = 8
            elif length > 128:
                print("Password too long! Setting length to 128.")
                length = 128
            break
        except ValueError:
            print("Error: Length must be a number. Please try again.")

    # Get options with validation
    while True:
        use_special = input("Include special characters? (y/n): ").lower()
        if use_special in ['y', 'n']:
            use_special = use_special == 'y'
            break
        print("Error: Please answer 'y' or 'n'.")

    while True:
        use_numbers = input("Include numbers? (y/n): ").lower()
        if use_numbers in ['y', 'n']:
            use_numbers = use_numbers == 'y'
            break
        print("Error: Please answer 'y' or 'n'.")

    # Call the function with selected parameters
    new_password = generate_password(length, use_special, use_numbers)
    print(f"\nYour secure password: {new_password}")

except ValueError as e:
    # Handle validation errors
    print(f"Error: {e}")
except Exception as e:
    # Handle unexpected errors
    print(f"Unexpected error: {e}")