ó ù2¢\c@s²dZddlZddlZyddlZWn!ek rQddlmZnXddlmZm Z ddl m Z ddl m Z ddlmZde fd „ƒYZdS( sA base class session manager.iÿÿÿÿN(tdbapi2(tgentweb(tLoggingConfigurable(t unicode_type(tInstancetSessionManagercBseZedƒZedƒZdZdZdddddhZe d„ƒZ e d„ƒZ d „Z d „Z d „Zd „Zejdddddd „ƒZejd„ƒZddddd„Zd„Zd„Zd„Zd„Zejd„ƒZRS(s<notebook.services.kernels.kernelmanager.MappingKernelManagers2notebook.services.contents.manager.ContentsManagert session_idtpathtnamettypet kernel_idcCs;|jdkr4|jjƒ|_|jjdƒn|jS(s5Start a cursor and create a database called 'session'sOCREATE TABLE session (session_id, path, name, type, kernel_id)N(t_cursortNonet connectiontcursortexecute(tself((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pyR s   cCs:|jdkr3tjdƒ|_tj|j_n|jS(sStart a database connections:memory:N(t _connectionR tsqlite3tconnecttRowt row_factory(R((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pyR)scCs,|jdk r(|jjƒd|_ndS(sClose the sqlite connectionN(R R tclose(R((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pyR1s cCs|jƒdS(s+Close connection once SessionManager closesN(R(R((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pyt__del__7scCs=|jjd|fƒ|jjƒ}|dkr5tStSdS(s2Check to see if the session of a given name existss"SELECT * FROM session WHERE path=?N(RRtfetchoneR tFalsetTrue(RRtreply((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pytsession_exists;s  cCsttjƒƒS(sCreate a uuid for a new session(Rtuuidtuuid4(R((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pytnew_session_idDsc csŠ|jƒ}|dk r*||jkr*n|j|||||ƒV}tj|j|d|d|d|d|ƒƒV}tj|ƒ‚dS(s'Creates a session and returns its modelRR R R N(R R tkernel_managertstart_kernel_for_sessionRt maybe_futuret save_sessiontReturn(RRR R t kernel_nameR Rtresult((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pytcreate_sessionHs +ccsM|jjd|ƒ}tj|jjd|d|ƒƒV}tj|ƒ‚dS(s'Start a new kernel for a given session.RR&N(tcontents_managertget_kernel_pathRR#R!t start_kernelR%(RRRR R R&t kernel_pathR ((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pyR"VscCs2|jjd|||||fƒ|jd|ƒS(süSaves 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 s&INSERT INTO session VALUES (?,?,?,?,?)R(RRt get_session(RRRR R R ((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pyR$as c Ks8|stdƒ‚ng}xF|jƒD]8}||jkrOtd|ƒ‚n|jd|ƒq(Wddj|ƒ}|jj|t|jƒƒƒy|jj ƒ}Wnt k rÂd }nX|d kr+g}x1|j ƒD]#\}}|jd||fƒqâWt jddd j|ƒƒ‚n|j|ƒS( sOReturns 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. smust specify a column to querysNo such column: %rs%s=?sSELECT * FROM session WHERE %ss AND s%s=%ri”uSession not found: %ss, N(t TypeErrortkeyst_columnstappendtjoinRRtlisttvaluesRtKeyErrorR titemsRt HTTPErrort row_to_model( Rtkwargst conditionstcolumntquerytrowtqtkeytvalue((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pyR-s&   "cKs§|jd|ƒ|sdSg}xG|jƒD]9}||jkrUtd|ƒ‚n|jd|ƒq-Wddj|ƒ}|jj|t|j ƒƒ|gƒdS(sUpdates 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. RNsNo such column: %rs%s=?s(UPDATE session SET %s WHERE session_id=?s, ( R-R/R0R.R1R2RRR3R4(RRR9tsetsR;R<((sHlib/python2.7/site-packages/notebook/services/sessions/sessionmanager.pytupdate_session¬scCs¶|d|jkr6|jjd|dfƒt‚ni|dd6|dd6|dd6|dd6|jj|dƒd6}|dd kr²i|dd6|dd6|d s