Check the import of datetime's capi succeeded or not in zoneinfo.

pganssle · May 25, 2021 · c54d700 · c54d700
1 parent 959cd91
commit c54d700
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/zoneinfo_module.c b/lib/zoneinfo_module.c
@@ -2647,6 +2647,9 @@ static int
 zoneinfomodule_exec(PyObject *m)
 {
     PyDateTime_IMPORT;
+    if (PyDateTimeAPI == NULL) {
+        goto error;
+    }
     PyZoneInfo_ZoneInfoType.tp_base = PyDateTimeAPI->TZInfoType;
     if (PyType_Ready(&PyZoneInfo_ZoneInfoType) < 0) {
         goto error;