scrubcsv 0.1.0

Remove bad lines from large CSV files and normalize the rest
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

//! Error-handling support implemented using the `[error-chain][]` crate.
//!
//! [error-chain]: https://docs.rs/error-chain

use csv;
use std::io;

// Declare nicer `Error` and `Result` types.
error_chain! {
    // Error types from other libraries that we want to just wrap
    // automatically.
    foreign_links {
        Csv(csv::Error);
        Io(io::Error);
    }

    errors {
        TooManyBadRows(bad: u64, total: u64) {
            description("a large portion of your rows were bad")
            display("a large portion of your rows ({} of {}) were bad", bad, total)
        }
    }
}

impl Error {
    /// Should we show a backtrace for this particular error?
    pub fn should_show_backtrace(&self) -> bool {
        match *self.kind() {
            ErrorKind::TooManyBadRows(_, _) => false,
            _ => true,
        }
    }

    /// What exit code should we return when the process exits?
    pub fn to_exit_code(&self) -> i32 {
        match *self.kind() {
            // This is only arguably an error, so return a special code for
            // people who want to try to ignore it.
            ErrorKind::TooManyBadRows(_, _) => 2,
            _ => 1,
        }
    }
}