o iv+@sdZddlZzddlZWneyddlmZYnwddlmZmZddl 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))SessionManagerznamepathtype kernel_id session_idcCs(|jdur|j|_|jd|jS)z5Start a cursor and create a database called 'session'NzNCREATE TABLE session (session_id, path, name, type, kernel_id))_cursor connectioncursorexecuteselfrd/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/services/sessions/sessionmanager.pyr+s   zSessionManager.cursorcCs&|jdurtd|_tj|j_|jS)zStart a database connectionNz:memory:) _connectionsqlite3connectRow row_factoryrrrrr4s   zSessionManager.connectioncCs"|jdur|jd|_dSdS)zClose the sqlite connectionN)rcloserrrrr<s   zSessionManager.closecCs |dS)z+Close connection once SessionManager closesN)rrrrr__del__B zSessionManager.__del__ccsRd}|jd|f|j}|dur$t|j|ddV}|dur$d}t|)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)ruuiduuid4rrrrnew_session_idWr zSessionManager.new_session_idccsV|}|dur||jvrn ||||||V}t|j|||||dV}t|)z'Creates a session and returns its modelN)r r rr)r+kernel_managerstart_kernel_for_sessionr save_sessionrr$)rr r r kernel_namerrresultrrrcreate_session[s zSessionManager.create_sessionccs0|jj|d}t|jj||dV}t|)z'Start a new kernel for a given session.)r )r r/)contents_managerget_kernel_pathrr, start_kernelrr$)rrr r rr/ kernel_pathrrrrr-is  z'SessionManager.start_kernel_for_sessionccs6|jd|||||ft|j|dV}t|)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)rrr get_sessionrr$)rrr r rrr0rrrr.ts   zSessionManager.save_sessionc ks$|stdg}|D]}||jvrtd|||dq dd|}|j|t|z|j }Wn t yGd}Ynw|durmg}| D]\}}||d|qRt dd d |z t||V} Wnt y} z t dd | d} ~ wwt| ) 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 queryzNo such column: %r=?zSELECT * FROM session WHERE z AND N=izSession not found: , ) TypeErrorkeys_columnsappendjoinrrlistvaluesr"KeyErroritemsr HTTPErrorrr#rr$) rkwargs conditionscolumnqueryr&qkeyvaluer'errrr7s6     zSessionManager.get_sessionckst|j|dV|sdSg}|D]}||jvr"td|||dqdd|d}|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. r6NzNo such column: r8zUPDATE session SET r:z WHERE session_id=?) rr7r<r=r;r>r?rrr@rA)rrrEsetsrGrHrrrupdate_sessions   zSessionManager.update_sessioncCs ||jvS)zRChecks if the kernel is still considered alive and returns true if its not found. )r,)rrrrr kernel_culleds zSessionManager.kernel_culledFccst||dV}|r5|jd|dfdj|d|dd}|r1|j|dtdt |t|j |dV}|d|d|d |d |d }|d d kra|d|d d |d <t|)z@Takes sqlite database session row and turns it into a dictionaryr&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...Nr r r)idr r rkernelnotebook)r r ) rrOrrformatlogwarningrr$rBr, kernel_model)rr&r!rOmsgrWr'rrrr#s.   zSessionManager.row_to_modelc csZ|jd}g}|D]}zt||V}||Wq ty'Yq wt|)z_Returns a list of dictionaries containing all the information from the session databasezSELECT * FROM session) rrfetchallrr#r>rBrr$)rcr0r&r'rrr list_sessions s    zSessionManager.list_sessionsccsBt|j|dV}t|j|ddV|jd|fdS)z=Deletes the row in the session database with given session_idr6rRrQrPN)rr7r,shutdown_kernelrr)rrsessionrrrdelete_sessionszSessionManager.delete_session)NNNNN)NNNN)F)__name__ __module__ __qualname__rr,r r2rrr=propertyrrrrr coroutiner(r+r1r-r.r7rNrOr#r[r^rrrrr sH        1  " r )__doc__r)r ImportError pysqlite2rtornadorrtraitlets.config.configurableripython_genutils.py3compatr traitletsrnotebook.utilsrnotebook.traittypesr r rrrrs