SearchForUsers.py

from db import get_connection
import sqlite3


def mostrar(resultado):

    if not resultado:
        print("\nNo records found.")
        return

    print("\n{:<3} {:<12} {:<15} {:<5} {:<10}".format(
        "ID", "Name", "Surname", "Age", "Status"))
    print("-" * 55)

    for linha in resultado:
        print("{:<3} {:<12} {:<15} {:<5} {:<10}".format(*linha))


# Verifica a conexão
try:
    conn = get_connection()
    cursor = conn.cursor()

    cursor.execute("SELECT COUNT(*) FROM users")
    total = cursor.fetchone()[0]

    print(f"\nDatabase connected successfully.")
    print(f"Total users: {total}")

    conn.close()

except sqlite3.Error as erro:
    print(f"Database Error: {erro}")
    exit()


while True:

    print("\n========== USER DATABASE ==========")
    print("1 - Search by Name")
    print("2 - Search by Surname")
    print("3 - Search by Age")
    print("4 - Search by Status")
    print("5 - Show All Users")
    print("0 - Exit")

    opcao = input("\nChoose an option: ").strip()

    try:

        conn = get_connection()
        cursor = conn.cursor()

        if opcao == "1":

            nome = input("Name: ").strip()

            cursor.execute(
                "SELECT * FROM users WHERE nome LIKE ?",
                (f"%{nome}%",)
            )

            mostrar(cursor.fetchall())

        elif opcao == "2":

            sobrenome = input("Surname: ").strip()

            cursor.execute(
                "SELECT * FROM users WHERE sobrenome LIKE ?",
                (f"%{sobrenome}%",)
            )

            mostrar(cursor.fetchall())

        elif opcao == "3":

            idade = input("Age: ").strip()

            cursor.execute(
                "SELECT * FROM users WHERE idade = ?",
                (idade,)
            )

            mostrar(cursor.fetchall())

        elif opcao == "4":

            status = input("Status (Active/Unactive): ").strip().capitalize()

            cursor.execute(
                "SELECT * FROM users WHERE status = ?",
                (status,)
            )

            mostrar(cursor.fetchall())

        elif opcao == "5":

            cursor.execute(
                "SELECT * FROM users ORDER BY nome"
            )

            mostrar(cursor.fetchall())

        elif opcao == "0":

            print("\nGoodbye!")
            break

        else:

            print("\nInvalid option.")

    except sqlite3.Error as erro:

        print(f"\nSQLite Error: {erro}")

    finally:

        conn.close()