a jS @sddlmZddlZddlZddlZddlmZm Z m Z ddl m Z ddl mZmZddlmZmZdd lmZmZmZgd ZGd d d Zdddd ddddddddddZddddddddZdS)) annotationsN)AnyCallableLiteral)NotFound)MapRequestRedirect)RequestResponse)ServerServerConnectionserve)route unix_routeRouterc@s|eZdZdZddddddd d Zd d dd ddZd dddddZd ddddZd d dd ddZd ddddZ dS)rz*WebSocket router supporting :func:`route`.Nwsr str | NonestrNone)url_map server_name url_schemereturncCs,||_||_||_|jD] }d|_qdS)NT)rrrZ iter_rulesZ websocket)selfrrrruler", endpoint=channel_handler), ... ]) with route(url_map, ...) as server: server.serve_forever() Refer to the documentation of :mod:`werkzeug.routing` for details. If you define redirects with ``Rule(..., redirect_to=...)`` in the URL map, when the server runs behind a reverse proxy that modifies the ``Host`` header or terminates TLS, you need additional configuration: * Set ``server_name`` to the name of the server as seen by clients. When not provided, websockets uses the value of the ``Host`` header. * Set ``ssl=True`` to generate ``wss://`` URIs without actually enabling TLS. Under the hood, this bind the URL map with a ``url_scheme`` of ``wss://`` instead of ``ws://``. There is no need to specify ``websocket=True`` in each rule. It is added automatically. Args: url_map: Mapping of URL patterns to connection handlers. server_name: Name of the server as seen by clients. If :obj:`None`, websockets uses the value of the ``Host`` header. ssl: Configuration for enabling TLS on the connection. Set it to :obj:`True` if a reverse proxy terminates TLS connections. create_router: Factory for the :class:`Router` dispatching requests to handlers. Set it to a wrapper or a subclass to customize routing. NrwssTrBprocess_requestrr r0r cs"||}|dur|S||S)N)r=)r!r"r+Z_process_requestZrouterrrrFs zroute..process_request)rpopr=rr9)rrrBrCrDr<rrFrrGrrIsE  r)rr5r<rcKst|fd|d|S)aK Create a WebSocket Unix server dispatching connections to different handlers. :func:`unix_route` combines the behaviors of :func:`route` and :func:`~websockets.sync.server.unix_serve`. Args: url_map: Mapping of URL patterns to connection handlers. path: File system path to the Unix socket. T)unixr5)r)rr5r<rrrrsr)N) __future__rr(rBZ ssl_module urllib.parser2typingrrrZwerkzeug.exceptionsrZwerkzeug.routingrrZhttp11r r serverr rr__all__rrrrrrrs  :i