Ensure the main is not present for --disable-main

certik · certik · commit 2d8145cf9480 · 2023-04-21T11:33:58.000-06:00
This fixes a regression introduced by lcompilers#1732.
diff --git a/src/lpython/semantics/python_ast_to_asr.cpp b/src/lpython/semantics/python_ast_to_asr.cpp
@@ -6731,6 +6731,16 @@ Result<ASR::TranslationUnit_t*> python_ast_to_asr(Allocator &al, LocationManager
                         throw LCompilersException("Verify failed");
                     };
         #endif
+        if (compiler_options.disable_main) {
+            // TODO: the main program should have not been created, but sometimes
+            // it is, so we explicitly remove it here.
+            for (auto &item : tu->m_global_scope->get_scope()) {
+                if (ASR::is_a<ASR::Program_t>(*item.second)) {
+                    tu->m_global_scope->erase_symbol(item.first);
+                    break;
+                }
+            }
+        }
     }
 
     return tu;
