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

use polars::prelude::{DataType, Schema};
use rusqlite::Statement;

/// Polars Schema from SQLite query (parsed from a string by `Connection::prepare`)
pub(crate) fn schema_from_sqlite(stmt: &Statement) -> Schema {
    Schema::from_iter(stmt.columns().iter().map(|col| {
        let name = col.name();
        let decl_type = col.decl_type();

        let dtype = match decl_type {
            Some("INTEGER") => DataType::Int64,
            Some("REAL") => DataType::Float64,
            Some("TEXT") => DataType::String,
            _ => DataType::String,
        };

        (name.into(), dtype)
    }))
}