Export SQLite Query Results to CSV in Python

import sqlite3
import csv

def export_query_to_csv(db_path, query, csv_path):
    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()
    cursor.execute(query)

    rows = cursor.fetchall()
    column_names = [description[0] for description in cursor.description]

    with open(csv_path, 'w', newline='', encoding='utf-8') as csvfile:
        writer = csv.writer(csvfile)
        writer.writerow(column_names)
        writer.writerows(rows)

    conn.close()
    print(f"Exported {len(rows)} rows to {csv_path}")

if __name__ == "__main__":
    # Create a sample in-memory database and export
    conn = sqlite3.connect(':memory:')
    conn.execute("CREATE TABLE employees (id INTEGER, name TEXT, department TEXT)")
    conn.executemany(
        "INSERT INTO employees VALUES (?, ?, ?)",
        [(1, 'Alice', 'Engineering'), (2, 'Bob', 'Marketing'), (3, 'Carol', 'Sales')]
    )
    conn.commit()
    conn.close()

    export_query_to_csv(
        ':memory:',
        "SELECT * FROM employees WHERE department != 'Sales'",
        'employees_export.csv'
    )