Forward the NsJail exit code from the entrypoint

1 files changed, 4 insertions, 0 deletions

diff --git a/snekbox/__main__.py b/snekbox/__main__.py
index f92a48c..1a33587 100644
--- a/snekbox/__main__.py
+++ b/snekbox/__main__.py
@@ -1,4 +1,5 @@
 import argparse
+import sys
 
 from snekbox.nsjail import NsJail
 
@@ -39,6 +40,9 @@ def main() -> None:
     result = NsJail().python3(args.code, nsjail_args=args.nsjail_args, py_args=args.py_args)
     print(result.stdout)
 
+    if result.returncode != 0:
+        sys.exit(result.returncode)
+
 
 if __name__ == "__main__":
     main()