diff options
Diffstat (limited to 'prettier.py')
| -rw-r--r-- | prettier.py | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/prettier.py b/prettier.py index 9425f80..75ccd66 100644 --- a/prettier.py +++ b/prettier.py @@ -1,3 +1,4 @@ +import os import sys import argparse @@ -18,6 +19,7 @@ def parse_args(): help="decrease vebosity", action="store_true") parser.add_argument("-l", "--no-log", help="disable result log", action="store_true") + parser.add_argument("-c", "--no-clear", help="disable terminal clear before output") parser.add_argument("-f", "--output-file", help="output file name") return vars(parser.parse_args(sys.argv[1:])) @@ -51,9 +53,14 @@ def write_logs(logs, options): filename = options["output_file"] with open(filename, "w") as log_file: for ko in logs["ko_info"]: - log_file.write("- " + ko["msg"] + "\n") - log_file.write(" " + ko["expected"] + "\n") - log_file.write(" " + ko["actual"] + "\n\n") + try: + log_file.write("- " + ko["msg"]+ "\n") + log_file.write(" " + ko["expected"] + "\n") + log_file.write(" " + ko["actual"] + "\n") + except UnicodeEncodeError: + log_file.write("Can't write detail\n") + finally: + log_file.write("\n") def print_logs(logs, options): @@ -75,6 +82,8 @@ def print_logs(logs, options): if __name__ == "__main__": print() options = parse_args() + if not options["no_clear"]: + os.system("clear") logs = parse() print_logs(logs, options) if not options["no_log"]: |