o ii6s@s<ddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZddlmZmZmZddlmZddlmZddlmZmZddlmZdd lmZmZdd lmZdd lm Z m!Z!dd l"m#Z#m$Z$m%Z%dd l&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-m.Z.m/Z/ddl0m1Z1m2Z2m3Z3m4Z4m5Z5ddl6m7Z7m8Z8Gddde9Z:GdddeZ;dGddZdJd+d,Z?e@d-ZAdKd/d0ZBGd1d2d2ZCGd3d4d4eCZDGd5d6d6eCZEGd7d8d8eCZFGd9d:d:eCZGed;ZHGd response`, and returns an ASGI application. scoper!receiver sendr"r*r+cs:t|||d fd d }t||||IdHdS) NrEr!rFr rGr"r*r+cs&IdH}||||IdHdSNr6)rErFrGresponse)frequestr6r7app<sz*request_response..app..apprEr!rFr rGr"r*r+)rrrErFrGrLrJ)rKr7rL9s zrequest_response..appNrM)r functoolspartialrrCrLr6rOr7request_response.s rS&Callable[[WebSocket], Awaitable[None]]csd fd d }|S) zM Takes a coroutine `func(session)`, and returns an ASGI application. rEr!rFr rGr"r*r+cs<t|||dd fd d }t||||IdHdS) N)rFrGrEr!rFr rGr"r*r+csIdHdSrHr6)rErFrG)rCsessionr6r7rLPsz+websocket_session..app..apprM)r#rrNrC)rUr7rLMszwebsocket_session..appNrMr6rRr6rVr7websocket_sessionEsrWendpointCallable[..., Any]r'cCst|d|jjS)Nr8)getattrr5r8)rXr6r6r7get_nameXsr[pathparam_convertorsdict[str, Convertor[Any]]r(dict[str, str]tuple[str, dict[str, str]]cCs^t|D]$\}}d|d|vr*||}||}|d|d|}||q||fS)N{})r.items to_stringreplacepop)r\r]r(keyvalue convertorr6r6r7replace_params\s  rjz4{([a-zA-Z_][a-zA-Z0-9_]*)(:[a-zA-Z_][a-zA-Z0-9_]*)?}3tuple[Pattern[str], str, dict[str, Convertor[Any]]]cCs|d }d}d}t}d}i}t|D]X}|d\}} | d} | tvs0Jd| dt| } |t||| 7}|d |d | j d 7}|||| 7}|d |7}||vrf| || ||<| }q|rd t|} t|dkrdnd} td| d| d||r||ddd} |t| d7}n |t||dd7}|||d7}t|||fS)a* Given a path string, like: "/{username:str}", or a host string, like: "{subdomain}.mydomain.org", return a three-tuple of (regex, format, {param_name:convertor}). regex: "/(?P[^/]+)" format: "/{username}" convertors: {"username": StringConvertor()} /^rr':zUnknown path convertor ''z(?P<>)z{%s}r,r>szDuplicated param name z at path N$) startswithset PARAM_REGEXfinditergroupslstriprreescapestartregexaddendr-sortedlen ValueErrorsplitcompile)r\is_host path_regex path_formatduplicated_paramsidxr]match param_nameconvertor_typerinamesendinghostnamer6r6r7 compile_pathns:     rc@s4eZdZdddZdd d ZdddZdddZdS) BaseRouterEr!r*tuple[Match, Scope]cCtrHNotImplementedError)r2rEr6r6r7matcheszBaseRoute.matchesr&r'r(r rcKrrHr)r2r&r(r6r6r7 url_path_forrzBaseRoute.url_path_forrFr rGr"r+cstrHrr2rErFrGr6r6r7handleszBaseRoute.handlecs||\}}|tjkr8|ddkr$tddd}||||IdHdS|ddkr6t}||||IdHdS||||||IdHdS)z A route may be used in isolation as a stand-alone ASGI app. This is a somewhat contrived case, as they'll almost always be used within a Router, but could be useful for some tooling and minimal apps. typehttp Not Found status_codeN websocket)rr=r@rr$updater)r2rErFrGr child_scoperIwebsocket_closer6r6r7__call__s     zBaseRoute.__call__NrEr!r*rr&r'r(r r*rrM)r8r9r:rrrrr6r6r6r7r    rc@sTeZdZdddddd)ddZd*ddZd+ddZd,d"d#Zd-d%d&Zd.d'd(ZdS)/RouteNT)methodsr&include_in_schema middlewarer\r'rXrYrCollection[str] | Noner& str | NonerboolrSequence[Middleware] | Noner*r+c Cs|ds Jd||_||_|durt|n||_||_|}t|tjr.|j }t|tjs%t |s8t |rEt ||_|durDdg}n||_|durct|D]\}} } ||jg| Ri| |_qP|durkd|_ndd|D|_d|jvr~|jdt|\|_|_|_dS)Nrl Routed paths must start with '/'GETcSsh|]}|qSr6)upper).0methodr6r6r7 sz!Route.__init__..HEAD)rvr\rXr[r&r isinstancerPrQrCinspect isfunctionismethodrSrLreversedrrrrrr]) r2r\rXrr&rrendpoint_handlerclsargskwargsr6r6r7r1s0      zRoute.__init__rEr!rc Cs|ddkrPt|}|j|}|rP|}|D]\}}|j||||<qt|di}| ||j |d}|j rK|d|j vrKt j |fSt j|fSt jifS)Nrrr(rXr(r)rrr groupdictrcr]convertdictgetrrXrr=rArBr@ r2rE route_pathrmatched_paramsrgrhr(rr6r6r7rs       z Route.matchesr(r rcK^t|}t|j}||jks||krt||t|j|j|\}}|r)Jt|ddS)Nrr\protocolrwr/r]r&r%rjrrr2r&r( seen_paramsexpected_paramsr\remaining_paramsr6r6r7r   zRoute.url_path_forrFr rGr"csv|jr/|d|jvr/dd|ji}d|vrtd|dtdd|d}||||IdHdS||||IdHdS)NrAllowr,rLi)rheaderszMethod Not Allowed)rr-rrrL)r2rErFrGrrIr6r6r7r s z Route.handleothercCs.t|to|j|jko|j|jko|j|jkSrH)rrr\rXrr2rr6r6r7__eq__s    z Route.__eq__cCs@|jj}t|jp g}|j|j}}|d|d|d|dS)N(path=, name=z , methods=rr)r5r8rrr\r&)r2 class_namerr\r&r6r6r7__repr__szRoute.__repr__)r\r'rXrYrrr&rrrrrr*r+rrrMrr r*rr*r' r8r9r:r1rrrrrr6r6r6r7rs )  rc@sPeZdZdddd%d dZd&ddZd'ddZd(ddZd)d!d"Zd*d#d$ZdS)+WebSocketRouteN)r&rr\r'rXrYr&rrrr*r+c Cs|ds Jd||_||_|durt|n||_|}t|tjr+|j}t|tjs"t |s5t |r;t ||_ n||_ |durYt|D]\}}}||j g|Ri||_ qFt|\|_|_|_dS)Nrlr)rvr\rXr[r&rrPrQrCrrrrWrLrrrrr]) r2r\rXr&rrrrrr6r6r7r1&s   zWebSocketRoute.__init__rEr!rc Cs|ddkrAt|}|j|}|rA|}|D]\}}|j||||<qt|di}| ||j |d}t j |fSt j ifS)Nrrr(r)rrrrrcr]rrrrrXr=rBr@rr6r6r7rCs      zWebSocketRoute.matchesr(r rcKr)Nrrrrr6r6r7rRrzWebSocketRoute.url_path_forrFr rGr"c||||IdHdSrHrLrr6r6r7r]zWebSocketRoute.handlerrcC"t|to|j|jko|j|jkSrH)rrr\rXrr6r6r7r`"zWebSocketRoute.__eq__cCs|jjd|jd|jdS)Nrrrr)r5r8r\r&r2r6r6r7rcszWebSocketRoute.__repr__) r\r'rXrYr&rrrr*r+rrrMrrrr6r6r6r7r%s   rc@sdeZdZ   d*ddd+ddZed,ddZd-ddZd.ddZd/d"d#Zd0d&d'Z d1d(d)Z dS)2MountNrr\r'rLASGIApp | NoneroutesSequence[BaseRoute] | Noner&rrrr*r+c Cs|dks |ds Jd|dus|dusJd|d|_|dur'||_nt|d|_|j|_|durLt|D]\}}}||jg|Ri||_q9||_t|jd\|_ |_ |_ dS)Nrnrlrz0Either 'app=...', or 'routes=' must be specified)rz /{path:path}) rvrstripr\ _base_appRouterrLrr&rrrr]) r2r\rLrr&rrrrr6r6r7r1hs   zMount.__init__list[BaseRoute]cCt|jdgSNr)rZrrr6r6r7rz Mount.routesrEr!rc Cs|ddvr_|dd}t|}|j|}|r_|}|D]\}}|j||||<q d|d}|dt | } t |di} | || |d||| |j d } t j| fSt jifS) Nrrr root_pathrnrlr\r( app_root_path)r(rrrX)rrrrrrcr]rrfrrrrLr=rBr@) r2rErrrrrgrhremaining_path matched_pathr(rr6r6r7rs&       z Mount.matchesr(r rc Ks(|jdur)||jkr)d|vr)|dd|d<t|j|j|\}}|s(t|dSnf|jdus6||jdr|jdur>|}n |t|jdd}|d}d|d<t|j|j|\}}|durd||d<|j phgD]%}z|j |fi|} t| dt | | j dWStyYqiwt||)Nr\rl)r\ror>rnr)r&r{rjrr]rrvrrrrrr'rr%) r2r&r(r\rremaining_name path_kwarg path_prefixrouteurlr6r6r7rs.   "  zMount.url_path_forrFr rGr"crrHrrr6r6r7rrz Mount.handlerrcCrrH)rrr\rLrr6r6r7rrz Mount.__eq__cC2|jj}|jpd}|d|jd|d|jdS)Nrnrr, app=rr)r5r8r&r\rLr2rr&r6r6r7r  zMount.__repr__)NNN) r\r'rLrrrr&rrrr*r+r*rrrrMrr r8r9r:r1propertyrrrrrrr6r6r6r7rgs   !  rc@sXeZdZd%d&d d Zed'd dZd(ddZd)ddZd*ddZd+d!d"Z d,d#d$Z dS)-HostNhostr'rLrr&rr*r+cCs<|dr Jd||_||_||_t|\|_|_|_dS)NrlzHost must not start with '/')rvrrLr&r host_regex host_formatr])r2rrLr&r6r6r7r1s z Host.__init__rcCrr)rZrLrr6r6r7rrz Host.routesrEr!rc Cs|ddvrMt|d}|dddd}|j|}|rM|}|D]\}}|j||||<q&t |di}| |||j d } t j | fSt jifS) NrrrErrnrorr()r(rX)rrrrrrrcr]rrrrLr=rBr@) r2rErrrrrgrhr(rr6r6r7rs       z Host.matchesr(r rc Ks|jdur&||jkr&d|vr&|d}t|j|j|\}}|s%t||dSnQ|jdus3||jdrw|jdur;|}n |t|jdd}t|j|j|\}}|jpTgD]!}z|j |fi|}tt ||j |dWSt yvYqUwt ||)Nr\)r\rror>)r\rr) r&rfrjrr]rrvrrrr'rr%) r2r&r(r\rrrrrr6r6r7rs&     zHost.url_path_forrFr rGr"crrHrrr6r6r7rrz Host.handlerrcCrrH)rrrrLrr6r6r7rrz Host.__eq__cCr)Nrnz(host=rrrr)r5r8r&rrLrr6r6r7rrz Host.__repr__rHrr'rLrr&rr*r+rrrrMrrrr6r6r6r7rs      r_Tc@s*eZdZdddZdddZdddZdS)_AsyncLiftContextManagercmAbstractContextManager[_T]cC ||_dSrH)_cm)r2r r6r6r7r1  z!_AsyncLiftContextManager.__init__r*r cs |jSrH)r __enter__rr6r6r7 __aenter__s z#_AsyncLiftContextManager.__aenter__exc_typetype[BaseException] | None exc_valueBaseException | None tracebacktypes.TracebackType | None bool | Nonecs|j|||SrH)r__exit__)r2rrrr6r6r7 __aexit__sz"_AsyncLiftContextManager.__aexit__N)r r )r*r )rrrrrrr*r)r8r9r:r1rrr6r6r6r7r  s  r lifespan_context)Callable[[Any], Generator[Any, Any, Any]]1Callable[[Any], AbstractAsyncContextManager[Any]]cs&t|tdfdd }|S)NrLr r*_AsyncLiftContextManager[Any]cs t|SrH)r rcmgrr6r7wrapper!s z+_wrap_gen_lifespan_context..wrapper)rLr r*r) contextlibcontextmanagerrPwraps)rr"r6r r7_wrap_gen_lifespan_contexts r&c@s4eZdZdddZdddZdd d ZdddZdS)_DefaultLifespanrouterrcCrrH)_router)r2r(r6r6r7r1)rz_DefaultLifespan.__init__r*r+cdSrHr6rr6r6r7r,z_DefaultLifespan.__aenter__exc_infoobjectcr*rHr6)r2r,r6r6r7r/r+z_DefaultLifespan.__aexit__r2r rLcCs|SrHr6)r2rLr6r6r7r2sz_DefaultLifespan.__call__N)r(r)r*r+)r,r-r*r+)r2r rLr-r*r )r8r9r:r1rrrr6r6r6r7r'(rr'c@seZdZ    d=ddd>ddZd?ddZd@dd Zd?d!d"Zd?d#d$Zd?d%d&ZdAd(d)Z dBdCd.d/Z dBdDd1d2Z   dEdFd8d9Z dBdGd;d<Z dS)HrNTrrrredirect_slashesrdefaultrlifespanLifespan[Any] | Nonerrr*r+c Cs|durgnt||_||_|dur|jn||_|dur"t||_n%t|r3t dt t ||_nt |rDt dt t||_n||_|j|_|rdt|D]\}}}||jg|Ri||_qQdSdS)Nzjasync generator function lifespans are deprecated, use an @contextlib.asynccontextmanager function insteadzdgenerator function lifespans are deprecated, use an @contextlib.asynccontextmanager function instead)r.rr. not_foundr/r'rrisasyncgenfunctionwarningswarnDeprecationWarningr isgeneratorfunctionr&rLmiddleware_stackr) r2rr.r/r0rrrrr6r6r7r17s0      zRouter.__init__rEr!rFr rGr"cs^|ddkrt}||||IdHdSd|vrtddtddd}||||IdHdS)NrrrLrrr)r$rr)r2rErFrGrrIr6r6r7r2^s   zRouter.not_foundr&r'r(r rc Ks@|jD]}z |j|fi|WStyYqwt||rH)rrr%)r2r&r(rr6r6r7rms   zRouter.url_path_forc sd}|d}|IdHzE||4IdH/}|dur,d|vr%td|d||ddiIdHd}|IdHWdIdHn 1IdHsMwYWn#tyvt}|rk|d |d IdH|d |d IdHw|dd iIdHdS) z{ Handle ASGI lifespan messages, which allows us to manage application startup and shutdown events. FrLNstatez:The server does not support "state" in the lifespan scope.rzlifespan.startup.completeTzlifespan.shutdown.failed)rmessagezlifespan.startup.failedzlifespan.shutdown.complete)rr RuntimeErrorr BaseExceptionr format_exc)r2rErFrGstartedrL maybe_stateexc_textr6r6r7r0us0  ( zRouter.lifespancs||||IdHdS)z; The main entry point to the Router class. N)r8rr6r6r7rszRouter.__call__c s|ddvs Jd|vr||d<|ddkr#||||IdHdSd}|jD]-}||\}}|tjkrH||||||IdHdS|tjkrU|durU|}|}q(|durk||||||IdHdSt|} |ddkr|j r| dkrt |} | dr| d d| d<n| dd| d<|jD]&}|| \}}|tj krt| d} tt| d } | |||IdHdSq||||IdHdS) Nr)rrr0r(r0rrlr\r)r)r0rrr=rBrrrArr.rendswithrr@rrr'r/) r2rErFrGrQrrr partial_scoperredirect_scope redirect_urlrIr6r6r7rLsJ         z Router.apprcCst|to |j|jkSrH)rrrrr6r6r7rsz Router.__eq__r\rLrrcCt|||d}|j|dSN)rLr&)rrappend)r2r\rLr&rr6r6r7mountz Router.mountrcCrErF)rrrG)r2rrLr&rr6r6r7rrIz Router.hostrXrDrrrcCs"t|||||d}|j|dS)N)rXrr&r)rrrG)r2r\rXrr&rrr6r6r7 add_routeszRouter.add_routerTcCrE)N)rXr&)rrrG)r2r\rXr&rr6r6r7add_websocket_routeszRouter.add_websocket_route)NTNN) rrr.rr/rr0r1rrr*r+rMrrrH)r\r'rLrr&rr*r+r )NNT) r\r'rXrDrrr&rrrr*r+)r\r'rXrTr&rr*r+)r8r9r:r1r2rr0rrLrrHrrJrKr6r6r6r7r6s,  '     2   r)rCrDr*r)rCrTr*r)rXrYr*r')r\r'r]r^r(r_r*r`)r\r'r*rk)rrr*r)M __future__rr#rPrr|rtypesr4collections.abcrrrrrrr r enumr r typingr rstarlette._exception_handlerrstarlette._utilsrrstarlette.concurrencyrstarlette.convertorsrrstarlette.datastructuresrrrstarlette.exceptionsrstarlette.middlewarerstarlette.requestsrstarlette.responsesrrrstarlette.typesrrr r!r"starlette.websocketsr#r$ Exceptionr%r=rSrWr[rjrrxrrrrrrr r r&r'rr6r6r6r7sT              8aBd>