liboxen 0.46.10

Oxen is a fast, unstructured data version control, to help version large machine learning datasets written in Rust.
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
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

use crate::error::OxenError;
use crate::opts::DFOpts;
use minus::Pager;
/// # Pretty Print
/// The default Polars DataFrame print implementation puts characters
/// that do not fit well in our docs, so this transform helps cleanup the output.
///
/// ## Polars Example
/// ┌────────┬──────────┬───────────────────┐
/// │ prompt ┆ category ┆ .oxen.diff.status │
/// │ ---    ┆ ---      ┆ ---               │
/// │ str    ┆ str      ┆ str               │
/// ╞════════╪══════════╪═══════════════════╡
/// │ 20*20  ┆ math     ┆ added             │
/// └────────┴──────────┴───────────────────┘
///
/// ## Oxen Example
/// +--------+----------+-------------------+
/// | prompt ┆ category ┆ .oxen.diff.status |
/// | ---    ┆ ---      ┆ ---               |
/// | str    ┆ str      ┆ str               |
/// +--------+----------+-------------------+
/// | 20*20  ┆ math     ┆ added             |
/// +--------+----------+-------------------+
use polars::prelude::*;
use std::cmp;
use std::env;
use std::fmt::Write;

fn write_to_pager(output: &mut Pager, text: &str) -> Result<(), OxenError> {
    match writeln!(output, "{text}") {
        Ok(_) => Ok(()),
        Err(_) => Err(OxenError::basic_str("Could not write to pager")),
    }
}

pub fn df_to_str(df: &DataFrame) -> String {
    let default_fmt = format!("{df:?}");

    default_fmt
        .replace(['┌', '└', '┬', '┴', '┐', '┘'], "+")
        .replace('─', "-")
        .replace('│', "|")
        .replace(['╞', '╡'], "+")
        .replace('═', "-")
        .replace('╪', "+")
        .replace('┆', "|")
}

pub fn df_to_pager(df: &DataFrame, opts: &DFOpts) -> Result<Pager, OxenError> {
    let height = df.height();
    let max_rows = height + 10;
    // Safety: We do not expect any concurrent usage of this environment variable.
    unsafe {
        env::set_var("POLARS_FMT_MAX_ROWS", max_rows.to_string());
    }

    let page_size: usize = opts.page_size.unwrap_or(10);
    let start: usize = if let Some(page) = opts.page {
        cmp::max(1, (page - 1) * page_size + 1)
    } else {
        1
    };

    let default_fmt = format!("{df:?}");
    let pretty_fmt = default_fmt
        .replace(['┌', '└', '┬', '┴', '┐', '┘'], "+")
        .replace('─', "-")
        .replace('│', "|")
        .replace(['╞', '╡'], "+")
        .replace('═', "-")
        .replace('╪', "+")
        .replace('┆', "|");

    let mut lines = pretty_fmt.lines();
    let first_line = lines.clone().nth(1).unwrap();

    let mut output = Pager::new();

    for _line in 0..5 {
        write_to_pager(&mut output, lines.next().unwrap())?;
    }

    if start > height {
        write_to_pager(&mut output, first_line)?;
        write_to_pager(&mut output, first_line)?;
        // Safety: We do not expect any concurrent usage of this environment variable.
        unsafe {
            env::set_var("POLARS_FMT_MAX_ROWS", "10");
        }
        return Ok(output);
    }

    for _line in 0..start {
        lines.next();
    }

    for line in 0..height - start + 1 {
        if line % page_size == 0 {
            write_to_pager(&mut output, first_line)?;
        }
        write_to_pager(&mut output, lines.next().unwrap())?;
    }

    write_to_pager(&mut output, first_line)?;
    // Safety: We do not expect any concurrent usage of this environment variable.
    unsafe {
        env::set_var("POLARS_FMT_MAX_ROWS", "10");
    }

    Ok(output)
}