add os._exit()

Author: glaukiol1

main

@@ -21,6 +21,7 @@ func init() {
 		py.MustNewMethod("getpid", getpid, 0, "Return the current process id."),
 		py.MustNewMethod("putenv", putenv, 0, "Set the environment variable named key to the string value."),
 		py.MustNewMethod("unsetenv", unsetenv, 0, "Unset (delete) the environment variable named key."),
+		py.MustNewMethod("_exit", _exit, 0, "Immediate program termination."),
 	}
 
 	globals := py.StringDict{
@@ -169,3 +170,25 @@ func unsetenv(self py.Object, args py.Tuple) (py.Object, error) {
 		return nil, py.ExceptionNewf(py.TypeError, "Expected 1 argument of type string")
 	}
 }
+
+// os._exit() immediate program termination; unline sys.exit(), which raises a SystemExit, this function will termninate the program immediately.
+func _exit(self py.Object, args py.Tuple) (py.Object, error) { // can never return
+	if len(args) == 0 {
+		os.Exit(0)
+		return nil, nil
+	} else if len(args) == 1 {
+		_ec, err := py.GetInt(args[0])
+		if err != nil {
+			os.Exit(1)
+		}
+		exit_code, err := _ec.GoInt()
+		if err != nil {
+			os.Exit(1)
+		}
+		os.Exit(exit_code)
+		return nil, nil
+	} else {
+		os.Exit(1)
+		return nil, nil
+	}
+}

os/os.module.go

@@ -109,4 +109,9 @@ def _fail(test_name):
 print()
 print(bcolors.ENDC+bcolors.BOLD+bcolors.HEADER+"----- Results -----")
 print(bcolors.OKGREEN+bcolors.BOLD, "PASSED: "+str(passed_tests)+" out of "+str(passed_tests+failed))
-print(bcolors.FAIL+bcolors.BOLD, "FAILED: "+str(failed)+" out of "+str(passed_tests+failed))
+print(bcolors.FAIL+bcolors.BOLD, "FAILED: "+str(failed)+" out of "+str(passed_tests+failed))
+print()
+print(bcolors.ENDC+bcolors.HEADER+bcolors.BOLD+"Running an extra test... | os._exit(0)")
+print(bcolors.OKCYAN+"If the program quits without any further output, the test is a pass...")
+os._exit(0)
+print(bcolors.ENDC+bcolors.BOLD+bcolors.FAIL+"Test failed!")