from database import postgresql_db_handler


def get_user_by_id(user_id):
    query = """
    SELECT * FROM users WHERE id = %s;
    """
    params = (user_id,)
    result = postgresql_db_handler.execute_query(query, params, fetchall=True)
    if result:
        user_data = result[0]
        return user_data
    else:
        return None


def get_user_by_email(email):
    query = """
    SELECT * FROM users WHERE email = %s;
    """
    params = (email,)
    result = postgresql_db_handler.execute_query(query, params, fetchall=True)
    if result:
        user_data = result[0]
        return user_data
    else:
        return None


def create_user_db(username, password, email):
    query = """
    INSERT INTO users (username, password, email)
    VALUES (%s, %s, %s);
    """
    params = (username, password, email)
    postgresql_db_handler.execute_query(query, params)


def update_user_db(user_id, new_username, new_password, new_email):
    query = """
    UPDATE users
    SET username = %s, password = %s, email = %s
    WHERE id = %s;
    """
    params = (new_username, new_password, new_email, user_id)
    postgresql_db_handler.execute_query(query, params)
    print(f"User with ID {user_id} updated successfully.")


def delete_user_db(user_id):
    query = """
    DELETE FROM users WHERE id = %s;
    """
    params = (user_id,)
    postgresql_db_handler.execute_query(query, params)
    print(f"User with ID {user_id} deleted successfully.")