fix: fix bugs and refactor code

Author: Mukundan314

codeforces/cf_run.py

@@ -1,7 +1,9 @@
 #!/usr/bin/env python
 
 import argparse
+import shlex
 import subprocess
+import tempfile
 import time
 
 import colorama
@@ -10,27 +12,42 @@
 
 colorama.init(autoreset=True)
 
+
 def main(argv=None):
     parser = argparse.ArgumentParser()
 
-    parser.add_argument('contestId', type=int,
-                        help=("Id of the contest. It is not the round number. "
-                              "It can be seen in contest URL."))
-
-    parser.add_argument('index', type=str,
-                        help=("A letter or a letter followed by a digit, that "
-                              "represent a problem index in a contest."))
-
-    parser.add_argument('program', type=str,
-                        help="Path to executable that needs to be tested")
+    parser.add_argument(
+        'contestId',
+        type=int,
+        help=
+        "Id of the contest. It is not the round number. It can be seen in contest URL."
+    )
 
-    parser.add_argument('-t', '--timeout', type=int, default=10,
-                        help=("Timeout for program in seconds, -1 for no time "
-                              "limit (default: 10)"))
+    parser.add_argument(
+        'index',
+        type=str,
+        help=
+        "A letter or a letter followed by a digit, that represent a problem index in a contest."
+    )
 
-    parser.add_argument('-g', '--gym', action='store_true',
-                        help=("If true open gym contest instead of regular "
-                              "contest. (default: false)"))
+    parser.add_argument(
+        'program', type=str, help="Path to executable that needs to be tested")
+
+    parser.add_argument(
+        '-t',
+        '--timeout',
+        type=int,
+        default=10,
+        help=
+        "Timeout for program in seconds, -1 for no time limit (default: 10)")
+
+    parser.add_argument(
+        '-g',
+        '--gym',
+        action='store_true',
+        help=
+        "If true open gym contest instead of regular contest. (default: false)"
+    )
 
     if argv:
         args = parser.parse_args(argv)
@@ -40,8 +57,7 @@ def main(argv=None):
     args.timeout = None if args.timeout == -1 else args.timeout
 
     title, time_limit, memory_limit, sample_tests = problem.get_info(
-        args.contestId, args.index, gym=args.gym
-    )
+        args.contestId, args.index, gym=args.gym)
 
     print(title)
     print("time limit per test:", time_limit)
@@ -50,15 +66,22 @@ def main(argv=None):
     print()
 
     for inp, ans in sample_tests:
+        tmp = tempfile.TemporaryFile('w')
+        tmp.write(inp)
+        tmp.seek(0)
+
         start = time.time()
 
-        out = subprocess.run(
-            args.program,
-            shell=True,
-            input=inp.encode('utf-8'),
-            capture_output=True,
-            timeout=args.timeout
-        ).stdout.decode('utf-8')
+        proc = subprocess.run(
+            shlex.split(args.program),
+            stdin=tmp,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            timeout=args.timeout,
+            universal_newlines=True)
+
+        stdout = proc.stdout
+        stderr = proc.stderr
 
         time_used = time.time() - start
 
@@ -70,7 +93,11 @@ def main(argv=None):
         print(inp)
 
         print(colorama.Style.BRIGHT + "Participant's output")
-        print(out)
+        print(stdout)
+
+        if stderr:
+            print(colorama.Style.BRIGHT + "stderr")
+            print(stderr)
 
         print(colorama.Style.BRIGHT + "Jury's answer")
         print(ans)