a Df@sddlZddlZddlZddlZddlmZddlmZddlm Z ddl Z ddl m Z ddl mZddlmZd d lmZGd d d e jZGd dde jZGdddeZddZddZddZeedZedD]Zeeej<qdS)N)run_path) MagicMock)parse_qs)web) WSGIContainer)entry_points_for)statec@seZdZdZdS) HTTPErrorz Custom HTTPError type N)__name__ __module__ __qualname____doc__rrZ/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/panel/io/rest.pyr sr c@seZdZddZdS) BaseHandlerc Ks|dd|jdrbd|vrbg}tj|dD]}||q2|td||j |din|td||j didS)N Content-Typeapplication/jsonZserve_tracebackexc_infoerror)codemessage traceback)rr) set_headersettingsgetrformat_exceptionappendfinishjsondumpsZ_reason)self status_codekwargslineslinerrr write_errors$   zBaseHandler.write_errorN)r r rr'rrrrrsrcs<eZdZfddZeddZeddZddZZS) ParamHandlerc s(|dd|_tj||fi|dS)Nroot)popr)super__init__)r"apprequestr$ __class__rrr,3szParamHandler.__init__csfdd|D}j|S)Ncsi|]}|t|qSr)getattr).0p parameterizedrr 9z*ParamHandler.serialize..)paramZserialize_parameters)clsr5 parametersvaluesrr4r serialize7szParamHandler.serializecsB|D]&}|jvrd|d}t|ddqfdd|DS)N'z!' query parameter not recognized.i)reasonr#cs i|]\}}|j||qSr)r8Zdeserialize_value)r2r3vr4rrr6Bsz,ParamHandler.deserialize..)r8r items)r9r5r:r3r>rr4r deserialize<s   zParamHandler.deserializecs|jj}|||jt|jd}tj|d\}}}|sBdSt|jj }| |d|}|dj j fi|| dd|||d|dS)N)NNNrrr)r.pathindexr)lenr Z_rest_endpointsrrqueryrAr8updaterwriter<)r"rBendpointr5r:_argsparamsrrrrEs   zParamHandler.get) r r rr, classmethodr<rAr __classcell__rrr/rr(1s    r(c Csddlm}m}ddlm}m}g}|D]}|dd}|dkrL||}n\|dkrz ddl} Wn.ty} ztd| WYd} ~ n d} ~ 00||}n|d || |j q(||d d d S) Nr)NotebookHandler ScriptHandler)UnsupportedFileTypemake_app.pyipynb4Please install nbconvert to serve Jupyter Notebooks.-{} is not a script (.py) or notebook (.ipynb)zPanel REST APIzrest/)prefix) Ztranquilizer.handlerrNrOZtranquilizer.mainrPrQsplit nbconvert ImportErrorformatextendZtranquilized_functions) filesrNrOrPrQZ functionsfilename extensionsourcerZerrrbuild_tranquilize_applicationTs     rccCs(t|}t|}d|tjt|dfgS)a Returns a Tranquilizer based REST API. Builds the API by evaluating the scripts and notebooks being served and finding all tranquilized functions inside them. Arguments --------- files: list(str) A list of paths being served endpoint: str The endpoint to serve the REST API on Returns ------- A Tornado routing pattern containing the route and handler z^/%s/.*)fallback)rcrrZFallbackHandlerdict)r^rHr-trrrrtranquilizer_rest_providerjsrgc Cs||D]:}|dd}|dkrPz t|Wn tyLtjjdYn0q|dkr2z ddl}WntytddYn0dd lm}|}| |\}}t j |} t jd | d d ^} | || zt| jd tidWn"tytjjdYn0Wdn1s&0Yqtd|q|r\|ds\|d7}|rjd|ndtt|dfgS)a] Returns a Param based REST API given the scripts or notebooks containing the tranquilized functions. Arguments --------- files: list(str) A list of paths being served endpoint: str The endpoint to serve the REST API on Returns ------- A Tornado routing pattern containing the route and handler rRrSrTz0Could not run app script on REST server startup.rUrNrV)ScriptExporterwT)modedirdeleteZ get_ipython)Z init_globalsz2Could not run app notebook on REST server startup.rW/z^/%s.*z^.*)r))rYr Exceptionr8mainwarningrZr[rh from_filenameosrBdirnametempfileNamedTemporaryFilerGflushnamer ValueErrorr\endswithr(re) r^rHr_r`rZrhZexporterrarI source_dirtmprrrparam_rest_providers6         6r|)Z tranquilizerr8z panel.io.rest)r rrrtrZrunpyrZ unittest.mockr urllib.parserr8ZtornadorZ tornado.wsgirZ entry_pointsrr r ZRequestHandlerrr(rcrgr|ZREST_PROVIDERS entry_pointloadrwrrrrs,       #0