add error checking for PyCapsule_Import call

PyO3 · Oct 19, 2024 · 8e02786 · 8e02786
1 parent 011364e
commit 8e02786
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pyo3-ffi/src/datetime.rs b/pyo3-ffi/src/datetime.rs
@@ -618,6 +618,10 @@ pub unsafe fn PyDateTime_IMPORT() {
         let py_datetime_c_api =
             PyCapsule_Import(PyDateTime_CAPSULE_NAME.as_ptr(), 1) as *mut PyDateTime_CAPI;
 
+        if py_datetime_c_api.is_null() {
+            return;
+        }
+
         // Protect against race conditions when the datetime API is concurrently
         // initialized in multiple threads. UnsafeCell.get() cannot panic so this
         // won't panic either.