Python bindings: Layer.WriteArrow(): add compatibility with DuckDB 1.…

…1 __arrow_c_stream__()
OSGeo · Sep 10, 2024 · 91ef2f1 · 91ef2f1
1 parent cf67174
commit 91ef2f1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/swig/include/python/ogr_python.i b/swig/include/python/ogr_python.i
@@ -451,7 +451,12 @@ def _WarnIfUserHasNotSpecifiedIfUsingExceptions():
             requested_schema = requested_schema.__arrow_c_schema__()
 
         if hasattr(obj, "__arrow_c_stream__"):
-            stream_capsule = obj.__arrow_c_stream__(requested_schema=requested_schema)
+            if requested_schema:
+                stream_capsule = obj.__arrow_c_stream__(requested_schema=requested_schema)
+            else:
+                # DuckDB 1.1 doesn't support the requested_schema argument at all
+                stream_capsule = obj.__arrow_c_stream__()
+
             return self.WriteArrowStreamCapsule(stream_capsule, createFieldsFromSchema, options)
 
         if hasattr(obj, "__arrow_c_array__"):