o i?M@sdZddlZddlZddlZddlmZmZmZmZm Z m Z ede Z ede Z zddlZWney=ddlmZYnwddlmZmZddlmZdd lmZdd lmZmZmZmZdd lmZdd l m!Z!Gd dde"Z#eGdddZ$GdddZ%GdddeZ&dS)zA base class session manager.N)AnyDictListNewTypeOptionalUnion KernelName ModelName)dbapi2) dataclassfields) ensure_async)web)Instance TraitErrorUnicodevalidate)LoggingConfigurable)InstanceFromClassesc@seZdZdZdS)KernelSessionRecordConflictzgException class to use when two KernelSessionRecords cannot merge because of conflicting data. N)__name__ __module__ __qualname____doc__rrj/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/jupyter_server/services/sessions/sessionmanager.pyrsrc@sNeZdZUdZdZeeed<dZeeed<de de fddZ d d d Z 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_idotherreturncCst|tr>|jo |j|jk}t|j|jk|jdup|jdug}t||gr(dSt|j|j|jk|j|jkgr>d}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) isinstancerrallranyr)selfr condition1 condition2msgrrr__eq__1s&     zKernelSessionRecord.__eq__cCszt|ts d}t||jr|jr|j|jkrd}t|t|D]}t||jr:t||jr:t ||jt||jq!dS)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$rr'fieldrrrupdateQs  zKernelSessionRecord.update)rrr N) rrrrrrstr__annotations__robjectboolr(r/rrrrr%s  rc@seZdZUdZeeed<defddZddZde ee fd e fd d Z d d Z de ee fd efddZded dfddZded dfddZdS)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. _recordsrecordscGsg|_|D]}||qdSzInitialize a record list.N)r5r/)r$r6recordrrr__init__ks z KernelSessionRecordList.__init__cC t|jS)z+The string representation of a record list.)r0r5r$rrr__str__q zKernelSessionRecordList.__str__r8r cCsHt|tr ||jvr dSt|tr"|jD] }||j|jfvr!dSqdS)z.Search for records by kernel_id and session_idTF)r!rr5r0rr)r$r8rrrr __contains__us  z$KernelSessionRecordList.__contains__cCr:)zThe length of the record list.)lenr5r;rrr__len__r=zKernelSessionRecordList.__len__cCslt|tr|jD]}||jks||jkr|Sqnt|tr-|jD] }||kr,|Sq"|d}t|)zsReturn a full KernelSessionRecord from a session_id, kernel_id, or incomplete KernelSessionRecord. z& not found in KernelSessionRecordList.)r!r0r5rrr ValueError)r$r8r>r'rrrgets     zKernelSessionRecordList.getNcCsDz|j|}|j||WdSty!|j|YdSw)z9Update a record in-place or append it if not in the list.N)r5indexr/rBappend)r$r8idxrrrr/s   zKernelSessionRecordList.updatecCs||jvr |j|dSdS)zYRemove a record if its found in the list. If it's not found, do nothing. N)r5remove)r$r8rrrrGs zKernelSessionRecordList.remove)rrrrrrr1r9r<rr0r3r?rArCr/rGrrrrr4`s   r4cseZdZdZedddjddZeddd Ze d Z e d d gZ fd dZ dZdZhdZeddZeddZddZddZddZdefddZ     d9deedeed eed!eed"eedeeeff d#d$Z d:deedeedeeeffd%d&Zd'edeedeed eed!eedef d(d)Z d;d*d+Z!d,d-Z"d.d/Z#d"ede$fd0d1Z%dr+rZtyperrcCs(|jdur|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 connectioncursorexecuter;rrrrgs  zSessionManager.cursorcCs,|jdurtj|jdd|_tj|j_|jS)zStart a database connectionN)isolation_level) _connectionsqlite3connectrMRow row_factoryr;rrrrfs  zSessionManager.connectioncCs"|jdur|jd|_dSdS)zClose the sqlite connectionN)recloser;rrrros   zSessionManager.closecCs |dS)z+Close connection once SessionManager closesN)ror;rrr__del__ zSessionManager.__del__csLd}|jd|f|j}|dur$|j|ddIdH}|dur$d}|S)z2Check to see if the session of a given name existsFz"SELECT * FROM session WHERE path=?NT)tolerate_culled)rgrhfetchone row_to_model)r$rZrTrowmodelrrrsession_existss zSessionManager.session_existsr cCs ttS)zCreate a uuid for a new session)r0uuiduuid4r;rrrnew_session_idrqzSessionManager.new_session_idrZr+rd kernel_namerc s|}t|d}|j||dur||jvrn ||||||IdH}||_|j||j|||||dIdH}|j||S)zCreates a session and returns its model Parameters ---------- name: ModelName(str) Usually the model name, like the filename associated with current kernel. rN)rZr+rdr) rzrr_r/kernel_managerstart_kernel_for_sessionr save_sessionrG) r$rZr+rdr{rrr8resultrrrcreate_session s        zSessionManager.create_sessioncCsB|dur|j|}tj||}t|tsJitjd|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_pathosrZjoinr!r0environ)r$rZr+cwdrrrget_kernel_env*s  zSessionManager.get_kernel_envrc sBt|jj|dIdH}|||}|jj|||dIdH}|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. )rZN)rZr{env)r contents_managerget_kernel_pathrr} start_kernel) r$rrZr+rdr{ kernel_path kernel_envrrrrr~=s  z'SessionManager.start_kernel_for_sessioncs0|jd|||||f|j|dIdH}|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 (?,?,?,?,?)r|N)rgrh get_session)r$rrZr+rdrrrrrr`s zSessionManager.save_sessionc s"|s d}t|g}|D]}||jvrd|}t||d|q dd|}|j|t|z|j}Wn t yId}Ynw|durng}| D]\}} ||d| qTt dd d |z | |IdH} W| St y} z t dd t| | d} ~ ww) 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)_columnsrErrgrhlistvaluesrsKeyErroritemsr HTTPErrorrtr0) r$rar' conditionscolumnqueryruqkeyrNrverrrrs:   zSessionManager.get_sessioncs~|j|dIdH|sdSg}|D]}||jvrtd||d|qdd|}|j|gt||dS)aUpdates the values in the session database. Changes the values of the session with the given session_id with the values from the keyword arguments. Parameters ---------- session_id : str a uuid that identifies a session in the sqlite3 database **kwargs : str the key must correspond to a column title in session database, and the value replaces the current value in the session with session_id. r|NzNo such column: %rrz(UPDATE session SET %s WHERE session_id=?r) rrr)rErrgrhrr)r$rrasetsrrrrrupdate_sessions  "zSessionManager.update_sessioncs ||jvS)zQChecks if the kernel is still considered alive and returns true if its not found.)r})r$rrrr kernel_culleds zSessionManager.kernel_culledFcst||dIdH}|r5|jd|dfdj|d|dd}|r1|j|ddSt|t|j |dIdH}|d|d|d |d |d }|d d krc|d|d d |d <|S)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...rZr+rd)idrZr+rdkernelnotebook)rZr+) r rrgrhformatlogwarningrr} kernel_model)r$rurrrr'rrvrrrrts.  zSessionManager.row_to_modelc sT|jd}g}|D]}z||IdH}||Wq ty'Yq w|S)z_Returns a list of dictionaries containing all the information from the session databasezSELECT * FROM sessionN)rgrhfetchallrtrEr)r$crrurvrrr list_sessionss   zSessionManager.list_sessionscsht|d}|j||j|dIdH}t|j|ddIdH|jd|f|j |dS)z=Deletes the row in the session database with given session_idr|Nrrr) rr_r/rr r}shutdown_kernelrgrhrG)r$rr8sessionrrrdelete_sessions  zSessionManager.delete_session)NNNNN)N)NNNN)F)*rrrrrtagrMrr]rr}rrr9rerjrpropertyrgrfrorprwr0rzrr rrrrrr~rrrr3rrtrr __classcell__rrrbrrHs      "   # 3 $rH)'rrrRrxtypingrrrrrrr0rr rk ImportError pysqlite2r dataclassesr r jupyter_core.utilsr tornador traitletsrrrrtraitlets.config.configurablerjupyter_server.traittypesr Exceptionrrr4rHrrrrs.         :C