diff --git a/utils/containers.py b/utils/containers.py index 27e990f3eb934fc4c418c409a5e21ecb382ed4a3..d598d445c484d2bff27b20b183f9238f18dee722 100644 --- a/utils/containers.py +++ b/utils/containers.py @@ -114,6 +114,14 @@ class ClusterConductor: for container in containers: if container and container_regex.match(container): self._dump_container_logs(dir, container) + def get_view(self) -> str: + return {"address": f"{self.ip}:{self.port}", "id": self.index} + + def get_nodes(self): + return self.nodes + + def get_node(self, index): + return self.nodes[index] def _dump_container_logs(self, dir, name: str) -> None: log_file = os.path.join(dir, f"{name}.log") @@ -503,7 +511,8 @@ class ClusterConductor: view_changed = True if view_changed and hasattr(self, "_parent"): self._parent.broadcast_view(self.get_full_view()) - + def get_node(self, index): + return self.nodes[index] def get_full_view(self): view = [] for node in self.nodes: