Feature/colorful (#73)

* colorful messages in envgen

* Add colors to tester and submit

Author: kyuridenamida

atcodertools/tools/envgen.py

@@ -9,6 +9,8 @@
 from time import sleep
 from typing import Tuple, Optional
 
+from colorama import Fore
+
 from atcodertools.client.atcoder import AtCoderClient, Contest, LoginError
 from atcodertools.client.models.problem import Problem
 from atcodertools.client.models.problem_content import InputFormatDetectionError, SampleDetectionError
@@ -21,6 +23,7 @@
 from atcodertools.fmtprediction.predict_format import NoPredictionResultError, \
     MultiplePredictionResultsError, predict_format
 from atcodertools.tools.models.metadata import Metadata
+from atcodertools.tools.utils import with_color
 
 script_dir_path = os.path.dirname(os.path.abspath(__file__))
 
@@ -76,7 +79,7 @@ def prepare_procedure(atcoder_client: AtCoderClient,
         pid)
 
     def emit_error(text):
-        logging.error("Problem {}: {}".format(pid, text))
+        logging.error(with_color("Problem {}: {}".format(pid, text), Fore.RED))
 
     def emit_warning(text):
         logging.warning("Problem {}: {}".format(pid, text))
@@ -143,7 +146,9 @@ def emit_info(text):
             code_file_path
         )
         emit_info(
-            "Prediction succeeded -- Saved auto-generated code to '{}'".format(code_file_path))
+            "{} -- Saved auto-generated code to '{}'".format(
+                with_color("Prediction succeeded", Fore.LIGHTGREEN_EX),
+                code_file_path))
     except (NoPredictionResultError, MultiplePredictionResultsError) as e:
         if isinstance(e, NoPredictionResultError):
             msg = "No prediction -- Failed to understand the input format"
@@ -154,7 +159,7 @@ def emit_info(text):
         shutil.copy(replacement_code_path, code_file_path)
         emit_warning(
             "{} -- Copied {} to {}".format(
-                msg,
+                with_color(msg, Fore.LIGHTRED_EX),
                 replacement_code_path,
                 code_file_path))
 

atcodertools/tools/submit.py

@@ -4,6 +4,10 @@
 import sys
 import os
 
+from colorama import Fore
+
+from atcodertools.tools.utils import with_color
+
 from atcodertools.client.atcoder import AtCoderClient, LoginError
 from atcodertools.tools import tester
 
@@ -90,9 +94,9 @@ def main(prog, args, credential_supplier=None, use_local_session_cache=True) ->
         if not args.unlock_safety:
             for submission in submissions:
                 if submission.problem_id == metadata.problem.problem_id:
-                    logging.error("Cancel submitting because you already sent some code to the problem. Please "
-                                  "specify -u to send the code. {}".format(
-                                      metadata.problem.contest.get_submissions_url(submission)))
+                    logging.error(with_color("Cancel submitting because you already sent some code to the problem. Please "
+                                             "specify -u to send the code. {}".format(
+                                                 metadata.problem.contest.get_submissions_url(submission)), Fore.LIGHTRED_EX))
                     return False
 
         code_path = os.path.join(args.dir, metadata.code_filename)
@@ -102,7 +106,8 @@ def main(prog, args, credential_supplier=None, use_local_session_cache=True) ->
         logging.info("Submitting {} as {}".format(code_path, detailed_lang))
         submission = client.submit_source_code(
             metadata.problem.contest, metadata.problem, detailed_lang, source)
-        logging.info("Done! {}".format(
+        logging.info("{} {}".format(
+            with_color("Done!", Fore.LIGHTGREEN_EX),
             metadata.problem.contest.get_submissions_url(submission)))
 
 

atcodertools/tools/tester.py

@@ -10,7 +10,10 @@
 from pathlib import Path
 from typing import List, Tuple
 
+from colorama import Fore
+
 from atcodertools.tools.models.metadata import Metadata
+from atcodertools.tools.utils import with_color
 
 
 class NoExecutableFileError(Exception):
@@ -121,13 +124,15 @@ def run_for_samples(exec_file: str, sample_pair_list: List[Tuple[str, str]], tim
 
         is_correct = exec_res.is_correct_output(answer_text)
         if is_correct:
-            message = "PASSED {elapsed} ms".format(elapsed=exec_res.elapsed_ms)
+            message = "{} {elapsed} ms".format(
+                with_color("PASSED", Fore.LIGHTGREEN_EX),
+                elapsed=exec_res.elapsed_ms)
             success_count += 1
         else:
             if exec_res.status == ExecStatus.NORMAL:
-                message = "WA"
+                message = with_color("WA", Fore.LIGHTRED_EX)
             else:
-                message = exec_res.status.name
+                message = with_color(exec_res.status.name, Fore.LIGHTYELLOW_EX)
 
         print("# {case_name} ... {message}".format(
             case_name=os.path.basename(in_sample_file),
@@ -198,13 +203,14 @@ def run_all_tests(exec_file, in_sample_file_list, out_sample_file_list, timeout_
         print("No test cases")
         return False
     elif success_count != len(samples):
-        print("Some cases FAILED (passed {success_count} of {total})".format(
+        print("{msg} (passed {success_count} of {total})".format(
+            msg=with_color("Some cases FAILED", Fore.LIGHTRED_EX),
             success_count=success_count,
             total=len(samples),
         ))
         return False
     else:
-        print("Passed all test cases!!!")
+        print(with_color("Passed all test cases!!!", Fore.LIGHTGREEN_EX))
         return True
 
 

atcodertools/tools/utils.py

@@ -0,0 +1,5 @@
+from colorama import Fore
+
+
+def with_color(msg, color):
+    return "{}{}{}".format(color, msg, Fore.RESET)