Function apply

tools/python/maps_generator/generator/stages.py:177–207

(obj: Stage, env: "Env", *args, **kwargs)

    def new_apply(method):
        def apply(obj: Stage, env: "Env", *args, **kwargs):
            name = get_stage_name(obj)
            logfile = os.path.join(env.paths.log_path, f"{name}.log")
            log_handler = create_file_handler(logfile)
            logger.addHandler(log_handler)
            # This message is used as an anchor for parsing logs.
            # See maps_generator/checks/logs/logs_reader.py STAGE_START_MSG_PATTERN
            logger.info(f"Stage {name}: start ...")
            t = time.time()
            try:
                if not env.is_accepted_stage(stage):
                    logger.info(f"Stage {name} was not accepted.")
                    return

                main_status = env.main_status
                main_status.init(env.paths.main_status_path, name)
                if main_status.need_skip():
                    logger.warning(f"Stage {name} was skipped.")
                    return

                main_status.update_status()
                env.set_subprocess_out(log_handler.stream)
                method(obj, env, *args, **kwargs)
            finally:
                d = time.time() - t
                # This message is used as an anchor for parsing logs.
                # See maps_generator/checks/logs/logs_reader.py STAGE_FINISH_MSG_PATTERN
                logger.info(
                    f"Stage {name}: finished in {str(datetime.timedelta(seconds=d))}"
                )
                logger.removeHandler(log_handler)

        return apply