From 9dc8a9098e4d3b61b156bdf440c1672949def9c7 Mon Sep 17 00:00:00 2001 From: cloudwithax Date: Fri, 10 Mar 2023 20:47:03 -0500 Subject: [PATCH] edit pool.disconnect to be a classmethod --- pomice/pool.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/pomice/pool.py b/pomice/pool.py index cabb5e2..6b5a6a1 100644 --- a/pomice/pool.py +++ b/pomice/pool.py @@ -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() \ No newline at end of file