:vjP:dZddlZddlZddlZddlmZmZmZedeZ edeZ ddl Z ddlmZmZddlmZdd lmZdd lmZmZmZmZdd lmZdd lmZGd deZ eGddZ!GddZ"GddeZ#y#e $r ddl mZ YiwxYw)zA base class session manager.N)AnyNewTypecast 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__/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_server/services/sessions/sessionmanager.pyrrsrrcReZdZUdZdZedzed<dZedzed<dede 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) isinstancerrallranyr)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 TypeErrorrrr hasattrnamegetattrsetattr)r&r r)fields rupdatezKernelSessionRecord.updateOs%!45GCC. ??t~~%//T^^2S^C-c2 2D\Euejj)geUZZ.Hejj'%*DE"r)r rr!N) rrrrrstr__annotations__robjectboolr*r2rrrrr#s?"Jd ! IsTz Ft@ FrrceZdZUdZeeed<defdZdZdee zde fdZ d Z dee zdefd Z dedd 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)r9r2)r&r:records r__init__z KernelSessionRecordList.__init__is F KK rc,t|jS)z+The string representation of a record list.)r3r9r&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#rr9r3rr)r&r=rs r __contains__z$KernelSessionRecordList.__contains__ssP f1 2v7N fc "]]allAKK88#rc,t|jS)zThe length of the record list.)lenr9r@s r__len__zKernelSessionRecordList.__len__~rBrct|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#r3r9rrr ValueError)r&r=rDr)s rgetzKernelSessionRecordList.getsv 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)r9indexr2rJappend)r&r=idxs rr2zKernelSessionRecordList.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)r9remove)r&r=s rrQzKernelSessionRecordList.removes' T]] " MM  ( #r)rrrrlistrr4r>rAr3r6rErHrKr2rQrrrr8r8^s&'' !4 " #6#<  " -3 8K )0)T))0)T)rr8ceZdZdZeddj dZeddZe d Z e d d gZ fd Z d Zd ZhdZedZedZdZdZdZdefdZ d&ded zded zded zded zded zdeeeff dZd'ded zded zdeeeffdZ d'deded zded zded zded zded zdefdZd(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.valuerUzL`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>r8_pending_sessions)r&argskwargs __class__s rr>zSessionManager.__init__s  $)&)!8!:rN>r.rftyperrc|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 connectioncursorexecuter@s rrtzSessionManager.cursorsE << ??113DL LL = ||rc|jEtj|jd|_tj|j_|jS)zStart a database connectionN)isolation_level) _connectionsqlite3connectrYRow row_factoryr@s rrszSessionManager.connectionsH    #&t/E/EW[\D +2;;D   (rc`|j"|jjd|_yy)zClose the sqlite connectionN)rrcloser@s rr~zSessionManager.closes( << # LL   DL $rc$|jy)z+Close connection once SessionManager closesN)r~r@s 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)rtrufetchone row_to_model)r&rfr`rowmodels rsession_existszSessionManager.session_existssi @4'Jkk""$ ? ++C+FFE  GsAA!A A!r!c<ttjS)zCreate a uuid for a new session)r3uuiduuid4r@s rnew_session_idzSessionManager.new_session_ids4::<  rrfr.rp kernel_namercK|j}t|}|jj||||jvrn|j ||||||d{}||_|jj||j|||||d{}|jj|td|S7k7-w)zCreates a session and returns its model Parameters ---------- name: ModelName(str) Usually the model name, like the filename associated with current kernel. rN)rrfr.rprr)rfr.rprzdict[str, Any]) rrrlr2kernel_managerstart_kernel_for_sessionr save_sessionrQr) r&rfr.rprrrr=results rcreate_sessionzSessionManager.create_sessions ((* $ ; %%f-  Y$2E2E%E ";;%'# <I% %%f-(( T49)   %%f-$f-- s$A#C%C&?C%C&,CCc|;|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_pathosrfjoinr#r3environ)r&rfr.cwds rget_kernel_envzSessionManager.get_kernel_env-s[  %%2248C77<<T*D$$$$7"**70$77rrcKt|jj|d{}|j||}|jj ||||d{}t d|S7I7w)atStart 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. kernel_id : str, optional client-supplied UUID to register the new kernel under. When provided, forwarded to ``kernel_manager.start_kernel`` so the kernel is created at that exact id. If omitted, the kernel manager mints a fresh id. )rfN)rfrenvrr3)r contents_managerget_kernel_pathrr start_kernelr) r&rrfr.rprr kernel_path kernel_envs rrz'SessionManager.start_kernel_for_session>s>))>)>)N)NTX)N)YZZ ((t4 --::# ;  E9%%[ s!)A9A58A9$A7%A97A9cK|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)rtru get_session)r&rrfr.rprrs rrzSessionManager.save_sessionhsM2  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,_columnsrNrrtrurRvaluesrKeyErroritemsr HTTPErrorrr3) r&rnr) conditionscolumnqueryrqkeyr[res rrzSessionManager.get_sessionsi$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&rs 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 dictionaryrN&DELETE FROM session WHERE session_id=?rzKernel '{kernel_id}' appears to have been culled or died unexpectedly, invalidating session '{session_id}'. The session has been removed.)rrz Continuing...rfr.rp)idrfr.rpkernelnotebook)rfr.) r rrtruformatlogwarningrr 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)rtrufetchallrrNr)r&crrrs r list_sessionszSessionManager.list_sessionssu KK   7 8::>wx?PQU?VWXXX DzmT %%f-@Xs"=B3B/1B31B12>B31B3)NNNNN)N)NNNN)F))rrrrrtagrYrrir rrrr>rrrxrpropertyrtrsr~rrr3rrrdictrrrrrrrr6rrrr __classcell__)ros@rrTrTs  O cc!"#(bcN* F @ ; GKBH     !!  !%)- $ $.Dj$.$$.Dj $.  $& $. : $. c3h$.L83:8Y5E8QUVY[^V^Q_80!%(&(&Dj(&$ (& Dj (&  $& (&:(& (&T@1f"eH4S4T4"H .rrT)$rrr^rtypingrrrr3rrry ImportError pysqlite2r dataclassesr r jupyter_core.utilsr tornador traitletsr rrrtraitlets.config.configurablerjupyter_server.traittypesr Exceptionrrr8rTrrrrs#  %% \3 ' K % , *+===9)  7F7F 7Ft@)@)F|.(|.e,+,sB BB