From fff7e43da93bdf28cda4a0c28716e4c6dee0d8f3 Mon Sep 17 00:00:00 2001 From: Maksim Andrianov Date: Wed, 27 May 2020 18:35:41 +0300 Subject: [PATCH] [python][generator] Added check_maps.py --- tools/python/maps_generator/check_maps.py | 53 +++++++++++++++++++++++ 1 file changed, 53 insertions(+) create mode 100644 tools/python/maps_generator/check_maps.py diff --git a/tools/python/maps_generator/check_maps.py b/tools/python/maps_generator/check_maps.py new file mode 100644 index 0000000000..c873c5a16f --- /dev/null +++ b/tools/python/maps_generator/check_maps.py @@ -0,0 +1,53 @@ +import argparse +import sys + +from maps_generator.checks.default_check_set import CheckType +from maps_generator.checks.default_check_set import get_mwm_check_sets_and_filters +from maps_generator.checks.default_check_set import run_checks_and_print_results + + +def get_args(): + parser = argparse.ArgumentParser( + description="This script checks maps generation and prints results." + ) + parser.add_argument( + "--old", type=str, required=True, help="Path to old mwm directory.", + ) + parser.add_argument( + "--new", type=str, required=True, help="Path to new mwm directory.", + ) + parser.add_argument( + "--categories", type=str, required=True, help="Path to categories file.", + ) + + parser.add_argument( + "--level", + type=str, + required=False, + choices=("low", "medium", "hard", "strict"), + default="hard", + help="Messages level.", + ) + parser.add_argument( + "--output", + type=str, + required=False, + default="", + help="Path to output file. stdout by default.", + ) + return parser.parse_args() + + +def main(): + args = get_args() + + s = get_mwm_check_sets_and_filters(args.old, args.new, args.categories) + run_checks_and_print_results( + s, + CheckType[args.level], + file=open(args.output, "w") if args.output else sys.stdout, + ) + + +if __name__ == "__main__": + main()