tabiew 0.16.0

A lightweight TUI application to view and query tabular data files, such as CSV, TSV, and parquet.
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

use clap::Parser;
use polars::frame::DataFrame;
use rayon::iter::{IntoParallelRefIterator, ParallelIterator};

use crate::{
    args::{Args, Type},
    parsers::auto_parser::AutoParser,
};

#[derive(Default, Clone)]
pub struct DataFrameParser<'a> {
    auto_parser: AutoParser<'a>,
}

impl<'a> DataFrameParser<'a> {
    pub fn with_bool(self, enabled: bool) -> Self {
        Self {
            auto_parser: self.auto_parser.with_bool(enabled),
        }
    }

    pub fn with_date(self, enabled: bool) -> Self {
        Self {
            auto_parser: self.auto_parser.with_date(enabled),
        }
    }

    pub fn with_datetime(self, enabled: bool) -> Self {
        Self {
            auto_parser: self.auto_parser.with_datetime(enabled),
        }
    }

    pub fn with_float(self, enabled: bool) -> Self {
        Self {
            auto_parser: self.auto_parser.with_float(enabled),
        }
    }

    pub fn with_int(self, enabled: bool) -> Self {
        Self {
            auto_parser: self.auto_parser.with_int(enabled),
        }
    }

    pub fn parse_and_update(&self, df: &mut DataFrame) {
        df.columns()
            .par_iter()
            .filter(|c| c.dtype().is_string())
            .filter_map(move |column| self.auto_parser.clone().parse_strict(column).ok())
            .collect::<Vec<_>>()
            .into_iter()
            .for_each(|column| {
                let _ = df.with_column(column);
            });
    }

    pub fn from_env_args() -> Self {
        let args = Args::parse_from(std::env::args_os());
        if args.no_type_inference {
            DataFrameParser::default()
        } else {
            args.infer_types
                .inner()
                .iter()
                .fold(DataFrameParser::default(), |p, t| match t {
                    Type::All => p
                        .with_int(true)
                        .with_float(true)
                        .with_bool(true)
                        .with_date(true)
                        .with_datetime(true),
                    Type::Int => p.with_int(true),
                    Type::Float => p.with_float(true),
                    Type::Boolean => p.with_bool(true),
                    Type::Date => p.with_date(true),
                    Type::Datetime => p.with_datetime(true),
                })
        }
    }
}