MhNpdZddlZddlZddlZddlmZmZmZmZm Z ede Z ede Z ddl Z n#e$r ddlmZ YnwxYwddlmZmZddlmZdd lmZdd lmZmZmZmZdd lmZdd lm Z Gd de!Z"eGddZ#GddZ$GddeZ%dS)zA base class session manager.N)AnyNewTypeOptionalUnioncast KernelName ModelName)dbapi2) dataclassfields) ensure_async)web)Instance TraitErrorUnicodevalidate)LoggingConfigurable)InstanceFromClassesceZdZdZdS)KernelSessionRecordConflictzgException class to use when two KernelSessionRecords cannot merge because of conflicting data. N)__name__ __module__ __qualname____doc__o/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/jupyter_server/services/sessions/sessionmanager.pyrrsrrc`eZdZUdZdZeeed<dZeeed<de de fdZ d dZ dS) 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_idotherreturncjt|tr|jo|j|jk}t|j|jk|jdup|jdug}t ||grdSt|j|j|jk|j|jkgrd}t |dS)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 1 1 7MDNeo,MJOu'77Nd*Eeo.EJ J +,, tOOu'77Neo5 72#666urc zt|tsd}t||jr(|jr!|j|jkrd}t |t |D]U}t ||jr>t||jr)t||jt||jVdS)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%!455 !GCC.. ? 3t~ 3%/T^2S2S^C-c22 2D\\ F FEuej)) FgeUZ.H.H Fej'%*D*DEEE F Fr)r"rr#N) rrrrr rstr__annotations__r!objectboolr,r4rrrrr#s!%J $$$#Ix}###Ft@ F F F F F FrrceZdZUdZeeed<defdZdZde ee fde fdZ d Z de ee fdefd Zdedd fd Zdedd fd Zd S)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. _recordsrecordscHg|_|D]}||dSzInitialize a record list.N)r;r4)r(r<records r__init__z KernelSessionRecordList.__init__is7   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 ||jvrdSt|tr|jD]}||j|jfvrdSdS)z.Search for records by kernel_id and session_idTF)r%rr;r5r r!)r(r?rs r __contains__z$KernelSessionRecordList.__contains__ssn f1 2 2 v7N7N4 fc " " ]  alAK888449urc*t|jS)zThe length of the record list.)lenr;rBs r__len__zKernelSessionRecordList.__len__~rDrct|tr|jD]}||j|jfvr|cSn)t|t r|jD] }||kr|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 " " "]  ak1<888HHH9  3 4 4 "] " "Q;;!MMM???oorNc |j|}|j||dS#t$r|j|YdSwxYw)z9Update a record in-place or append it if not in the list.N)r;indexr4rLappend)r(r?idxs rr4zKernelSessionRecordList.updatesw )-%%f--C M#  % %f - - - - - ) ) ) M  ( ( ( ( ( ( )s:>$A&%A&cP||jvr|j|dSdS)zYRemove a record if its found in the list. If it's not found, do nothing. N)r;remove)r(r?s rrSzKernelSessionRecordList.removes5 T] " " M  ( ( ( ( ( # "r)rrrrlistrr6r@rCrr5r8rGrJrMr4rSrrrr:r:^s&'''' !4    """ 5)Q    )0)T)))))0)T))))))rr:cVeZdZdZedddZeddZe 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_filepathc|d}|dkr|Stj|}|r|rd}t |t |d5}|d}dddn #1swxYwY|ds|dkrd }t ||S) 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 oo%eT"" %a % % % % % % % % % % % % % % %$$%788 &Vs]]F oo% s.BBBzBjupyter_server.services.kernels.kernelmanager.MappingKernelManagerz8jupyter_server.services.contents.manager.ContentsManagerz2notebook.services.contents.manager.ContentsManagerc`tj|i|t|_dSr>)superr@r:_pending_sessions)r(argskwargs __class__s rr@zSessionManager.__init__s3$)&)))!8!:!:rN>r0rhtyper!r c|j8|j|_|jd|jS)z5Start a cursor and create a database called 'session'Nz\CREATE TABLE IF NOT EXISTS session (session_id, path, name, type, kernel_id))_cursor connectioncursorexecuterBs rrvzSessionManager.cursorsJ < ?1133DL L =   |rc|j6tj|jd|_tj|j_|jS)zStart a database connectionN)isolation_level) _connectionsqlite3connectr[Row row_factoryrBs rruzSessionManager.connections>   #&t/EW[\\\D +2;D  (rcX|j"|jd|_dSdS)zClose the sqlite connectionN)rtcloserBs rrzSessionManager.closes1 < # L   DLLL $ #rc.|dS)z+Close connection once SessionManager closesN)rrBs r__del__zSessionManager.__del__s rcKd}|jd|f|j}|!||dd{V}|d}|S)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_existssz @4'JJJk""$$ ? ++C+FFFFFFFFE  rr#cBttjS)zCreate a uuid for a new session)r5uuiduuid4rBs rnew_session_idzSessionManager.new_session_ids4:<<   rrhr0rr kernel_namer!cK|}t|}|j|| ||jvrn||||||d{V}||_|j|||||||d{V}|j|ttttf|S)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$2E%E%E ";;D$kI% %%f---(( T49)         %%f---DcNF+++rc|:|j|}tj||}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. NJPY_SESSION_NAME)r cwd_for_pathosrhjoinr%r5environ)r(rhr0cwds rget_kernel_envzSessionManager.get_kernel_env(sc  %22488C7<<T**D$$$$$$7"*70$777rr cKt|j|d{V}|||}|j|||d{V}t t|S)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)Y)YZZZZZZZZ ((t44 -::#;        C###rcK|jd|||||f||d{V}|S)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^sb2  4 tT9 5   '':'>>>>>>>> rcK|sd}t|g}|D]7}||jvrd|}t||d|z8dd|z}|j|t | |j}n#t$rd}YnwxYw|ag}| D]\}} ||d|  tj dd d |z | |d{V} n8#t$r+} tj dd t| z| d} ~ wwxYw| S) 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~s$ !2CC..   / /FT]**111nn$   fvo . . . .0GLL4L4LM E4 #8#8999 +&&((CC   CCC  ;A$llnn - - UC++%++,,,,-%< ! %MNN N N++C00000000EE N N N-%>> KK ( ( ( (:diiooN E#GT&--//%:%:#GJ#GHHH 4& 5 5 e K  NQ[P\   %)K$8$8$:$: !D$   * *YDDWDWX\^bDcDc * d d d d d  e ercK||jvS)zQChecks if the kernel is still considered alive and returns true if its not found.)r)r(r!s r kernel_culledzSessionManager.kernel_culleds 333rFc:Kt||dd{V}|ru|jd|dfd|d|d}|r|j|ddSt|t|j |dd{V}|d|d|d |d |d }|d d kr|d|d d |d <|S)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_modelsa$01C1CC DT1U1U$V$VVVVVVV   K   H3|K\J^ _ _ _UU[U[!+.3|;LV\VV     C!8!8!8999t3-- )$*=*J*J3{K[*\*\]]]]]]]] l#KKK"    v;* $ $),Vc&k J JE*  rcK|jd}g}|D]C} ||d{V}||4#t $rY@wxYw|S)z_Returns a list of dictionaries containing all the information from the session databasezSELECT * FROM sessionN)rvrwfetchallrrPr)r(crrrs r list_sessionszSessionManager.list_sessionss K   7 8 8::<<  C "//44444444 e$$$$     s0A'' A43A4ctKt|}|j|||d{V}t |j|ddd{V|jd|f|j |dS)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?VWWXXXXXXXXX DzmTTT %%f-----r)NNNNN)N)NNNN)F)*rrrrrtagr[rrkrrrrr@rtrzrpropertyrvrurrrr5rrr rrrrrrrrrr8rrrr __classcell__)rqs@rrVrVs  O ccX!""#"(XbccN** F @ ;;;;; GKBBBH X  X     !!!!! #$(",0#' ,,sm,y!,sm , j) , C= , c3h,,,,D@D88SM8)1))<8 c3h8888&!$!$sm!$y! !$ sm !$ j) !$ !$!$!$!$F@111f"e"e"eH4S4T4444""""H   .......rrV)&rrr`rtypingrrrrrr5rr r{ ImportError pysqlite2r dataclassesr r jupyter_core.utilsr tornador traitletsrrrrtraitlets.config.configurablerjupyter_server.traittypesr Exceptionrrr:rVrrrrs##  66666666666666 W\3 ' ' GK % % ,NNNN,,,++++++++,*)))))))++++++==================999999)  7F7F7F7F7F7F7F 7Ft@)@)@)@)@)@)@)@)Fr.r.r.r.r.(r.r.r.r.r.s; A A