Jj!UdZddlZddlZddlZddlZddlZddlmZddlZddl Zddl m Z m Z m Z mZddlmZmZddlZddlmZdZGdd ZiZeeefed <daeejed <d Zd efdZdZGddeZ GddeZ!GddeZ"GddeZ#GddeZ$de fdZ%dS)zJupyter notebook execution endpoints. Provides per-cell execution with multi-session support. Each session gets its own Jupyter kernel via nbclient. Requires the ``notebooks`` optional extra:: pip install open-terminal[notebooks] N)Optional) APIRouterDepends HTTPExceptionQuery) BaseModelField)NotebookClientic&eZdZdZdZdedefdZdS)_Sessionz/Wraps a NotebookClient for a specific notebook.)idpathnbclientbusy created_at last_used session_idrc||_||_||_||_d|_t j|_t j|_dS)NF)r rrrrtimerr)selfrrrrs q/home/rurouni/.cache/uv/archive-v0/6mr0qdqUAO9GXgnL/lib/python3.11/site-packages/open_terminal/utils/notebooks.py__init__z_Session.__init__%sC   )++N)__name__ __module__ __qualname____doc__ __slots__strrrrr r sA99QI%3%c%%%%%%rr _sessions _cleanup_taskcK tjdd{VtjfdtD}|D]}t |d{Vm)z>Periodically remove sessions idle for more than _IDLE_TIMEOUT.T<NcNg|]!\}}|jz tk|j|"Sr!)r _IDLE_TIMEOUTr).0sidsnows r z&_idle_cleanup_loop..8s@   QQ[ =000 000r)asynciosleeprr"items_destroy_session)staler)r+s @r_idle_cleanup_loopr23s (mBikk    #//++     ( (C"3'' ' ' ' ' ' ' ' ' (rrcKt|d}|r:|jr5 |jd{VdS#t$rYdSwxYwdSdSN)r"popr_async_cleanup_kernel Exception)rsessions rr0r0AsmmJ--G7> .6688 8 8 8 8 8 8 8 8 8    DD sA AActtr"tjt adSdSr4)r#doner- create_taskr2r!rr_ensure_cleanup_taskr<Js< 2 2 4 4+,>,@,@AA rc4eZdZUeddZeed<dS)CreateSessionRequest.z!Absolute path to the .ipynb file. descriptionrN)rrrr rr __annotations__r!rrr>r>Us2c'JKKKD#KKKKKrr>c.eZdZUeed<eed<eed<dS)CreateSessionResponser kernelstatusNrrrr rAr!rrrCrCYs+ GGG KKK KKKKKrrCcdeZdZUeddZeed<eddZee ed<dS)ExecuteCellRequest.z!Zero-based cell index to execute.r? cell_indexNzJOverride cell source. If omitted, uses the source already in the notebook.source) rrrr rIintrArJrr r!rrrHrH_sgeC-PQQQJQQQ!E fFHSMrrHcVeZdZUeed<dZeeed<ee Z e ed<dS)ExecuteCellResponserENexecution_count)default_factoryoutputs) rrrr rArNrrKr listrPr!rrrMrMfsL KKK%)OXc])))E$///GT/////rrMc8eZdZUeed<eed<eed<eed<dS)SessionStatusResponser rrDrENrFr!rrrSrSls4 GGG III KKK KKKKKrrSreturnctddgt|g}|dtdddd d tfd }|d t d ddd dt d tfd}|dtddddt fd}| dddddt fd}|S)z;Create the notebooks router with the given auth dependency.z /notebooks notebooks)prefixtags dependenciescreate_notebook_sessionzCreate a notebook sessionzaStart a Jupyter kernel for the given notebook. Returns a session ID for subsequent execute calls.F)response_model operation_idsummaryr@include_in_schemareqcKttj|j}tjj|d{Vst dd|t j|d4d{V}|d{V}dddd{Vn#1d{VswxYwY tj |d}n&#t$r}t dd |d}~wwxYw|j d i d d }t||d } ||d{V|d{Vn)#t$r}t dd|d|d}~wwxYwt%jjdd}t+||||t,|<t/||dS)NzNotebook not found:  status_codedetailutf-8encoding) as_versionzInvalid notebook: kernelspecnamepython3x) kernel_nametimeoutizFailed to start kernel 'z': ready)r rDrE)r<osrabspathaiofilesisfileropenreadnbformatreadsr7metadatagetr create_kernel_managerasync_start_new_kernelasync_start_new_kernel_clientuuiduuid4hexr r"rC) r`rfcontentrerprrs rcreate_sessionz/create_notebooks_router..create_sessions wsx(([%,,T22222222 WC8Ut8U8UVVV V=888 % % % % % % %AFFHHnnnnnnG % % % % % % % % % % % % % % % % % % % % % % % % % % % RA666BB R R RC8PQ8P8PQQQ Q RkoolB77;;FINN   SIII   ( ( * * *//11 1 1 1 1 1 1 16688 8 8 8 8 8 8 8 8   E+EE!EE   Z\\%crc* (T2v F F *$+g    sCB<< C CC%% D/DD AF F< F77F<z/{session_id}/executeexecute_notebook_cellzExecute a notebook cellzExecute a single cell in the given session. Optionally override the cell source. Updates the .ipynb file in place after execution.rc Kt|}|stdd|jrtdd|j}|jdks|jt |jkr2tdd|jd t |jd z d |j|j}|j |j|_d |_tj |_ |j ||jd{Vnb#t$rU}d |_tddt|jt#|t#|gdgcYd}~Sd}~wwxYwd |_tj |_ g}|jD]B}t'|}d|vrt'|d|d<||C|d} t+j|} t/j|jdd4d{V} | | d{Vdddd{Vn#1d{VswxYwYn#t$rYnwxYwtd| |S)NrbSession not foundrcizCell already executingrrkz cell_index z out of range (0..)TFerror) output_typeenameevalue traceback)rErPdatarNwrfrgok)rErNrP)r"r}rrrrIlencellsrJrrrasync_execute_cellr7rMtyperr rPdictappendrzwritesrvrxrwrite) rr`r8rcellrrPoodecnb_jsonrs r execute_cellz-create_notebooks_router..execute_cellsq-- ++ MC8KLLL L < RC8PQQQ Q Z >A  3rx==!@!@[S^[[s28}}WXGX[[[  x' : !*DK  IKK .33D#.II I I I I I I I I    GL&(/!%a!1"%a&&&)!ffX             IKK  AaB||!"V*--6 NN2     XX' ( ( ob))G}W\3III ' ' ' ' ' ' 'Qggg&&&&&&&&& ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' '    D #W    sU.&D E4A E/)E4/E486I..I I. I&&I.)I&*I.. I;:I;z /{session_id}get_notebook_sessionzGet notebook session status)r\r]r^r_cKt|}|stdd|jjdidd}|jrdnd}t |j|j|| S) Nrbrrcrlrmrnrrs)r rrDrE) r"r}rrr|rrSr r)rr8rprEs r get_sessionz,create_notebooks_router..get_sessions-- ++ MC8KLLL Lj)--lB??CC I  #<4W$z [    rdelete_notebook_sessionzStop a notebook session)r]r^r_clK|tvrtddt|d{VddiS)NrbrrcrEstopped)r"rr0)rs rdelete_sessionz/create_notebooks_router..delete_sessionsR Y & &C8KLLL Lz*********)$$r) rrpostrCr>rMr rHr}rSdelete)verify_api_keyrouterrrrrs rcreate_notebooks_routerrxsu]n--.F [[ ,.+w $ "6$ $ $ $ L [[*,)<? s? 1C? ? ? ? B ZZ,+-   c       ]].)  %%%%  % Mr)&rr-jsonrtrrtypingrrv aiofiles.osfastapirrrrpydanticrr rznbclientr r'r r"rr rAr#Taskr2r0r<r>rCrHrMrSrr!rrrsd <<<<<<<<<<<<%%%%%%%%######  % % % % % % % %"$ 4X ###(, x %,,, ( ( (sBBBLLLLL9LLLI 00000)000 Icyccccccr