fix: make thing do work

ooliver1 · Oct 22, 2022 · 113b37a · 113b37a
1 parent 5aead6a
commit 113b37a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/mafic/pool.py b/mafic/pool.py
@@ -4,22 +4,24 @@
 
 from logging import getLogger
 from random import choice
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Generic, TypeVar
 
+from .__libraries import Client
 from .node import Node
 
 if TYPE_CHECKING:
     from typing import ClassVar
 
     from aiohttp import ClientSession
 
-    from .__libraries import Client
+
+ClientT = TypeVar("ClientT", bound=Client)
 
 
 _log = getLogger(__name__)
 
 
-class NodePool:
+class NodePool(Generic[ClientT]):
     __slots__ = ()
     _nodes: ClassVar[dict[str, Node]] = {}
 
@@ -35,7 +37,7 @@ async def create_node(
         port: int,
         label: str,
         password: str,
-        client: Client,
+        client: ClientT,
         secure: bool = False,
         heartbeat: int = 30,
         timeout: float = 10,

diff --git a/mafic/typings/http.py b/mafic/typings/http.py
@@ -2,11 +2,9 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, TypedDict, Union
+from typing import TYPE_CHECKING, Literal, TypedDict, Union
 
 if TYPE_CHECKING:
-    from typing import Literal
-
     from .misc import FriendlyException
 
 __all__ = (