edit pool.disconnect to be a classmethod

This commit is contained in:
cloudwithax 2023-03-10 20:47:03 -05:00
parent 145634ce79
commit 9dc8a9098e
1 changed files with 7 additions and 3 deletions

View File

@ -723,7 +723,11 @@ class NodePool:
cls._nodes[node._identifier] = node
return node
async def disconnect(self) -> None:
"""Disconnects all nodes from the node pool."""
for node in self._nodes.copy().values():
@classmethod
async def disconnect(cls) -> None:
"""Disconnects all available nodes from the node pool."""
available_nodes: List[Node] = [node for node in cls._nodes.values() if node._available]
for node in available_nodes:
await node.disconnect()