def _is_node_in_replacement_valid()

in src/slurm_plugin/clustermgtd.py [0:0]

• 20 lines of code
• 6 McCabe index (conditional complexity)

    def _is_node_in_replacement_valid(self, node: SlurmNode, check_node_is_valid):
        """
        Check node is replacement timeout or in replacement.

        If check_node_is_valid=True, check whether a node is in replacement,
        If check_node_is_valid=False, check whether a node is replacement timeout.
        """
        log.debug(f"Checking if node is in replacement {node}")
        if (
            node.is_backing_instance_valid(
                self._config.ec2_instance_missing_max_count,
                self._nodes_without_backing_instance_count_map,
                log_warn_if_unhealthy=True,
            )
            and node.name in self._static_nodes_in_replacement
        ):
            # Set `time_is_expired` to `False` if `node.instance` is `None` since we don't have a launch time yet
            time_is_expired = (
                False
                if not node.instance
                else time_is_up(
                    node.instance.launch_time, self._current_time, grace_time=self._config.node_replacement_timeout
                )
            )
            log.debug(f"Node {node} is in replacement and timer expired? {time_is_expired}, instance? {node.instance}")
            return not time_is_expired if check_node_is_valid else time_is_expired
        return False