jKddlmZddlZddlZddlZddlmZm Z m Z m Z ddl m Z mZddlmZddlmZmZmZgd Z dd lmZdd lmZmZdddd d"dZ d#d$dZn#e$rdddd d"dZ d#d$dZYnwxYwGd d!ZdS)%) annotationsN)Any AwaitableCallableLiteral)RequestResponse)PathLike)ServerServerConnectionserve)route unix_routeRouter)NotFound)MapRequestRedirect) server_namessl create_routerurl_maprargsrr str | Noner,ssl_module.SSLContext | Literal[True] | Nonertype[Router] | Nonekwargsreturnr c |dnd}|dur|||d<|t}|||| |dd j}nd fd }t jg|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 event to exit the server stop = asyncio.Event() async with route(url_map, ...) as server: await stop.wait() 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. Like :func:`~websockets.sync.server.serve`, :func:`route` returns a :class:`~websockets.sync.server.Server` that you can also run with :meth:`~websockets.sync.server.Server.serve_forever`. 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 connectionrrequestr rResponse | NonecK||}t|tr|d{V}||S||SN) isinstancer route_request)r$r%response_process_requestrouters IC:\Projects\Muse2API\.venv\Lib\site-packages\websockets/asyncio/router.pyr#zroute..process_requests_,+J@@h 22.%-~~~~~~H'#O++J@@@r$rr%r rr&)rpopr*rhandler) rrrrrr url_schemer#r,r-s @@r.rr)sV![TTe d??sF5M  "Mw Z@@ JJ($ / /   #$ O A A A A A A A N    ,     r/pathPathLike | Nonec "t|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)unixr4)rrr4rs r.rrs! W=4d==f===r/c td)Nzroute() requires werkzeug ImportError)rrrrrrs r.rrs5666r/c  td)Nzunix_route() requires werkzeugr:r8s r.rr s :;;;r/cHeZdZdZ ddd ZddZddZddZddZddZ dS) rz*WebSocket router supporting :func:`route`.Nr!rrrrr3strrNonecv||_||_||_|jD] }d|_ dS)NT)rrr3 iter_rules websocket)selfrrr3rules r.__init__zRouter.__init__sJ  &$L++-- " "D!DNN " "r/r$rr%r c8|j |jdS|jS)NHost)rheaders)rCr$r%s r.get_server_namezRouter.get_server_names!   #?6* *# #r/urlr cj|tjjd|}||jd<|S)Nz Found at Location)respondhttp HTTPStatusFOUNDrH)rCr$rJr+s r.redirectzRouter.redirects7%%do&;=N=N=NOO'*$r/cL|tjjdS)Nz Not Found)rMrNrO NOT_FOUNDrCr$s r. not_foundzRouter.not_founds!!$/";[IIIr/r&c|j||||j} tj|j}||j|j \}}nR#t$r%}| ||j cYd}~Sd}~wt$r||cYSwxYw||c|_|_dS)zRoute incoming request.)rr3) path_info query_argsN)rbindrIr3urllibparseurlparser4matchqueryrrQnew_urlrrUr2handler_kwargs)rCr$r%url_map_adapterparsedr2rrQs r.r*zRouter.route_requests,++,,ZAA,   .\**7<88F-33 +!<4OGVV ? ? ?==X-=>> > > > > > > . . .>>*-- - - - .8?5 J5ts$AB C B*$C*"CCc8K|j|fi|jd{VS)zHandle a connection.N)r2r`rTs r.r2zRouter.handlers6'Z' PPj6OPPPPPPPPPr/)Nr!)rrrrr3r>rr?)r$rr%r rr>)r$rrJr>rr )r$rrr r0)r$rrr?) __name__ __module__ __qualname____doc__rErIrQrUr*r2r/r.rrs44 #' " " " " "$$$$  JJJJ*QQQQQQr/r)rrrrrrrrrrrrrr r()rrr4r5rrrr ) __future__rrNr ssl_module urllib.parserZtypingrrrrhttp11r r r serverr rr__all__werkzeug.exceptionsrwerkzeug.routingrrrrr;rrhr/r.rrs"""""" 444444444444&&&&&&&&3333333333 , + +]>,,,,,,555555554#'<@-1 r r r r r r l!%>>>>>>S<<< #'<@-1 777777!%<<<<<<<