Skip to content

Commit bddb940

Browse files
committed
🎨 Refactor socket server key definitions for consistency
1 parent 6547364 commit bddb940

File tree

2 files changed

+17
-13
lines changed

2 files changed

+17
-13
lines changed

services/web/server/src/simcore_service_webserver/socketio/_utils.py

Lines changed: 15 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -2,42 +2,46 @@
22
from collections.abc import Awaitable, Callable
33
from functools import wraps
44
from types import ModuleType
5-
from typing import Any, Final
5+
from typing import Any, Final, TypeAlias
66

77
from aiohttp import web
88
from socketio import AsyncServer # type: ignore[import-untyped]
99

10-
APP_CLIENT_SOCKET_DECORATED_HANDLERS_APPKEY: Final = web.AppKey(
10+
_CLIENT_SOCKET_DECORATED_HANDLERS_APPKEY: Final = web.AppKey(
1111
"APP_CLIENT_SOCKET_DECORATED_HANDLERS", list[Callable]
1212
)
13-
APP_CLIENT_SOCKET_SERVER_APPKEY: Final = web.AppKey(
13+
CLIENT_SOCKET_SERVER_APPKEY: Final = web.AppKey(
1414
# NOTE: AsyncServer stub library is missing
1515
"APP_CLIENT_SOCKET_SERVER",
1616
AsyncServer, # type: ignore[var-annotated]
1717
)
1818

1919

2020
def get_socket_server(app: web.Application) -> AsyncServer:
21-
return app[APP_CLIENT_SOCKET_SERVER_APPKEY]
21+
return app[CLIENT_SOCKET_SERVER_APPKEY]
2222

2323

2424
# The socket ID that was assigned to the client
25-
SocketID = str
25+
SocketID: TypeAlias = str
2626

2727
# The environ argument is a dictionary in standard WSGI format containing the request information, including HTTP headers
28-
EnvironDict = dict[str, Any]
28+
EnvironDict: TypeAlias = dict[str, Any]
2929

3030
# Connect event
31-
SocketioConnectEventHandler = Callable[
31+
SocketioConnectEventHandler: TypeAlias = Callable[
3232
[SocketID, EnvironDict, web.Application], Awaitable[None]
3333
]
3434

3535
# Disconnect event
36-
SocketioDisconnectEventHandler = Callable[[SocketID, web.Application], Awaitable[None]]
36+
SocketioDisconnectEventHandler: TypeAlias = Callable[
37+
[SocketID, web.Application], Awaitable[None]
38+
]
3739

3840
# Event
39-
AnyData = Any
40-
SocketioEventHandler = Callable[[SocketID, AnyData, web.Application], Awaitable[None]]
41+
AnyData: TypeAlias = Any
42+
SocketioEventHandler: TypeAlias = Callable[
43+
[SocketID, AnyData, web.Application], Awaitable[None]
44+
]
4145

4246
_socketio_handlers_registry: list[
4347
(
@@ -82,7 +86,7 @@ def register_socketio_handlers(app: web.Application, module: ModuleType):
8286
partial_fcts = [
8387
_socket_io_handler(app)(func_handler) for func_handler in member_fcts
8488
]
85-
app[APP_CLIENT_SOCKET_DECORATED_HANDLERS_APPKEY] = partial_fcts
89+
app[_CLIENT_SOCKET_DECORATED_HANDLERS_APPKEY] = partial_fcts
8690

8791
# register the fcts
8892
for func in partial_fcts:

services/web/server/src/simcore_service_webserver/socketio/server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from ..rabbitmq_settings import get_plugin_settings as get_rabbitmq_settings
1111
from . import _handlers
1212
from ._utils import (
13-
APP_CLIENT_SOCKET_SERVER_APPKEY,
13+
CLIENT_SOCKET_SERVER_APPKEY,
1414
get_socket_server,
1515
register_socketio_handlers,
1616
)
@@ -30,7 +30,7 @@ async def _socketio_server_cleanup_ctx(app: web.Application) -> AsyncIterator[No
3030
)
3131
sio_server.attach(app)
3232

33-
app[APP_CLIENT_SOCKET_SERVER_APPKEY] = sio_server
33+
app[CLIENT_SOCKET_SERVER_APPKEY] = sio_server
3434

3535
register_socketio_handlers(app, _handlers)
3636

0 commit comments

Comments
 (0)