feat: add register_view method to SessionContext for DataFrame registration

Author: kosiew

python/datafusion/context.py

@@ -704,6 +704,18 @@ def from_polars(self, data: polars.DataFrame, name: str | None = None) -> DataFr
         """
         return DataFrame(self.ctx.from_polars(data, name))
 
+    # https://github.com/apache/datafusion-python/pull/1016#discussion_r1983239116
+    # is the discussion on how we arrived at adding register_view
+    def register_view(self, name: str, df: DataFrame):
+        """Register a :py:class: `~datafusion.detaframe.DataFrame` as a view.
+
+        Args:
+            name (str): The name to register the view under.
+            df (DataFrame): The DataFrame to be converted into a view and registered.
+        """
+        view = df.into_view()
+        self.ctx.register_table(name, view)
+
     def register_table(self, name: str, table: Table) -> None:
         """Register a :py:class: `~datafusion.catalog.Table` as a table.
 

python/tests/test_view.py

@@ -27,11 +27,8 @@ def test_register_filtered_dataframe():
     df = ctx.from_pydict(data, "my_table")
 
     df_filtered = df.filter(col("a") > literal(2))
-    view = df_filtered.into_view()
 
-    assert view.kind == "view"
-
-    ctx.register_table("view1", view)
+    ctx.register_view("view1", df_filtered)
 
     df_view = ctx.sql("SELECT * FROM view1")