Fix exit codes; add a simple test case

Author: Eclips4

Lib/test/test_pyrepl/test_pyrepl.py

@@ -1339,3 +1339,12 @@ def test_readline_history_file(self):
     def test_keyboard_interrupt_after_isearch(self):
         output, exit_code = self.run_repl(["\x12", "\x03", "exit"])
         self.assertEqual(exit_code, 0)
+
+    def test_exit_code(self):
+        _, exit_code = self.run_repl(["exit(128)"])
+        self.assertEqual(exit_code, 128)
+
+        env = os.environ.copy()
+        env["PYTHON_BASIC_REPL"] = "1"
+        _, exit_code = self.run_repl(["exit(64)"], env=env)
+        self.assertEqual(exit_code, 64)

Modules/main.c

@@ -569,10 +569,10 @@ pymain_run_stdin(PyConfig *config)
         || _Py_GetEnv(config->use_environment, "PYTHON_BASIC_REPL")) {
         PyCompilerFlags cf = _PyCompilerFlags_INIT;
         int run = PyRun_AnyFileExFlags(stdin, "<stdin>", 0, &cf);
-        return (run != 0);
+        return run;
     }
     int run = pymain_run_module(L"_pyrepl", 0);
-    return (run != 0);
+    return run;
 }