Add __repr__ where missing

Author: timsaucer

python/datafusion/catalog.py

@@ -34,6 +34,10 @@ def __init__(self, catalog: df_internal.Catalog) -> None:
         """This constructor is not typically called by the end user."""
         self.catalog = catalog
 
+    def __repr__(self) -> str:
+        """Print a string representation of the catalog."""
+        self.catalog.__repr__()
+
     def names(self) -> list[str]:
         """Returns the list of databases in this catalog."""
         return self.catalog.names()
@@ -50,6 +54,10 @@ def __init__(self, db: df_internal.Database) -> None:
         """This constructor is not typically called by the end user."""
         self.db = db
 
+    def __repr__(self) -> str:
+        """Print a string representation of the database."""
+        self.db.__repr__()
+
     def names(self) -> set[str]:
         """Returns the list of all tables in this database."""
         return self.db.names()
@@ -66,6 +74,10 @@ def __init__(self, table: df_internal.Table) -> None:
         """This constructor is not typically called by the end user."""
         self.table = table
 
+    def __repr__(self) -> str:
+        """Print a string representation of the table."""
+        self.table.__repr__()
+
     @property
     def schema(self) -> pa.Schema:
         """Returns the schema associated with this table."""

python/datafusion/context.py

@@ -496,6 +496,10 @@ def __init__(
 
         self.ctx = SessionContextInternal(config, runtime)
 
+    def __repr__(self) -> str:
+        """Print a string representation of the Session Context."""
+        self.ctx.__repr__()
+
     @classmethod
     def global_ctx(cls) -> SessionContext:
         """Retrieve the global context as a `SessionContext` wrapper.

python/datafusion/expr.py

@@ -1172,6 +1172,10 @@ def __init__(
                 end_bound = end_bound.cast(pa.uint64())
         self.window_frame = expr_internal.WindowFrame(units, start_bound, end_bound)
 
+    def __repr__(self) -> str:
+        """Print a string representation of the window frame."""
+        self.window_frame.__repr__()
+
     def get_frame_units(self) -> str:
         """Returns the window frame units for the bounds."""
         return self.window_frame.get_frame_units()

python/datafusion/user_defined.py

@@ -102,6 +102,10 @@ def __init__(
             name, func, input_types, return_type, str(volatility)
         )
 
+    def __repr__(self) -> str:
+        """Print a string representation of the Scalar UDF."""
+        self._udf.__repr__()
+
     def __call__(self, *args: Expr) -> Expr:
         """Execute the UDF.
 
@@ -268,6 +272,10 @@ def __init__(
             str(volatility),
         )
 
+    def __repr__(self) -> str:
+        """Print a string representation of the Aggregate UDF."""
+        self._udaf.__repr__()
+
     def __call__(self, *args: Expr) -> Expr:
         """Execute the UDAF.
 
@@ -604,6 +612,10 @@ def __init__(
             name, func, input_types, return_type, str(volatility)
         )
 
+    def __repr__(self) -> str:
+        """Print a string representation of the Window UDF."""
+        self._udwf.__repr__()
+
     def __call__(self, *args: Expr) -> Expr:
         """Execute the UDWF.