Class AbstractTournament

codeclash/tournaments/tournament.py:12–57

class AbstractTournament:
    def __init__(self, config: dict, *, name: str, output_dir: Path, **kwargs):
        self.config: dict = config
        self.name: str = name
        self.tournament_id: str = f"{self.name}.{config['game']['name']}.{time.strftime('%y%m%d%H%M%S')}"
        self._output_dir: Path = output_dir
        self._metadata: dict = {
            "name": self.name,
            "tournament_id": self.tournament_id,
            "config": self.config,
            "created_timestamp": int(time.time()),
            "aws": get_aws_metadata(),
        }
        self.logger = get_logger(self.name, log_path=self.local_output_dir / "tournament.log", emoji="🏆")
        self._root_file_handler = add_root_file_handler(self.local_output_dir / "everything.log")

    @property
    def local_output_dir(self) -> Path:
        if "PYTEST_CURRENT_TEST" in os.environ:
            return Path("/tmp/codeclash") / self._output_dir.name
        return self._output_dir.resolve()

    def get_metadata(self) -> dict:
        return self._metadata

    def cleanup_handlers(self) -> None:
        """Close and remove file handlers to prevent file descriptor leaks."""
        # Close the root file handler
        remove_file_handler(logging.getLogger(), self._root_file_handler)

        # Close logger's file handlers
        for handler in self.logger.handlers[:]:  # Copy list to avoid modification during iteration
            if isinstance(handler, logging.FileHandler):
                self.logger.removeHandler(handler)
                handler.close()

    def _copy_game_log_to_agent(self, agent, round_num: int, log_output: str, dest_path: str = None) -> None:
        """Copy round log to agent environment."""
        try:
            create_file_in_container(
                container=agent.environment,
                content=log_output,
                dest_path=dest_path if dest_path else f"logs/round_{round_num}.log",
            )
        except Exception:
            self.logger.error(f"Error creating round log in {agent.name}'s container: {traceback.format_exc()}")