o ^iN@sdZddlZddlZddlZddlmZmZmZmZm Z ede Z ede Z zddl Z Wney;ddlmZ YnwddlmZmZddlmZdd lmZdd lmZmZmZmZdd lmZdd lm Z Gd dde!Z"eGdddZ#GdddZ$GdddeZ%dS)zA base class session manager.N)AnyNewTypeOptionalUnioncast 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/home/ubuntu/hpml_nyu/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__/s&     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&fieldrrrupdateOs  zKernelSessionRecord.update)rrrN) 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)r4r.)r#r5recordrrr__init__is z KernelSessionRecordList.__init__cC t|jS)z+The string representation of a record list.)r/r4r#rrr__str__o zKernelSessionRecordList.__str__r7rcCsHt|tr ||jvr dSt|tr"|jD] }||j|jfvr!dSqdS)z.Search for records by kernel_id and session_idTF)r rr4r/rr)r#r7rrrr __contains__ss  z$KernelSessionRecordList.__contains__cCr9)zThe length of the record list.)lenr4r:rrr__len__~r<zKernelSessionRecordList.__len__cCsht|tr|jD]}||j|jfvr|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 r/r4rrr ValueError)r#r7r=r&rrrgets     zKernelSessionRecordList.getNcCsDz|j|}|j||WdSty!|j|YdSw)z9Update a record in-place or append it if not in the list.N)r4indexr.rAappend)r#r7idxrrrr.s   zKernelSessionRecordList.updatecCs||jvr |j|dSdS)zYRemove a record if its found in the list. If it's not found, do nothing. N)r4remove)r#r7rrrrFs zKernelSessionRecordList.remove)rrrrlistrr0r8r;rr/r2r>r@rBr.rFrrrrr3^s   r3cseZdZdZedddjddZeddd 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_existsrcCs ttS)zCreate a uuid for a new session)r/uuiduuid4r:rrrnew_session_idrqzSessionManager.new_session_idrZr*rd kernel_namerc s|}t|d}|j||dur||jvrn ||||||IdH}||_|j||j|||||dIdH}|j|t t t t f|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_sessionrFrdictr/r) r#rZr*rdr{rrr7resultrrrcreate_sessions        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 r/environ)r#rZr*cwdrrrget_kernel_env(s  zSessionManager.get_kernel_envrc sHt|jj|dIdH}|||}|jj|||dIdH}tt|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_kernelrr/) 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(_columnsrDrrgrhrGvaluesrsKeyErroritemsr HTTPErrorrtr/) r#rar& conditionscolumnqueryruqkeyrNrverrrr~s:   zSessionManager.get_sessionc s|j|dIdH|sdSg}|D]}||jvrtd||d|qdd|}|j|gt||t |j dra|jd|g|j \}}}|j j || ||d dSdS) 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 update_envz