46 lines
1.3 KiB
Rust
46 lines
1.3 KiB
Rust
|
use polars::frame::DataFrame;
|
||
|
|
use polars::prelude::*;
|
||
|
|
use std::io;
|
||
|
|
use std::io::Read;
|
||
|
|
|
||
|
|
/// Read CSV format from stdin and return a Polars DataFrame
|
||
|
|
pub fn load_csv_from_stdin() -> DataFrame {
|
||
|
|
let mut buffer = String::new();
|
||
|
|
let _res: () = match io::stdin().read_to_string(&mut buffer) {
|
||
|
|
Ok(_ok) => (),
|
||
|
|
Err(_e) => (),
|
||
|
|
};
|
||
|
|
let cursor = io::Cursor::new(buffer.as_bytes());
|
||
|
|
let df = match CsvReader::new(cursor).finish() {
|
||
|
|
Ok(df) => df,
|
||
|
|
Err(_e) => DataFrame::default(),
|
||
|
|
};
|
||
|
|
df
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Take a Polars Dataframe and write it as CSV to stdout
|
||
|
|
pub fn dump_csv_to_stdout(df: &mut DataFrame) {
|
||
|
|
let _res: () = match CsvWriter::new(io::stdout().lock()).finish(df) {
|
||
|
|
Ok(_ok) => (),
|
||
|
|
Err(_e) => (),
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Read parquet format from stdin and return a Polars DataFrame
|
||
|
|
pub fn load_parquet_from_stdin() -> DataFrame {
|
||
|
|
let mut buffer: String = String::new();
|
||
|
|
let _res: () = match io::stdin().read_to_string(&mut buffer) {
|
||
|
|
Ok(_ok) => (),
|
||
|
|
Err(_e) => (),
|
||
|
|
};
|
||
|
|
let cursor = io::Cursor::new(buffer.as_bytes());
|
||
|
|
let df = match ParquetReader::new(cursor).finish() {
|
||
|
|
Ok(df) => df,
|
||
|
|
Err(e) => {
|
||
|
|
eprintln!("{e}");
|
||
|
|
DataFrame::default()
|
||
|
|
}
|
||
|
|
};
|
||
|
|
df
|
||
|
|
}
|