Adding Rust set_ascii_tables

Author: Bidek56

__tests__/config.test.ts

@@ -0,0 +1,44 @@
+import pl from "@polars";
+
+describe("config", () => {
+    pl.Config.setAsciiTables(true);
+
+    test("setAsciiTables", () => {
+        const df = pl.DataFrame({
+            int: [1, 2],
+            str: ["a", "b"],
+            bool: [true, null],
+            list: [[1, 2], [3]],
+        });
+
+        const asciiTable = `shape: (2, 4)
++-----+-----+------+------------+
+| int | str | bool | list       |
+| --- | --- | ---  | ---        |
+| f64 | str | bool | list[f64]  |
++===============================+
+| 1.0 | a   | true | [1.0, 2.0] |
+| 2.0 | b   | null | [3.0]      |
++-----+-----+------+------------+`;
+
+    pl.Config.setAsciiTables(true);    
+    expect(df.toString()).toEqual(asciiTable);
+
+    pl.Config.setAsciiTables(false);
+
+    const utf8Table = `shape: (2, 4)
+┌─────┬─────┬──────┬────────────┐
+│ int ┆ str ┆ bool ┆ list       │
+│ --- ┆ --- ┆ ---  ┆ ---        │
+│ f64 ┆ str ┆ bool ┆ list[f64]  │
+╞═════╪═════╪══════╪════════════╡
+│ 1.0 ┆ a   ┆ true ┆ [1.0, 2.0] │
+│ 2.0 ┆ b   ┆ null ┆ [3.0]      │
+└─────┴─────┴──────┴────────────┘`;
+
+    expect(df.toString()).toEqual(utf8Table);
+    pl.Config.setAsciiTables();
+
+    expect(df.toString()).toEqual(utf8Table);
+  });
+});

polars/cfg.ts

@@ -1,3 +1,5 @@
+import pli from "./internals/polars_internal";
+
 /**
  * Configure polars; offers options for table formatting and more.
  */
@@ -6,12 +8,38 @@ export interface Config {
   setUtf8Tables(): Config;
   /** Use ascii characters to print tables */
   setAsciiTables(): Config;
+  setAsciiTables(active: boolean): Config;
   /** Set the number of character used to draw the table */
   setTblWidthChars(width: number): Config;
   /** Set the number of rows used to print tables */
   setTblRows(n: number): Config;
   /** Set the number of columns used to print tables */
   setTblCols(n: number): Config;
+
+  // set_auto_structify
+  // set_decimal_separator
+  //   set_engine_affinity
+  //   set_float_precision
+  //   set_fmt_float
+  //   set_fmt_str_lengths
+  //   set_fmt_table_cell_list_len
+  //   set_streaming_chunk_size
+  //   set_tbl_cell_alignment
+  //   set_tbl_cell_numeric_alignment
+  //   set_tbl_cols
+  //   set_tbl_column_data_type_inline
+  //   set_tbl_dataframe_shape_below
+  //   set_tbl_formatting
+  //   set_tbl_hide_column_data_types
+  //   set_tbl_hide_column_names
+  //   set_tbl_hide_dataframe_shape
+  //   set_tbl_hide_dtype_separator
+  //   set_tbl_rows
+  //   set_tbl_width_chars
+  //   set_thousands_separator
+  //   set_trim_decimal_zeros
+  //   set_verbose
+
 }
 
 /**
@@ -22,8 +50,8 @@ export const Config: Config = {
     process.env["POLARS_FMT_NO_UTF8"] = undefined;
     return this;
   },
-  setAsciiTables() {
-    process.env["POLARS_FMT_NO_UTF8"] = "1";
+  setAsciiTables(active?: boolean) {
+    pli.setAsciiTables(active);
     return this;
   },
   setTblWidthChars(width) {

src/cfg.rs

@@ -0,0 +1,16 @@
+
+/// Set the `POLARS_FMT_TABLE_FORMATTING` environment variable.
+///
+/// - `Some(true)`  => `"ASCII_FULL_CONDENSED"`
+/// - `Some(false)` => `"UTF8_FULL_CONDENSED"`
+/// - `None`        => remove the variable
+#[napi(catch_unwind)]
+pub fn set_ascii_tables(enabled: Option<bool>) {
+    const ENV_VAR: &str = "POLARS_FMT_TABLE_FORMATTING";
+
+    match enabled {
+        Some(true) => std::env::set_var(ENV_VAR, "ASCII_FULL_CONDENSED"),
+        Some(false) => std::env::set_var(ENV_VAR, "UTF8_FULL_CONDENSED"),
+        None => std::env::remove_var(ENV_VAR),
+    }
+}

src/lib.rs

@@ -23,6 +23,7 @@ pub mod series;
 pub mod set;
 pub mod sql;
 pub mod utils;
+pub mod cfg;
 
 pub use polars_core;