asdl: use _PyOnceFlag in Python-ast.c

colesbury · Apr 23, 2023 · 70856f1 · 70856f1
1 parent 212fef4
commit 70856f1
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/Include/internal/pycore_ast_state.h b/Include/internal/pycore_ast_state.h
diff --git a/Parser/asdl_c.py b/Parser/asdl_c.py
@@ -1467,6 +1467,7 @@ def visit(self, object):
 def generate_ast_state(module_state, f):
     f.write('struct ast_state {\n')
     f.write('    int initialized;\n')
+    f.write('    _PyOnceFlag once;\n')
     f.write('    int recursion_depth;\n')
     f.write('    int recursion_limit;\n')
     for s in module_state:
@@ -1544,8 +1545,12 @@ def generate_module_def(mod, metadata, f, internal_h):
         {
             PyInterpreterState *interp = _PyInterpreterState_GET();
             struct ast_state *state = &interp->ast;
-            if (!init_types(state)) {
-                return NULL;
+            if (_PyBeginOnce(&state->once)) {
+                if (!init_types(state)) {
+                    _PyEndOnceFailed(&state->once);
+                    return NULL;
+                }
+                _PyEndOnce(&state->once);
             }
             return state;
         }

diff --git a/Python/Python-ast.c b/Python/Python-ast.c