csv-slice 0.1.0

Extract rows or columns from CSV files without loading the entire file
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

// create cli interface 

use clap::{Parser, Subcommand};

#[derive(Parser)]
#[command(name = "csv-slice")]
#[command(about = "Extract rows or columns from CSV files", long_about = None)]
struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    Rows {
        #[arg(short, long)]
        input: String,
        #[arg(short, long)]
        start: usize,
        #[arg(short, long)]
        end: usize,
    },
    Columns {
        #[arg(short, long)]
        input: String,
        #[arg(short, long)]
        columns: Vec<String>,
    },
}

use csv_slice::CsvSliceError;

fn main() -> Result<(), CsvSliceError> {
    let cli = Cli::parse();

    match &cli.command {
        Commands::Rows { input, start, end } => {
            let rows = csv_slice::extract_rows(input, *start, *end)?;
            for row in rows {
                println!("{}", row.iter().collect::<Vec<_>>().join(","));
            }
        }
        Commands::Columns { input, columns } => {
            let cols = csv_slice::extract_columns(input, &columns.iter().map(|s| s.as_str()).collect::<Vec<_>>())?;
            for row in cols {
                println!("{}", row.join(","));
            }
        }
    }
    Ok(())
}