polite 0.1.3

rusqlite × Polars bridge library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use polite::prelude::*;
use tempfile::NamedTempFile;

#[test]
fn test_repo_example() -> anyhow::Result<()> {
    // Create a temporary file-backed SQLite DB
    let db = NamedTempFile::new()?;
    let db_path = db.path().to_str().unwrap();

    // Create a connection
    let conn = connect_sqlite(Some(db_path))?;

    // Create and populate a table
    execute_query(
        &conn,
        "CREATE TABLE IF NOT EXISTS users (id INTEGER, name TEXT)",
    )?;
    execute_query(&conn, "INSERT INTO users VALUES (1, 'Alice')")?;
    execute_query(&conn, "INSERT INTO users VALUES (2, 'Bob')")?;

    // Query into a Polars DataFrame
    let df = to_dataframe(db_path, "SELECT * FROM users")?;
    println!("Loaded DataFrame:\n{df:?}");

    // Write it back into a new table
    from_dataframe(&conn, "users_copy", &df)?;
    println!("Saved DataFrame into table `users_copy`");

    Ok(())
}