o _i`@sddlmZddlZddlZddlZddlmZm Z m Z m Z ddl m Z mZddlmZmZmZgdZzdd lmZdd lmZmZWney^dddd d"ddZ d#d$ddZYnwdddd d"ddZ d#d$ddZGd d!d!ZdS)%) annotationsN)Any AwaitableCallableLiteral)RequestResponse)ServerServerConnectionserve)route unix_routeRouter)NotFound)MapRequestRedirect) server_namessl create_routerurl_maprargsrr str | Noner,ssl_module.SSLContext | Literal[True] | Nonertype[Router] | NonekwargsreturnAwaitable[Server]cOtd)Nzroute() requires werkzeug ImportError)rrrrrrr"[/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/websockets/asyncio/router.pyrsrpathcKr)Nzunix_route() requires werkzeugr rr$rr"r"r#rsrcs|durdnd}|dur|dur||d<|durt}|||||dddur.j}ndfd d }tjg|Rd|i|S)a Create a WebSocket server dispatching connections to different handlers. This feature requires the third-party library `werkzeug`_: .. code-block:: console $ pip install werkzeug .. _werkzeug: https://werkzeug.palletsprojects.com/ :func:`route` accepts the same arguments as :func:`~websockets.sync.server.serve`, except as described below. The first argument is a :class:`werkzeug.routing.Map` that maps URL patterns to connection handlers. In addition to the connection, handlers receive parameters captured in the URL as keyword arguments. Here's an example:: from websockets.asyncio.router import route from werkzeug.routing import Map, Rule async def channel_handler(websocket, channel_id): ... url_map = Map([ Rule("/channel/", endpoint=channel_handler), ... ]) # set this future to exit the server stop = asyncio.get_running_loop().create_future() async with route(url_map, ...) as server: await stop 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 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. NwswssTrprocess_request connectionr requestrrResponse | Nonecs8||}t|tr|IdH}|dur|S||SN) isinstancer route_request)r)r*response_process_requestrouterr"r#r(s    zroute..process_requestr)r r*rrr+)rpopr.r handler)rrrrrr url_schemer(r"r0r#r(sI   cKst|fd|d|S)aj Create a WebSocket Unix server dispatching connections to different handlers. :func:`unix_route` combines the behaviors of :func:`route` and :func:`~websockets.asyncio.server.unix_serve`. Args: url_map: Mapping of URL patterns to connection handlers. path: File system path to the Unix socket. T)unixr$)rr%r"r"r#rsc@sReZdZdZ  dd d d Zd!ddZd"ddZd#ddZd$ddZd%ddZ dS)&rz*WebSocket router supporting :func:`route`.Nr&rrrrr6strrNonecCs,||_||_||_|jD]}d|_qdS)NT)rrr6 iter_rules websocket)selfrrr6ruler"r"r#__init__s zRouter.__init__r)r r*rcCs|jdur |jdS|jS)NHost)rheaders)r<r)r*r"r"r#get_server_names  zRouter.get_server_nameurlr cCs$|tjjd|}||jd<|S)Nz Found at Location)respondhttp HTTPStatusFOUNDr@)r<r)rBr/r"r"r#redirects zRouter.redirectcCs|tjjdS)Nz Not Found)rDrErF NOT_FOUNDr<r)r"r"r# not_foundszRouter.not_foundr+c Cs|jj||||jd}ztj|j}|j|j|j d\}}Wn&t y:}z | ||j WYd}~Sd}~wt yG||YSw|||_|_dS)zRoute incoming request.)rr6) path_info query_argsN)rbindrAr6urllibparseurlparser$matchqueryrrHnew_urlrrKr5handler_kwargs)r<r)r*url_map_adapterparsedr5rrHr"r"r#r.s$  zRouter.route_requestcs|j|fi|jIdHS)zHandle a connection.N)r5rUrJr"r"r#r5szRouter.handler)Nr&)rrrrr6r8rr9)r)r r*rrr8)r)r rBr8rr )r)r rr r3)r)r rr9) __name__ __module__ __qualname____doc__r>rArHrKr.r5r"r"r"r#rs   r)rrrrrrrrrrrrrrr,)rrr$rrrrr) __future__rrEr ssl_module urllib.parserOtypingrrrrhttp11rr serverr r r __all__werkzeug.exceptionsrwerkzeug.routingrrr!rrrr"r"r"r#s4    n