bi$UddlmZmZddlmZddlmZddlZddlZddl Z ddl Z ddl m Z mZddlmZddlmZddlZddlmZdd lmZd d lmZmZd d lmZiZe eefed <e j@Z!defdZ"dZ#GddeeZ$Gdde$Z%Gdde$Z&Gdde$Z'Gdde$Z(Gdde$Z)Gdde$Z*Gdd e$Z+Gd!d"e$Z,Gd#d$e$Z-Gd%d&e$Z.Gd'd(e$Z/Gd)d*e$Z0d+Z1y),)ABCabstractmethod)Path) HTTPStatusN)DictList) APIHandler) url_path_join)KernelProvisionerFactory)KernelProvisionerBase) NsightTooltools) WebRTCManagerkernel_id_to_nsight_tool provisionerc>fd}|j_y)Ncbtjdttffd }|S)Ncmdctjjd}||j|z}|fi|SN)rget kernel_idlaunch_kernel_cmd)rkwargstoolfuncrs \/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jupyterlab_nvidia_nsight/handlers.py launch_kernelzFwrap_provisioner..launch_kernel_wrapper..launch_kernel#sD+// 0E0EtLD,,.4&v& &) functoolswrapsrstr)rrrs` rlaunch_kernel_wrapperz/wrap_provisioner..launch_kernel_wrapper"s1   'tCy '  ' r )r)rr$s` rwrap_provisionerr%!s!6k6O6O PKr cHd}|tjt_y)NcBtjfd}|S)Nc.|i|}t||Sr)r%)argsrrrs rcreate_provisioner_instancezqwrap_kernel_provisioner_factory..create_provisioner_instance_wrapper..create_provisioner_instance0s //K [ ) r )r!r")rr*s` r#create_provisioner_instance_wrapperzLwrap_kernel_provisioner_factory..create_provisioner_instance_wrapper/s&     +*r )r r*)r+s rwrap_kernel_provisioner_factoryr,.s!+<_ <<<>8r cTeZdZedZej jdZy) HandlerBasec yr)selfrs r post_implzHandlerBase.post_impl;s r c * |jdi|jxsiy#t$rb}|jtj j |jtjdt|iYd}~yd}~wwxYw)Nmessager0) r2 get_json_body BaseException set_statusrINTERNAL_SERVER_ERRORvaluefinishjsondumpsr#)r1es rpostzHandlerBase.post?so 9 DNN :d0028b : 9 OOJ<<BB C KK Is1v#67 8 8 9s$' BAB  BN) __name__ __module__ __qualname__rr2tornadoweb authenticatedr>r0r rr.r.:s2   [[99r r.ceZdZdZy)DoIsInsideDockerHandlercl|jtjdtjiy)N inside_docker)r:r;r<rrH)r1s rr2z!DoIsInsideDockerHandler.post_implIs" DJJ1L1LMNOr Nr?r@rAr2r0r rrFrFHsPr rFc ,eZdZddddedededefdZy) DoEnableHandler)r)installation_path tool_typerr)rMc t||||}|jVd|d}|jtjj |j tjd|iy|t|<|j tj|jxsdt|jdy)NzCould not find z\ executable. Please set the installation path in jupyterlab-nvidia-nsight extension settingsr4rL) target_exer) rrPr7rr8r9r:r;r<rr#r)r1rNrr)rMrr4s rr2zDoEnableHandler.post_implNsY +@[] J  J-0 J8; JUX Jr rKceZdZdefdZy)DoDisableHandlerrcntj|}|j|jyr)rpopcleanupr:)r1rrs rr2zDoDisableHandler.post_impl]s$'++I6  r NrQr0r rrSrS\scr rSceZdZdedefdZy)DoSetVersionHandlerrversionc  t|j|}|r'|jtjd|iy|jy#t j $rb}|jtjj|jtjdt|iYd}~yd}~wwxYw)Nwarningr4) r set_versionr:r;r<r VersionErrorr7rr8r9r#)r1rrYr[r=s rr2zDoSetVersionHandler.post_implds 9.y9EEgNG DJJ 7';<= && 9 OOJ<<BB C KK Is1v#67 8 8 9sAAAC 'ACC NrQr0r rrXrXcs 9c 9C 9r rXceZdZdefdZy)DoGetStartCodeHandlerrc t|jdi|}|jtjd|iyNcoder0)rget_start_coder:r;r<)r1r start_kwargsrbs rr2zDoGetStartCodeHandler.post_implqs3' 2AAQLQ DJJ~./r NrQr0r rr_r_ps0c0r r_ceZdZdZy)DoGetStopCodeHandlerc t|jdi|}|jtjd|iyra)r get_stop_coder:r;r<)r1r stop_kwargsrbs rr2zDoGetStopCodeHandler.post_implws3' 2@@O;O DJJ~./r NrIr0r rrfrfvs0r rfc&eZdZeddededefdZy)DoStartWebRTCHandler)host report_pathrNrlc 2 tj|tj||}|P|j t j j|jtjddiy|jtjdd|d|iy#t$rb}|j t j j|jtjdt|iYd}~yd}~wwxYw)Nr4z Failed to start WebRTC containerurlzhttp://:) rrunrnormalize_pathr7rr8r9r:r;r< Exceptionrepr)r1rmrNrlportr=s rr2zDoStartWebRTCHandler.post_impl}s : $$Y 0I0I+0VX\]D| @ @ F FG DJJ 3U'VWX DJJav/F'GHI : OOJ<<BB C KK ItAw#78 9 9 :sA;B+>,B++ D4ADDN)r?r@rA default_hostr#r2r0r rrkrk|sIU : : :3 :r rkceZdZdefdZy)DoStopWebRTCHandlerrmcttjtj||j yr)rstoprrrr:)r1rms rr2zDoStopWebRTCHandler.post_impls$:44[AB r NrQr0r rrxrxsr rxceZdZdefdZy)DoIsFileExistsHandlerpathc|jtjdtj|j iy)Nexists)r:r;r<rrrr)r1r}s rr2zDoIsFileExistsHandler.post_impls0 DJJ**C*CD*I*P*P*RSTUr NrQr0r rr|r|sVVr r|ceZdZdefdZy)DoIsWebRTCPulledHandlerrNcX tjtj||jt j ddiy#t jj$r;}|jt j d|jdYd}~yd}~wwxYw)NpulledTF)rr4) rget_docker_imageget_docker_clientr:r;r<dockererrors ImageNotFound explanation)r1rNr=s rr2z!DoIsWebRTCPulledHandler.post_implsx Q  * *=+J+J+Li X KK Hd#34 5}}** Q KK e #NO P P QsAAB).1B$$B)NrQr0r rrrsQcQr rceZdZdefdZy)DoPullWebRTCHandlerrNcNtj||jyr)r pull_imager:)r1rNs rr2zDoPullWebRTCHandler.post_impls  + r NrQr0r rrrscr rceZdZdefdZy)DoCheckOutputDirr}c tj|}|jsT|jtj j |jtjdd|diy|jsT|jtj j |jtjdd|diy|dz } |j|j|jy#t$rV|jtj j |jtjdd|diYywxYw)Nr4z"Output directory does not exist: ""z&Output directory is not a directory: "z,jupyterlab_nvidia_nsight_permission_test.txtz#Output directory is not writable: ")rrrrr7rr8r9r:r;r<is_dirtouchunlinkPermissionError)r1r} test_paths rr2zDoCheckOutputDir.post_impls*((.{{} OOJ<<BB C KK I1STXSYYZ/[#\] ^ {{} OOJ<<BB C KK I1WX\W]]^/_#`a b II  ` OO      KKM ` OOJ<<BB C KK I1TUYTZZ[/\#]^ _ `s$0DAE43E4NrQr0r rrrs``r rc bt|jdd}dtfdtfdtfdt fdt fdtfd tfd tfd tfd tfd tfdtff Dcgc]\}}t|||f}}}|jd|tycc}}w)Nbase_urlzjupyterlab-nvidia-nsightzis-inside-dockerenabledisablez set-versionzget-start-codez get-stop-codez start-webrtcz stop-webrtczis-file-existszis-webrtc-pulledz pull-webrtczcheck-output-directoryz.*$)r settingsrFrKrSrXr_rfrkrxr|rrr add_handlersr,)web_appbase_route_patternnamehandlerhandlerss rsetup_handlersrs&w'7'7 'CE_` !8 9  ' ( ) / 0 4 5 2 3 1 2 / 0 4 5 !8 9 / 0 %'7 8   D' )4 0':H" )#%%s3B+)2abcrrpathlibrhttprr!r;socket docker.errorsrtypingrrjupyter_server.base.handlersr jupyter_server.utilsr rBjupyter_client.provisioningr ,jupyter_client.provisioning.provisioner_baser nsight_toolsrrwebrtc_containerrrr#__annotations__getfqdnrvr%r,r.rFrKrSrXr_rfrkrxr|rrrrr0r rrs$# 3.@N++35$sJ/4v~~  Q"7 Q > 9*c 9PkP Jk J{ 9+ 90K0 0;0 :; :+ VKV QkQ+ `{`,&r