Read csv from stdin too
ci/woodpecker/push/woodpecker Pipeline was successful Details

main
Guillem Borrell 1 year ago
parent e4e9b71674
commit 1e18c9ae9f

@ -110,31 +110,31 @@ fn main() {
) )
.get_matches(); .get_matches();
if let Some(_matches) = matches.subcommand_matches("csv") { if let Some(_matches) = matches.subcommand_matches("csv") {
if let Some(path) = _matches.get_one::<String>("path") { let mut ldf = if _matches.get_flag("stdin") {
let mut ldf = LazyFrame::default(); io::load_csv_from_stdin()
if _matches.get_flag("stdin") { } else {
ldf = io::load_csv_from_stdin(); let path = _matches
} else { .get_one::<String>("path")
ldf = io::read_csv(path.to_string()); .expect("Please, provide a file");
} io::read_csv(path.to_string())
if let Some(query) = _matches.get_one::<String>("query") { };
ldf = sql::execute(ldf, query); if let Some(query) = _matches.get_one::<String>("query") {
} ldf = sql::execute(ldf, query);
if _matches.get_flag("summary") { }
let df = ldf.collect().expect("Could not collect"); if _matches.get_flag("summary") {
println!("{}", df.describe(None)); let df = ldf.collect().expect("Could not collect");
} else if _matches.get_flag("head") { println!("{}", df.describe(None));
let df = ldf.fetch(5).expect("Could not fetch"); } else if _matches.get_flag("head") {
println!("{}", df) let df = ldf.fetch(5).expect("Could not fetch");
println!("{}", df)
} else {
if _matches.get_flag("text") {
io::dump_csv_to_stdout(ldf);
} else { } else {
if _matches.get_flag("text") { if let Some(path) = _matches.get_one::<String>("parquet") {
io::dump_csv_to_stdout(ldf); io::write_parquet(ldf, path.to_string());
} else { } else {
if let Some(path) = _matches.get_one::<String>("parquet") { io::write_ipc(ldf);
io::write_parquet(ldf, path.to_string());
} else {
io::write_ipc(ldf);
}
} }
} }
} }

Loading…
Cancel
Save