rename to PyDataframeDisplayConfig

Author: kosiew

src/context.rs

@@ -73,21 +73,69 @@ use pyo3::types::{PyCapsule, PyDict, PyList, PyTuple, PyType};
 use tokio::task::JoinHandle;
 
 /// Configuration for displaying DataFrames
+#[pyclass(name = "DataframeDisplayConfig", module = "datafusion", subclass)]
+#[derive(Clone)]
+pub struct PyDataframeDisplayConfig {
+    /// Maximum bytes to display for table presentation (default: 2MB)
+    #[pyo3(get, set)]
+    pub max_table_bytes: usize,
+    /// Minimum number of table rows to display (default: 20)
+    #[pyo3(get, set)]
+    pub min_table_rows: usize,
+    /// Maximum length of a cell before it gets minimized (default: 25)
+    #[pyo3(get, set)]
+    pub max_cell_length: usize,
+    /// Maximum number of rows to display in repr string output (default: 10)
+    #[pyo3(get, set)]
+    pub max_table_rows_in_repr: usize,
+}
+
+#[pymethods]
+impl PyDataframeDisplayConfig {
+    #[new]
+    #[pyo3(signature = (max_table_bytes=None, min_table_rows=None, max_cell_length=None, max_table_rows_in_repr=None))]
+    fn new(
+        max_table_bytes: Option<usize>,
+        min_table_rows: Option<usize>,
+        max_cell_length: Option<usize>,
+        max_table_rows_in_repr: Option<usize>,
+    ) -> Self {
+        let default = Self::default();
+        Self {
+            max_table_bytes: max_table_bytes.unwrap_or(default.max_table_bytes),
+            min_table_rows: min_table_rows.unwrap_or(default.min_table_rows),
+            max_cell_length: max_cell_length.unwrap_or(default.max_cell_length),
+            max_table_rows_in_repr: max_table_rows_in_repr
+                .unwrap_or(default.max_table_rows_in_repr),
+        }
+    }
+}
+
+impl Default for PyDataframeDisplayConfig {
+    fn default() -> Self {
+        Self {
+            max_table_bytes: 2 * 1024 * 1024, // 2 MB
+            min_table_rows: 20,
+            max_cell_length: 25,
+            max_table_rows_in_repr: 10,
+        }
+    }
+}
 
 /// Configuration options for a SessionContext
 #[pyclass(name = "SessionConfig", module = "datafusion", subclass)]
 #[derive(Clone, Default)]
 pub struct PySessionConfig {
     pub config: SessionConfig,
     #[pyo3(get, set)]
-    pub display_config: DataframeDisplayConfig,
+    pub display_config: PyDataframeDisplayConfig,
 }
 
 impl From<SessionConfig> for PySessionConfig {
     fn from(config: SessionConfig) -> Self {
         Self {
             config,
-            display_config: DataframeDisplayConfig::default(),
+            display_config: PyDataframeDisplayConfig::default(),
         }
     }
 }
@@ -106,7 +154,7 @@ impl PySessionConfig {
 
         Self {
             config,
-            display_config: DataframeDisplayConfig::default(),
+            display_config: PyDataframeDisplayConfig::default(),
         }
     }
 
@@ -166,7 +214,7 @@ impl PySessionConfig {
         Self::from(self.config.clone().with_repartition_file_min_size(size))
     }
 
-    fn with_dataframe_display_config(&self, display_config: DataframeDisplayConfig) -> Self {
+    fn with_dataframe_display_config(&self, display_config: PyDataframeDisplayConfig) -> Self {
         let mut config = self.clone();
         config.display_config = display_config;
         config
@@ -181,55 +229,6 @@ impl PySessionConfig {
     }
 }
 
-#[pyclass(name = "DataframeDisplayConfig", module = "datafusion", subclass)]
-#[derive(Clone)]
-pub struct DataframeDisplayConfig {
-    /// Maximum bytes to display for table presentation (default: 2MB)
-    #[pyo3(get, set)]
-    pub max_table_bytes: usize,
-    /// Minimum number of table rows to display (default: 20)
-    #[pyo3(get, set)]
-    pub min_table_rows: usize,
-    /// Maximum length of a cell before it gets minimized (default: 25)
-    #[pyo3(get, set)]
-    pub max_cell_length: usize,
-    /// Maximum number of rows to display in repr string output (default: 10)
-    #[pyo3(get, set)]
-    pub max_table_rows_in_repr: usize,
-}
-
-#[pymethods]
-impl DataframeDisplayConfig {
-    #[new]
-    #[pyo3(signature = (max_table_bytes=None, min_table_rows=None, max_cell_length=None, max_table_rows_in_repr=None))]
-    fn new(
-        max_table_bytes: Option<usize>,
-        min_table_rows: Option<usize>,
-        max_cell_length: Option<usize>,
-        max_table_rows_in_repr: Option<usize>,
-    ) -> Self {
-        let default = Self::default();
-        Self {
-            max_table_bytes: max_table_bytes.unwrap_or(default.max_table_bytes),
-            min_table_rows: min_table_rows.unwrap_or(default.min_table_rows),
-            max_cell_length: max_cell_length.unwrap_or(default.max_cell_length),
-            max_table_rows_in_repr: max_table_rows_in_repr
-                .unwrap_or(default.max_table_rows_in_repr),
-        }
-    }
-}
-
-impl Default for DataframeDisplayConfig {
-    fn default() -> Self {
-        Self {
-            max_table_bytes: 2 * 1024 * 1024, // 2 MB
-            min_table_rows: 20,
-            max_cell_length: 25,
-            max_table_rows_in_repr: 10,
-        }
-    }
-}
-
 /// Runtime options for a SessionContext
 #[pyclass(name = "RuntimeEnvBuilder", module = "datafusion", subclass)]
 #[derive(Clone)]

src/lib.rs

@@ -82,7 +82,7 @@ fn _internal(py: Python, m: Bound<'_, PyModule>) -> PyResult<()> {
     m.add_class::<context::PyRuntimeEnvBuilder>()?;
     m.add_class::<context::PySessionConfig>()?;
     m.add_class::<context::PySessionContext>()?;
-    m.add_class::<context::DataframeDisplayConfig>()?;
+    m.add_class::<context::PyDataframeDisplayConfig>()?;
     m.add_class::<context::PySQLOptions>()?;
     m.add_class::<dataframe::PyDataFrame>()?;
     m.add_class::<udf::PyScalarUDF>()?;