XhNBdZddlZddlZddlZddlmZmZmZmZm Z ede Z ede Z ddl Z ddlmZmZddlmZdd lmZdd lmZmZmZmZdd lmZdd lm Z Gd de!Z"eGddZ#GddZ$GddeZ%y#e$r ddlmZ YiwxYw)zA base class session manager.N)AnyNewTypeOptionalUnioncast KernelName ModelName)dbapi2) dataclassfields) ensure_async)web)Instance TraitErrorUnicodevalidate)LoggingConfigurable)InstanceFromClassesceZdZdZy)KernelSessionRecordConflictzgException class to use when two KernelSessionRecords cannot merge because of conflicting data. N)__name__ __module__ __qualname____doc__c/var/www/html/myenv/lib/python3.12/site-packages/jupyter_server/services/sessions/sessionmanager.pyrrsrrcReZdZUdZdZeeed<dZeeed<de de fdZ d dZ y) KernelSessionRecordzA record object for tracking a Jupyter Server Kernel Session. Two records that share a session_id must also share a kernel_id, while kernels can have multiple session (and thereby) session_ids associated with them. N session_id kernel_idotherreturnct|tr|jxr|j|jk(}t|j|jk(|jduxs|jdug}t ||gryt|j|j|jk(|j|jk7gr d}t |y)z Whether a record equals another.NTzA single session_id can only have one kernel_id associated with. These two KernelSessionRecords share the same session_id but have different kernel_ids. This should not be possible and is likely an issue with the session records.F) isinstancerr!allr anyr)selfr" condition1 condition2msgs r__eq__zKernelSessionRecord.__eq__/s e0 1MDNNeoo,MJOOu'7'77NNd*Eeoo.EJ J +,OOOOu'7'77NNeoo52#66rc t|ts d}t||jr2|jr&|j|jk7r d}t |t |D][}t ||jst||js1t||jt||j]y)zCUpdates in-place a kernel from other (only accepts positive updatesz3'other' must be an instance of KernelSessionRecord.zJCould not update the record from 'other' because the two records conflict.N) r%r TypeErrorr!rr hasattrnamegetattrsetattr)r(r"r+fields rupdatezKernelSessionRecord.updateOs%!45GCC. ??t~~%//T^^2S^C-c2 2D\ FEuejj)geUZZ.Hejj'%*DE Fr)r"rr#N) rrrrr rstr__annotations__r!objectboolr,r4rrrrr#s?!%J $#Ix}#Ft@ FrrceZdZUdZeeed<defdZdZde ee fde fdZ d Z de ee fdefd Zdedd fd Zdedd fd Zy )KernelSessionRecordLista.An object for storing and managing a list of KernelSessionRecords. When adding a record to the list, the KernelSessionRecordList first checks if the record already exists in the list. If it does, the record will be updated with the new information; otherwise, it will be appended. _recordsrecordscBg|_|D]}|j|yzInitialize a record list.N)r;r4)r(r<records r__init__z KernelSessionRecordList.__init__is$  F KK  rc,t|jS)z+The string representation of a record list.)r5r;r(s r__str__zKernelSessionRecordList.__str__o4==!!rr?r#ct|tr||jvryt|tr-|jD]}||j|j fvsyy)z.Search for records by kernel_id and session_idTF)r%rr;r5r r!)r(r?rs r __contains__z$KernelSessionRecordList.__contains__ssU f1 2v7N fc "]] allAKK88 rc,t|jS)zThe length of the record list.)lenr;rBs r__len__zKernelSessionRecordList.__len__~rDrct|tr0|jD] }||j|jfvs|cSn*t|t r|jD] }||k(s |cS|d}t |)zsReturn a full KernelSessionRecord from a session_id, kernel_id, or incomplete KernelSessionRecord. z& not found in KernelSessionRecordList.)r%r5r;r!r r ValueError)r(r?rFr+s rgetzKernelSessionRecordList.gets fc "]] akk1<<88H  3 4]] "Q;!M ">?orNc |jj|}|j|j|y#t$r|jj |YywxYw)z9Update a record in-place or append it if not in the list.N)r;indexr4rLappend)r(r?idxs rr4zKernelSessionRecordList.updatesS )--%%f-C MM#  % %f - ) MM  ( )s9<$A#"A#cX||jvr|jj|yy)zYRemove a record if its found in the list. If it's not found, do nothing. N)r;remove)r(r?s rrSzKernelSessionRecordList.removes' T]] " MM  ( #r)rrrrlistrr6r@rCrr5r8rGrJrMr4rSrrrr:r:^s&'' !4 " 5)Q )0)T))0)T)rr:ceZdZdZeddj dZeddZe d Z e d d gZ fd Z d Zd ZhdZedZedZdZdZdZdefdZ d&deedeedeedeedeedeeeff dZ d'deedeedeeeffdZdedeedeedeedeedef dZ d(dZ!d Z"d!Z#dede$fd"Z%d)d#Z&d$Z'd%Z(xZ)S)*SessionManagerzA session manager.:memory:zThe filesystem path to SQLite Database file (e.g. /path/to/session_database.db). By default, the session database is stored in-memory (i.e. `:memory:` setting from sqlite3) and does not persist when the current Jupyter Server shuts down.) default_valuehelpT)configdatabase_filepathcL|d}|dk(r|Stj|}|jrf|jr d}t |t |d5}|j d}dddjds|dk7r d }t ||S#1swY.xYw) zValidate a database file path.valuerWzL`database_filepath` expected a file path, but the given path is a directory.rbdNsSQLite format 3rz.The given file is not an SQLite database file.)pathlibPathexistsis_dirropenread startswith)r(proposalr]pathr+fheaders r_validate_database_filepathz*SessionManager._validate_database_filepaths! J L||E" ;;={{}d o%eT" %a %$$%78Vs]F o%  % %s BB#zBjupyter_server.services.kernels.kernelmanager.MappingKernelManagerz8jupyter_server.services.contents.manager.ContentsManagerz2notebook.services.contents.manager.ContentsManagercBt||i|t|_yr>)superr@r:_pending_sessions)r(argskwargs __class__s rr@zSessionManager.__init__s  $)&)!8!:rN>r0rhtyper!r c|j:|jj|_|jjd|jS)z5Start a cursor and create a database called 'session'z\CREATE TABLE IF NOT EXISTS session (session_id, path, name, type, kernel_id))_cursor connectioncursorexecuterBs rrvzSessionManager.cursorsE << ??113DL LL = ||rc|jEtj|jd|_tj|j_|jS)zStart a database connectionN)isolation_level) _connectionsqlite3connectr[Row row_factoryrBs rruzSessionManager.connectionsH    #&t/E/EW[\D +2;;D   (rc`|j"|jjd|_yy)zClose the sqlite connectionN)rtcloserBs rrzSessionManager.closes( << # LL   DL $rc$|jy)z+Close connection once SessionManager closesN)rrBs r__del__zSessionManager.__del__s  rcKd}|jjd|f|jj}||j|dd{}|d}|S7 w)z2Check to see if the session of a given name existsFz"SELECT * FROM session WHERE path=?NT)tolerate_culled)rvrwfetchone row_to_model)r(rhrbrowmodels rsession_existszSessionManager.session_existssi @4'Jkk""$ ? ++C+FFE  GsAA!A A!r#c<ttjS)zCreate a uuid for a new session)r5uuiduuid4rBs rnew_session_idzSessionManager.new_session_ids4::<  rrhr0rr kernel_namer!cK|j}t|}|jj||||jvrn|j |||||d{}||_|jj||j|||||d{}|jj|ttttf|S7|7>w)zCreates a session and returns its model Parameters ---------- name: ModelName(str) Usually the model name, like the filename associated with current kernel. r N)rhr0rrr!) rrrnr4kernel_managerstart_kernel_for_sessionr! save_sessionrSrdictr5r) r(rhr0rrrr!r r?results rcreate_sessionzSessionManager.create_sessions ((* $ ; %%f-  Y$2E2E%E ";;D$kI% %%f-(( T49)   %%f-DcNF++  s$A!C$#C $?C$#C"$=C$"C$c|;|jj|}tjj ||}t |t sJitjd|iS)a[Return the environment variables that need to be set in the kernel Parameters ---------- path : str the url path for the given session. name: ModelName(str), optional Here the name is likely to be the name of the associated file with the current kernel at startup time. JPY_SESSION_NAME)r cwd_for_pathosrhjoinr%r5environ)r(rhr0cwds rget_kernel_envzSessionManager.get_kernel_env(s[  %%2248C77<<T*D$$$$7"**70$77rr cKt|jj|d{}|j||}|jj |||d{}t t|S7L7w)aQStart a new kernel for a given session. Parameters ---------- session_id : str uuid for the session; this method must be given a session_id path : str the path for the given session - seem to be a session id sometime. name : str Usually the model name, like the filename associated with current kernel. type : str the type of the session kernel_name : str the name of the kernel specification to use. The default kernel name will be used if not provided. )rhN)rhrenv)r contents_managerget_kernel_pathrr start_kernelrr5) r(r rhr0rrr kernel_path kernel_envr!s rrz'SessionManager.start_kernel_for_session;s2))>)>)N)NTX)N)YZZ ((t4 --::#;  C##[ s!)A<A87A<#A:$A<:A<cK|jjd|||||f|j|d{}|S7w)aSaves the items for the session with the given session_id Given a session_id (and any other of the arguments), this method creates a row in the sqlite session database that holds the information for a session. Parameters ---------- session_id : str uuid for the session; this method must be given a session_id path : str the path for the given session name : str the name of the session type : str the type of the session kernel_id : str a uuid for the kernel associated with this session Returns ------- model : dict a dictionary of the session model z&INSERT INTO session VALUES (?,?,?,?,?)rN)rvrw get_session)r(r rhr0rrr!rs rrzSessionManager.save_session^sM2  4 tT9 5 '':'>> ?s7AAAcK|s d}t|g}|D]4}||jvrd|}t||jd|z6ddj|z}|jj |t |j |jj}|Xg}|jD]\}} |j|d| tjdd d j|z |j|d{} | S#t$rd}YwxYw7#t$r(} tjdd t| z| d} ~ wwxYww) a:Returns the model for a particular session. Takes a keyword argument and searches for the value in the session database, then returns the rest of the session's info. Parameters ---------- **kwargs : dict must be given one of the keywords and values from the session database (i.e. session_id, path, name, type, kernel_id) Returns ------- model : dict returns a dictionary that includes all the information from the session described by the kwarg. zmust specify a column to queryzNo such column: %s=?zSELECT * FROM session WHERE %sz AND N=izSession not found: %s, )r._columnsrPrrvrwrTvaluesrKeyErroritemsr HTTPErrorrr5) r(rpr+ conditionscolumnqueryrqkeyr]res rrzSessionManager.get_session~ss$2CC.   /FT]]*(1n$   fvo .  / 1GLL4LM E4 #89 ++&&(C ;A$lln - UC5%+, ---%< ! %MN N N++C00E  C 1 N--%> KK ( );diioN E#GT&--/%:#GJ#GH 4&& 5 KK  NQ[P\ %)KK$8$8$: !D$    * *YDDWDWX\^bDc * d 6 6sD'D$D D'c&K||jvSw)zQChecks if the kernel is still considered alive and returns true if its not found.)r)r(r!s r kernel_culledzSessionManager.kernel_culleds 3 333scKt|j|dd{}|re|jjd|dfdj |d|d}|r|j j |dyt|t|jj|dd{}|d|d|d |d |d }|d d k(r|d|d d |d <|S772w)z@Takes sqlite database session row and turns it into a dictionaryr!N&DELETE FROM session WHERE session_id=?r zKernel '{kernel_id}' appears to have been culled or died unexpectedly, invalidating session '{session_id}'. The session has been removed.)r!r z Continuing...rhr0rr)idrhr0rrkernelnotebook)rhr0) r rrvrwformatlogwarningrr kernel_model)r(rrrr+rrs rrzSessionManager.row_to_models$01C1CC DT1U$VV  KK   H3|K\J^ _UU[U[!+.3|;LV\V    C5!893- )$*=*=*J*J3{K[*\]] l#KKK"   v;* $),Vc&k JE*  AW*^s"!C.C*BC.9C,:1C.,C.cK|jjd}g}|jD]-} |j|d{}|j |/|S7#t $rY?wxYww)z_Returns a list of dictionaries containing all the information from the session databasezSELECT * FROM sessionN)rvrwfetchallrrPr)r(crrrs r list_sessionszSessionManager.list_sessionssz KK   7 8::< C "//44 e$   5  s:1A3A$A" A$A3"A$$ A0-A3/A00A3cjKt|}|jj||j|d{}t |j j |ddd{|jjd|f|jj|y7o7?w)z=Deletes the row in the session database with given session_idrNrrr) rrnr4rr rshutdown_kernelrvrwrS)r(r r?sessions rdelete_sessionzSessionManager.delete_session s$ ; %%f-((J(??4..>>wx?PQU?VWXXX DzmT %%f-@Xs"=B3B/1B31B12>B31B3)NNNNN)N)NNNN)F)*rrrrrtagr[rrkrrrrr@rtrzrpropertyrvrurrrr5rrr rrrrrrrrrr8rrrr __classcell__)rqs@rrVrVs  O cc!"#(bcN* F @ ; GKBH     !! #$(",0#' ,sm,y!,sm , j) , C= , c3h,D@D8SM8)1))<8 c3h8&!$!$sm!$y! !$ sm !$ j) !$ !$F@1f"eH4S4T4"H .rrV)&rrr`rtypingrrrrrr5rr r{ ImportError pysqlite2r dataclassesr r jupyter_core.utilsr tornador traitletsrrrrtraitlets.config.configurablerjupyter_server.traittypesr Exceptionrrr:rVrrrrs#  66 \3 ' K % , *+===9)  7F7F 7Ft@)@)Fr.(r.e,+,sB BB