satori-python-server 0.14.0__tar.gz → 0.14.2__tar.gz

{satori_python_server-0.14.0 → satori_python_server-0.14.2}/PKG-INFO

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

{satori_python_server-0.14.0 → satori_python_server-0.14.2}/pyproject.toml

@@ -27,7 +27,7 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
     "Operating System :: OS Independent",
 ]
-version = "0.14.0"
+version = "0.14.2"
 
 [project.license]
 text = "MIT"

{satori_python_server-0.14.0 → satori_python_server-0.14.2}/src/satori/server/__init__.py

@@ -20,7 +20,7 @@ from graia.amnesia.builtins.asgi import UvicornASGIService
 from launart import Launart, Service, any_completed
 from loguru import logger
 from starlette.applications import Starlette
-from starlette.datastructures import FormData
+from starlette.datastructures import FormData as FormData
 from starlette.requests import Request as StarletteRequest
 from starlette.responses import JSONResponse, Response
 from starlette.routing import Route, WebSocketRoute
@@ -34,7 +34,7 @@ from satori.model import Event, ModelBase, Opcode
 from .adapter import Adapter as Adapter
 from .conection import WebsocketConnection
 from .deque import Deque
-from .formdata import parse_content_disposition
+from .formdata import parse_content_disposition as parse_content_disposition
 from .model import Provider as Provider
 from .model import Request as Request
 from .model import Router as Router
@@ -94,6 +94,7 @@ class Server(Service, RouterMixin):
         self.path = path
         if self.path and not self.path.startswith("/"):
             self.path = f"/{self.path}"
+        self._url = f"http://{host}:{port}{self.path}/{version}"
         self._adapters = []
         self.providers = []
         self.routers = []
@@ -107,14 +108,13 @@ class Server(Service, RouterMixin):
 
     def apply(self, item: Provider | Router | Adapter):
         if isinstance(item, Adapter):
+            item.ensure_net(self._url)
             self._adapters.append(item)
             self.providers.append(item)
-            for proxy_url_pf in item.proxy_urls():
-                self.proxy_url_mapping[proxy_url_pf] = item
+            self.proxy_url_mapping[item.id] = item.proxy_urls()
         elif isinstance(item, Provider):
             self.providers.append(item)
-            for proxy_url_pf in item.proxy_urls():
-                self.proxy_url_mapping[proxy_url_pf] = item
+            self.proxy_url_mapping[item.id] = item.proxy_urls()
         elif isinstance(item, Router):
             self.routers.append(item)
         else:
@@ -160,7 +160,9 @@ class Server(Service, RouterMixin):
             if not provider.authenticate(token):
                 return await ws.close(code=3000, reason="Unauthorized")
             logins.extend(await provider.get_logins())
-        sequence = body.get("sequence", -1)
+        sequence = body.get("sequence")
+        if sequence is None:
+            sequence = -1
         await connection.send({"op": Opcode.READY, "body": {"logins": [lo.dump() for lo in logins]}})
         self.connections.append(connection)
         logger.debug(f"New connection: {id(connection)}")
@@ -230,10 +232,11 @@ class Server(Service, RouterMixin):
             for provider in self.providers:
                 if provider.ensure(platform, self_id):
                     return await provider.download_uploaded(platform, self_id, path)
-        for proxy_url_pf, provider in self.proxy_url_mapping.items():
-            if url.startswith(proxy_url_pf):
-                async with self.session.get(url) as resp:
-                    return await resp.read()
+        for provider in self.providers:
+            for proxy_url_pf in self.proxy_url_mapping[provider.id]:
+                if url.startswith(proxy_url_pf):
+                    async with self.session.get(url) as resp:
+                        return await resp.read()
         raise ValueError(f"Unknown proxy url: {url}")
 
     def get_local_file(self, url: str):

{satori_python_server-0.14.0 → satori_python_server-0.14.2}/src/satori/server/adapter.py

@@ -9,6 +9,8 @@ from .route import RouterMixin
 
 
 class Adapter(Service, RouterMixin):
+    server_url: str
+
     @abstractmethod
     def get_platform(self) -> str: ...
 
@@ -38,3 +40,6 @@ class Adapter(Service, RouterMixin):
     @property
     def id(self):
         return f"satori-python.adapter.{self.get_platform()}#{id(self)}"
+
+    def ensure_net(self, url: str):
+        self.server_url = url

{satori_python_server-0.14.0 → satori_python_server-0.14.2}/src/satori/server/model.py

@@ -22,6 +22,9 @@ class Request(Generic[TP]):
 
 @runtime_checkable
 class Provider(Protocol):
+    @property
+    def id(self) -> str: ...
+
     def publisher(self) -> AsyncIterator[Event]: ...
 
     def authenticate(self, token: Optional[str]) -> bool: ...