nambrogi
/
pomice
mirror of https://github.com/cloudwithax/pomice.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore: rename to type adapter instead

This commit is contained in:
NiceAesth 2024-02-23 15:30:42 +02:00
parent ac49c81609
commit d486245428
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pomice/models/payloads.py
View File

@ -31,5 +31,5 @@ class ResumePayloadV4(BaseModel):
resuming: bool = True resuming: bool = True
_ResumePayloadType = Union[ResumePayloadV3, ResumePayloadV4] ResumePayloadType = Union[ResumePayloadV3, ResumePayloadV4]
ResumePayloadType = lambda **kwargs: TypeAdapter(_ResumePayloadType).validate_python(kwargs) ResumePayloadTypeAdapter = lambda **kwargs: TypeAdapter(ResumePayloadType).validate_python(kwargs)

8
pomice/pool.py
View File

@ -42,7 +42,7 @@ from .routeplanner import RoutePlanner
from .utils import ExponentialBackoff from .utils import ExponentialBackoff
from .utils import NodeStats from .utils import NodeStats
from .utils import Ping from .utils import Ping
from pomice.models.payloads import ResumePayloadType from pomice.models.payloads import ResumePayloadTypeAdapter
from pomice.models.payloads import ResumePayloadV4 from pomice.models.payloads import ResumePayloadV4
from pomice.models.version import LavalinkVersion from pomice.models.version import LavalinkVersion
@ -302,8 +302,10 @@ class Node:
if not self._resume_key: if not self._resume_key:
return return
data = ResumePayloadType( data = ResumePayloadTypeAdapter(
version=self._version, timeout=self._resume_timeout, resuming_key=self._resume_key, version=self._version,
timeout=self._resume_timeout,
resuming_key=self._resume_key,
).model_dump() ).model_dump()
if isinstance(data, ResumePayloadV4): if isinstance(data, ResumePayloadV4):