a D b+@sdZddlZz ddlZWney6ddlmZYn0ddlmZmZddl m Z ddl m Z ddl mZddlmZdd lmZGd d d e ZdS) zA base class session manager.N)dbapi2)genweb)LoggingConfigurable) unicode_type)Instance) maybe_future)InstanceFromClassesc@seZdZedZeddgdZdZdZhdZ e ddZ e d d Z d d Z d dZejddZddZejd&ddZejddZejd'ddZejddZejddZddZejd(d d!Zejd"d#Zejd$d%ZdS))SessionManagerz session_idpathtype kernel_idnamecCs(|jdur"|j|_|jd|jS)z5Start a cursor and create a database called 'session'NzNCREATE TABLE session (session_id, path, name, type, kernel_id))_cursor connectioncursorexecuteselfrHlib/python3.9/site-packages/notebook/services/sessions/sessionmanager.pyr+s   zSessionManager.cursorcCs&|jdur td|_tj|j_|jS)zStart a database connectionNz:memory:) _connectionsqlite3ZconnectZRowZ row_factoryrrrrr4s   zSessionManager.connectioncCs|jdur|jd|_dS)zClose the sqlite connectionN)rcloserrrrr<s  zSessionManager.closecCs |dS)z+Close connection once SessionManager closesN)rrrrr__del__BszSessionManager.__del__ccsTd}|jd|f|j}|durFt|j|ddV}|durFd}t|dS)z2Check to see if the session of a given name existsFz"SELECT * FROM session WHERE path=?NT)tolerate_culled)rrfetchoner row_to_modelrReturn)rr existsrowmodelrrrsession_existsFs zSessionManager.session_existscCs ttS)zCreate a uuid for a new session)ruuidZuuid4rrrrnew_session_idWszSessionManager.new_session_idccsX|}|dur||jvrn||||||V}t|j|||||dV}t|dS)z'Creates a session and returns its modelN)r rr r)r%kernel_managerstart_kernel_for_sessionr save_sessionrr)rr rr kernel_namerr resultrrrcreate_session[szSessionManager.create_sessionccs2|jj|d}t|jj||dV}t|dS)z'Start a new kernel for a given session.)r )r r)N)contents_managerZget_kernel_pathrr&Z start_kernelrr)rr r rr r)Z kernel_pathrrrrr'is z'SessionManager.start_kernel_for_sessionccs8|jd|||||ft|j|dV}t|dS)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: string 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)rrr get_sessionrr)rr r rr rr*rrrr(ts  zSessionManager.save_sessionc ks.|s tdg}|D]&}||jvr0td||d|qdd|}|j|t|z|j }Wnt yd}Yn0|durg}| D]\}}|d||fqt dd d |zt||V} Wn:t y} z t dd t| WYd} ~ n d} ~ 00t| dS) aFReturns 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 : keyword argument 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 queryNo such column: %r%s=?zSELECT * FROM session WHERE %sz AND Nz%s=%rizSession not found: %s, ) TypeErrorkeys_columnsappendjoinrrlistvaluesrKeyErroritemsrZ HTTPErrorrrstrrr) rkwargsZ conditionscolumnqueryr!qkeyvaluer"errrr.s.     *zSessionManager.get_sessioncks~t|j|dV|sdSg}|D](}||jvr@td||d|q&dd|}|j|t | |gdS)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-Nr/r0z(UPDATE session SET %s WHERE session_id=?r1) rr.r3r4r2r5r6rrr7r8)rr r<Zsetsr=r>rrrupdate_sessions   zSessionManager.update_sessioncCs ||jvS)zRChecks if the kernel is still considered alive and returns true if its not found. )r&)rrrrr kernel_culledszSessionManager.kernel_culledFccst||dV}|rh|jd|dfdj|d|dd}|r`|j|dtdt |t|j |dV}|d|d|d |d |d }|d d kr|d|d d |d <t|dS)z@Takes sqlite database session row and turns it into a dictionaryr&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.)rr z Continuing...Nr rr )idr rr kernelZnotebook)r r) rrDrrformatlogZwarningrrr9r& kernel_model)rr!rrDmsgrJr"rrrrs,   zSessionManager.row_to_modelc cs^|jd}g}|D]6}zt||V}||WqtyLYq0qt|dS)z_Returns a list of dictionaries containing all the information from the session databasezSELECT * FROM sessionN) rrZfetchallrrr5r9rr)rcr*r!r"rrr list_sessions s   zSessionManager.list_sessionsccs@t|j|dV}t|j|ddV|jd|fdS)z=Deletes the row in the session database with given session_idr-rGrFrEN)rr.r&Zshutdown_kernelrr)rr Zsessionrrrdelete_sessionszSessionManager.delete_session)NNNNN)NNNN)F)__name__ __module__ __qualname__rr&r r,rrr4propertyrrrrr coroutiner#r%r+r'r(r.rCrDrrMrNrrrrr sF        1  " r )__doc__r$r ImportErrorZ pysqlite2rZtornadorrZtraitlets.config.configurablerZipython_genutils.py3compatrZ traitletsrZnotebook.utilsrZnotebook.traittypesr r rrrrs