64b0967292
Fix abstract renderer signatures and add small stubs so type checkers can
see expected attributes (e.g. username, _decrypt). This removes several
mypy false-positives that were caused by mixin/ABC mismatches.
Preserve message text containing ':' by using split(':', 1) in both
DefaultClientRenderer and RichClientRenderer.
Normalize renderer APIs: print_chat(...) now takes the response mapping
and returns None (matches runtime behavior).
Make RSA symmetric-key request more robust: read r.content instead of a
fixed-size r.raw.read(999), avoiding truncated key material.
Improve _connect_ws exception handling in client to ensure a valid
Exception is re-raised if connection attempts fail.
Correct server/service typing: memory_msgs is now typed as
list[Message] and we null-check incoming payload text before creating a
new Message.
Replace manual package list in setup.py with setuptools.find_packages()
so packaging uses valid Python package names.
Installed types-requests in the project venv so mypy no longer flags the
requests import.
Verification: ran python -m compileall and mypy cmd_chat — no issues
remain.
Notes:
Wire format still uses Python literal evaluation in some places (existing
behavior); switching to JSON for client/server payloads is recommended as a
follow-up for robustness and security.
37 lines
735 B
Python
37 lines
735 B
Python
import ast
|
|
from sanic import Websocket
|
|
from cmd_chat.server.models import Message
|
|
|
|
|
|
async def _get_bytes_and_serialize(
|
|
ws: Websocket
|
|
) -> dict:
|
|
return ast.literal_eval(await ws.recv())
|
|
|
|
|
|
async def _check_ws_for_close_status(
|
|
response: dict,
|
|
ws: Websocket
|
|
) -> None:
|
|
if "action" in response.keys():
|
|
if response["action"] == "close":
|
|
await ws.close()
|
|
|
|
|
|
async def _generate_new_message(
|
|
message: str
|
|
) -> Message:
|
|
return Message(message = message)
|
|
|
|
|
|
async def _generate_update_payload(
|
|
memory_msgs: list[Message],
|
|
users_structure: dict
|
|
) -> str:
|
|
return str({
|
|
"messages": [i.message for i in memory_msgs],
|
|
"users_in_chat": list(users_structure.keys())
|
|
})
|
|
|
|
|