Add log level option to convert_data.py

scripts/convert_data.py

@@ -12,7 +12,7 @@ from probs_runner import probs_convert_data, Datasource, load_datasource
 
 # Import logging helpers
 from doit_utils import log_to_file, setup_logging_to_console
-setup_logging_to_console()
+import logging
 
 
 CODE_DIR = Path(__file__).parent
@@ -21,6 +21,9 @@ ONTOLOGY_DIR = Path(__file__).parent.resolve().parent / 'ontology'
 
 def parse_arguments():
     parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument("--log-level",
+                        help="log level (debug, info, warning, error)",
+                        default="warning")
     parser.add_argument("type", help="data type of file to convert")
     parser.add_argument('input_file',
                         type=Path,
@@ -71,4 +74,5 @@ def convert_data_prodcom(data_type, data_csv, output_path):
 
 if __name__ == "__main__":
     args = parse_arguments()
+    setup_logging_to_console(level=getattr(logging, args.log_level.upper()))
     convert_data_prodcom(args.type, args.input_file, args.output_file)