datui-lib 0.2.53

Data Exploration in the Terminal (library)
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

use polars::prelude::*;
use ratatui::{
    buffer::Buffer,
    layout::Rect,
    style::{Modifier, Style},
    text::Span,
    widgets::{Block, BorderType, Borders, Cell, Row, Table, Widget},
};

struct SchemaView<'a> {
    lf: &'a LazyFrame,
}

impl<'a> Widget for &'a SchemaView<'a> {
    fn render(self, area: Rect, buf: &mut Buffer) {
        let block = Block::default()
            .title("Schema")
            .borders(Borders::ALL)
            .border_type(BorderType::Rounded);

        let rows: Vec<Row> = if let Ok(schema) = self.lf.clone().collect_schema() {
            schema
                .iter()
                .map(|(name, dtype)| {
                    let cells: Vec<Cell> = vec![
                        Cell::from(Span::raw(name.to_string())),
                        Cell::from(Span::raw(dtype.to_string())),
                    ];
                    Row::new(cells)
                })
                .collect()
        } else {
            vec![]
        };
        // iterate the schema and create two colum table,
        // one column for the name and one for the type

        // calculate widths as fractions of total width, equal per column
        let widths = vec![area.width / 2; 2];
        // Create and render the table
        let table = Table::new(rows, widths)
            .header(Row::new(vec![
                Cell::from(Span::styled(
                    "Column",
                    Style::default().add_modifier(Modifier::BOLD),
                )),
                Cell::from(Span::styled(
                    "Type",
                    Style::default().add_modifier(Modifier::BOLD),
                )),
            ]))
            .block(block);

        Widget::render(table, area, buf);
    }
}