Sadmin
f20080305d
Определение специфических рекомендаций (матрица профилей §8):
1. Ж/д слой (закрыт мёртвый railway ×2.5 у РАС):
- /api/v1/water/{case_id} отдаёт railway=rail как LineString
(без service/industrial/military веток), кэш общий v2;
- railway_warning «перекрыть/проверить немедленно» по профилям;
- SearchMap: Polyline слой ж/д (тёмно-красный), счётчики 💧/🚂.
2. cant_swim → профиль не_умеет_плавать (water ×3.0, без изменения
радиуса, critical_warning «обследовать водоёмы НЕМЕДЛЕННО»):
- раньше чекбокс влиял только на текст, в скоринге был пробел;
- derive в analyze._derive_profiles — работает и для closed_cases.
3. unmodeled_profiles: ДЦП/слабое зрение/слух — честная пометка
«вне поведенческой модели» с пояснением (vector_tasks B12:
профили без аналога не выдавать за учтённые); блок на фронте
в карточке здоровья.
Площадь воды: сферический эксцесс, проверен на квадрате 53° (744017 м²
vs 743272 точного). Тесты: 202 passed (новый test_cant_swim_profile).
74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Mapping
|
|
from typing import Any
|
|
|
|
from starlette._exception_handler import (
|
|
ExceptionHandlers,
|
|
StatusHandlers,
|
|
wrap_app_handling_exceptions,
|
|
)
|
|
from starlette.exceptions import HTTPException, WebSocketException
|
|
from starlette.requests import Request
|
|
from starlette.responses import PlainTextResponse, Response
|
|
from starlette.types import ASGIApp, ExceptionHandler, Receive, Scope, Send
|
|
from starlette.websockets import WebSocket
|
|
|
|
|
|
class ExceptionMiddleware:
|
|
def __init__(
|
|
self,
|
|
app: ASGIApp,
|
|
handlers: Mapping[Any, ExceptionHandler] | None = None,
|
|
debug: bool = False,
|
|
) -> None:
|
|
self.app = app
|
|
self.debug = debug # TODO: We ought to handle 404 cases if debug is set.
|
|
self._status_handlers: StatusHandlers = {}
|
|
self._exception_handlers: ExceptionHandlers = {
|
|
HTTPException: self.http_exception,
|
|
WebSocketException: self.websocket_exception,
|
|
}
|
|
if handlers is not None: # pragma: no branch
|
|
for key, value in handlers.items():
|
|
self.add_exception_handler(key, value)
|
|
|
|
def add_exception_handler(
|
|
self,
|
|
exc_class_or_status_code: int | type[Exception],
|
|
handler: ExceptionHandler,
|
|
) -> None:
|
|
if isinstance(exc_class_or_status_code, int):
|
|
self._status_handlers[exc_class_or_status_code] = handler
|
|
else:
|
|
assert issubclass(exc_class_or_status_code, Exception)
|
|
self._exception_handlers[exc_class_or_status_code] = handler
|
|
|
|
async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
|
|
if scope["type"] not in ("http", "websocket"):
|
|
await self.app(scope, receive, send)
|
|
return
|
|
|
|
scope["starlette.exception_handlers"] = (
|
|
self._exception_handlers,
|
|
self._status_handlers,
|
|
)
|
|
|
|
conn: Request | WebSocket
|
|
if scope["type"] == "http":
|
|
conn = Request(scope, receive, send)
|
|
else:
|
|
conn = WebSocket(scope, receive, send)
|
|
|
|
await wrap_app_handling_exceptions(self.app, conn)(scope, receive, send)
|
|
|
|
async def http_exception(self, request: Request, exc: Exception) -> Response:
|
|
assert isinstance(exc, HTTPException)
|
|
if exc.status_code in {204, 304}:
|
|
return Response(status_code=exc.status_code, headers=exc.headers)
|
|
return PlainTextResponse(exc.detail, status_code=exc.status_code, headers=exc.headers)
|
|
|
|
async def websocket_exception(self, websocket: WebSocket, exc: Exception) -> None:
|
|
assert isinstance(exc, WebSocketException)
|
|
await websocket.close(code=exc.code, reason=exc.reason) # pragma: no cover
|