satori-python-client 0.13.2__tar.gz → 0.14.0__tar.gz

{satori_python_client-0.13.2 → satori_python_client-0.14.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: satori-python-client
-Version: 0.13.2
+Version: 0.14.0
 Summary: Satori Protocol SDK for python, specify client part
 Home-page: https://github.com/RF-Tar-Railt/satori-python
 Author-Email: RF-Tar-Railt <rf_tar_railt@qq.com>

{satori_python_client-0.13.2 → satori_python_client-0.14.0}/pyproject.toml

@@ -24,7 +24,7 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
     "Operating System :: OS Independent",
 ]
-version = "0.13.2"
+version = "0.14.0"
 
 [project.license]
 text = "MIT"
@@ -82,7 +82,7 @@ extra_standard_library = [
 
 [tool.ruff]
 line-length = 110
-target-version = "py38"
+target-version = "py39"
 exclude = [
     "exam.py",
 ]
@@ -107,7 +107,7 @@ ignore = [
 
 [tool.pyright]
 pythonPlatform = "All"
-pythonVersion = "3.8"
+pythonVersion = "3.9"
 typeCheckingMode = "basic"
 reportShadowedImports = false
 disableBytesTypePromotions = true

{satori_python_client-0.13.2 → satori_python_client-0.14.0}/src/satori/client/__init__.py

@@ -4,8 +4,9 @@ import asyncio
 import functools
 import signal
 import threading
+from collections.abc import Awaitable, Iterable
 from functools import wraps
-from typing import TYPE_CHECKING, Any, Awaitable, Callable, Iterable, Literal, TypeVar, overload
+from typing import TYPE_CHECKING, Any, Callable, Literal, TypeVar, overload
 
 from creart import it
 from graia.amnesia.builtins.aiohttp import AiohttpClientService

{satori_python_client-0.13.2 → satori_python_client-0.14.0}/src/satori/client/account.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import asyncio
 from dataclasses import dataclass
-from typing import TypeVar
+from typing import Generic, TypeVar
 
 from yarl import URL
 
@@ -11,6 +11,7 @@ from satori.model import Login
 from .protocol import ApiProtocol
 
 TP = TypeVar("TP", bound="ApiProtocol")
+TP1 = TypeVar("TP1", bound="ApiProtocol")
 
 
 @dataclass
@@ -29,14 +30,14 @@ class ApiInfo:
         return URL(f"http://{self.host}:{self.port}{self.path}") / "v1"
 
 
-class Account:
+class Account(Generic[TP]):
     def __init__(
         self,
         platform: str,
         self_id: str,
         self_info: Login,
         config: ApiInfo,
-        protocol_cls: type[ApiProtocol] = ApiProtocol,
+        protocol_cls: type[TP] = ApiProtocol,
     ):
         self.platform = platform
         self.self_id = self_id
@@ -45,14 +46,16 @@ class Account:
         self.protocol = protocol_cls(self)  # type: ignore
         self.connected = asyncio.Event()
 
-    def custom(self, config: ApiInfo | None = None, protocol_cls: type[TP] = ApiProtocol, **kwargs) -> TP:
+    def custom(
+        self, config: ApiInfo | None = None, protocol_cls: type[TP1] = ApiProtocol, **kwargs
+    ) -> "Account[TP1]":
         return Account(
             self.platform,
             self.self_id,
             self.self_info,
-            config or ApiInfo(**kwargs),
+            config or (ApiInfo(**kwargs) if kwargs else self.config),
             protocol_cls,  # type: ignore
-        ).protocol
+        )
 
     @property
     def identity(self):

{satori_python_client-0.13.2 → satori_python_client-0.14.0}/src/satori/client/account.pyi

@@ -1,5 +1,6 @@
 import asyncio
-from typing import Any, Iterable, Protocol, TypeVar, overload
+from collections.abc import Iterable
+from typing import Any, Generic, Protocol, TypeVar, overload
 
 from yarl import URL
 
@@ -23,6 +24,7 @@ from satori.model import (
 from .protocol import ApiProtocol
 
 TP = TypeVar("TP", bound="ApiProtocol")
+TP1 = TypeVar("TP1", bound="ApiProtocol")
 
 class Api(Protocol):
     token: str | None = None
@@ -35,12 +37,12 @@ class ApiInfo(Api):
         self, host: str = "localhost", port: int = 5140, path: str = "", token: str | None = None
     ): ...
 
-class Account:
+class Account(Generic[TP]):
     platform: str
     self_id: str
     self_info: Login
     config: Api
-    protocol: ApiProtocol
+    protocol: TP
     connected: asyncio.Event
 
     def __init__(
@@ -49,16 +51,18 @@ class Account:
         self_id: str,
         self_info: Login,
         config: Api,
-        protocol_cls: type[ApiProtocol] = ApiProtocol,
+        protocol_cls: type[TP] = ApiProtocol,
     ): ...
     @property
     def identity(self) -> str: ...
     @overload
-    def custom(self, config: Api, protocol_cls: type[TP] = ApiProtocol) -> TP: ...
+    def custom(self, config: Api, protocol_cls: type[TP1] = ApiProtocol) -> Account[TP1]: ...
+    @overload
+    def custom(self, *, protocol_cls: type[TP1]) -> Account[TP1]: ...
     @overload
     def custom(
-        self, *, protocol_cls: type[TP] = ApiProtocol, host: str, port: int, token: str | None = None
-    ) -> TP: ...
+        self, *, protocol_cls: type[TP1] = ApiProtocol, host: str, port: int, token: str | None = None
+    ) -> Account[TP1]: ...
     async def send(self, event: Event, message: str | Iterable[str | Element]) -> list[MessageObject]:
         """发送消息。返回一个 `MessageObject` 对象构成的数组。
 

{satori_python_client-0.13.2 → satori_python_client-0.14.0}/src/satori/client/network/webhook.py

@@ -61,7 +61,7 @@ class WebhookNetwork(BaseNetwork[WebhookInfo]):
         op = data["op"]
         if op != Opcode.EVENT:
             return web.Response(status=202)
-        logger.debug(f"Received payload: {data}")
+        logger.trace(f"Received payload: {data}")
         self.post_event(data["body"])
         return web.Response()
 

{satori_python_client-0.13.2 → satori_python_client-0.14.0}/src/satori/client/network/websocket.py

@@ -37,6 +37,7 @@ class WsNetwork(BaseNetwork[WebsocketsInfo]):
                 return
             elif msg.type == aiohttp.WSMsgType.TEXT:
                 data: dict = json.loads(cast(str, msg.data))
+                logger.trace(f"Received payload: {data}")
                 if data["op"] == Opcode.EVENT:
                     self.post_event(data["body"])
                 elif data["op"] > 4:
@@ -118,9 +119,8 @@ class WsNetwork(BaseNetwork[WebsocketsInfo]):
     async def _heartbeat(self):
         """心跳"""
         while True:
-            if self.sequence:
-                with suppress(Exception):
-                    await self.send({"op": 1})
+            with suppress(Exception):
+                await self.send({"op": 1})
             await asyncio.sleep(9)
 
     async def daemon(self, manager: Launart, session: aiohttp.ClientSession):

{satori_python_client-0.13.2 → satori_python_client-0.14.0}/src/satori/client/protocol.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Iterable, cast, overload
+from collections.abc import Iterable
+from typing import TYPE_CHECKING, Any, cast, overload
 
 from aiohttp import FormData
 from graia.amnesia.builtins.aiohttp import AiohttpClientService