(args, task_num, log_level=INFO)
| 391 | # |
| 392 | |
| 393 | def init_logger(args, task_num, log_level=INFO): |
| 394 | filenameOutPrefixSeed = get_file_name(args, task_num) |
| 395 | logger = logging.getLogger() |
| 396 | formatter = logging.Formatter("[%(asctime)s][%(levelname)s\t] %(message)s", |
| 397 | datefmt='%Y-%m-%d %H:%M:%S') |
| 398 | logger.setLevel(log_level) |
| 399 | |
| 400 | ch = logging.StreamHandler() |
| 401 | ch.setLevel(log_level) |
| 402 | ch.setFormatter(formatter) |
| 403 | logger.addHandler(ch) |
| 404 | logging_dir = args["output_path"] |
| 405 | if logging_dir: |
| 406 | os.makedirs(logging_dir, exist_ok=True) |
| 407 | filename = f"{filenameOutPrefixSeed}.log" |
| 408 | fh = logging.FileHandler(filename) |
| 409 | fh.setLevel(log_level) |
| 410 | fh.setFormatter(formatter) |
| 411 | if logger.hasHandlers(): |
| 412 | logger.handlers.clear() |
| 413 | logger.addHandler(fh) |
| 414 | return logger |
| 415 | |
| 416 | def main(): |
| 417 | args = parse_args() |
no test coverage detected