fix: improve error handling and streamline capsule registration process

Author: kosiew

examples/raw_capsule_registration_failure.py

@@ -36,15 +36,14 @@ def main() -> None:
     """Attempt to use the capsule the same way existing callers do."""
 
     ctx = SessionContext()
-    capsule = make_table_provider_capsule()
-
     try:
-        Table(capsule)
-    except ValueError as err:
-        print("Constructing Table(capsule) failed:", err)
+        capsule = make_table_provider_capsule()
+    except Exception as err:
+        print("Creating the PyCapsule failed:", err)
+        return
 
     try:
-        ctx.register_table("capsule", capsule)
+        ctx.read_table(capsule)
     except ValueError as err:
         print("Registering capsule with SessionContext failed:", err)
 

examples/table_capsule_failure.py

@@ -12,7 +12,6 @@
 
 from datafusion import SessionContext, Table, udtf
 
-
 def main() -> None:
     """Register a Python table UDTF that returns a ``Table`` and trigger it."""