| 347 | |
| 348 | |
| 349 | def prepare(args: argparse.Namespace) -> None: |
| 350 | # prepare result dir |
| 351 | result_dir = args.result_dir |
| 352 | if not result_dir: |
| 353 | result_dir = f"cache/results_{time.strftime('%Y%m%d%H%M%S', time.localtime())}" |
| 354 | if not Path(result_dir).exists(): |
| 355 | Path(result_dir).mkdir(parents=True, exist_ok=True) |
| 356 | args.result_dir = result_dir |
| 357 | logger.info(f"Create result dir: {result_dir}") |
| 358 | |
| 359 | if not (Path(result_dir) / "traces").exists(): |
| 360 | (Path(result_dir) / "traces").mkdir(parents=True) |
| 361 | |
| 362 | # log the log file |
| 363 | with open(os.path.join(result_dir, "log_files.txt"), "a+") as f: |
| 364 | f.write(f"{LOG_FILE_NAME}\n") |
| 365 | |
| 366 | |
| 367 | @beartype |