feat: add into_view method to register DataFrame as a view

Author: kosiew

python/datafusion/context.py

@@ -636,7 +636,7 @@ def from_pylist(
 
     def from_pydict(
         self, data: dict[str, list[Any]], name: str | None = None
-    ) -> DataFrame:
+    ) -> DataFramee
         """Create a :py:class:`~datafusion.dataframe.DataFrame` from a dictionary.
 
         Args:

python/datafusion/dataframe.py

@@ -121,6 +121,12 @@ def __init__(self, df: DataFrameInternal) -> None:
         """
         self.df = df
 
+    def into_view(self) -> pa.Table:
+        """Register this DataFrame as a view in the current session.
+
+        """
+        return self.df.into_view()
+        
     def __getitem__(self, key: str | List[str]) -> DataFrame:
         """Return a new :py:class`DataFrame` with the specified column or columns.
 

python/tests/test_view.py

@@ -22,6 +22,7 @@ def test_register_filtered_dataframe():
 
     # Filter the DataFrame (for example, keep rows where a > 2)
     df_filtered = df.filter(col("a") > literal(2))
+    df_filtered = df_filtered.into_view()
 
     # Register the filtered DataFrame as a table called "view1"
     ctx.register_table("view1", df_filtered)

src/dataframe.rs

@@ -33,6 +33,7 @@ use datafusion::dataframe::{DataFrame, DataFrameWriteOptions};
 use datafusion::execution::SendableRecordBatchStream;
 use datafusion::parquet::basic::{BrotliLevel, Compression, GzipLevel, ZstdLevel};
 use datafusion::prelude::*;
+use datafusion::sql::sqlparser::ast::Table;
 use pyo3::exceptions::PyValueError;
 use pyo3::prelude::*;
 use pyo3::pybacked::PyBackedStr;