Ibv+dZddlZ ddlZddlmZmZddl m Z ddl m Z ddl mZddlmZdd lmZGd d e Zy#e$r ddlmZY?wxYw) zA base class session manager.N)dbapi2)genweb)LoggingConfigurable) unicode_type)Instance) maybe_future)InstanceFromClassesceZdZedZeddgZdZdZhdZ e dZ e dZ d Z d Zej d Zd Zej dd Zej dZej ddZej dZej dZdZej ddZej dZej dZy)SessionManagerznamepathtype kernel_id session_idc|j:|jj|_|jjd|jS)z5Start a cursor and create a database called 'session'zNCREATE TABLE session (session_id, path, name, type, kernel_id))_cursor connectioncursorexecuteselfs K/usr/lib/python3/dist-packages/notebook/services/sessions/sessionmanager.pyrzSessionManager.cursor+sC << ??113DL LL "= >||c|j9tjd|_tj|j_|jS)zStart a database connectionz:memory:) _connectionsqlite3connectRow row_factoryrs rrzSessionManager.connection4s?    #&z:D +2;;D   (rc`|j"|jjd|_yy)zClose the sqlite connectionN)rclosers rr#zSessionManager.close<s( << # LL   DL $rc$|jy)z+Close connection once SessionManager closesN)r#rs r__del__zSessionManager.__del__Bs  rc#Kd}|jjd|f|jj}|"t|j |d}|d}t j |w)z2Check to see if the session of a given name existsFz"SELECT * FROM session WHERE path=?T)tolerate_culled)rrfetchoner row_to_modelrReturn)rrexistsrowmodels rsession_existszSessionManager.session_existsFst @4'Jkk""$ ? 't'8'8d'8'STTE jj  sA3A5c<ttjS)zCreate a uuid for a new session)ruuiduuid4rs rnew_session_idzSessionManager.new_session_idWsDJJL))rc #K|j}|||jvrn|j|||||}t|j |||||}t j |w)z'Creates a session and returns its model)rrrr)r2kernel_managerstart_kernel_for_sessionr save_sessionrr*)rrrr kernel_namerrresults rcreate_sessionzSessionManager.create_session[s((*  Y$2E2E%E ";;JdTXZeffI#   jt$TU^  _  jj  sA/A1c#K|jj|}t|jj ||}t j |w)z'Start a new kernel for a given session.)r)rr7)contents_managerget_kernel_pathr r4 start_kernelrr*)rrrrrr7 kernel_pathrs rr5z'SessionManager.start_kernel_for_sessionisZ++;;;F &    , ,+; , W  jj##sAAc#K|jjd|||||ft|j|}t j |w)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*)rrrrrrr8s rr6zSessionManager.save_sessiontsW4 D tT9 5 $D$4$4 $4$KLLjj  sAAc+K|s tdg}|jD]0}||jvr td||j|d2ddj |}|j j |t|j |j j}|Xg}|jD]\}}|j|d|tjdd d j | t|j|} t!j"| #t$rd}YwxYw#t$r} tjdd | d} ~ wwxYww) 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 HTTPErrorr r)rr*) rkwargs conditionscolumnqueryr,qkeyvaluer-es rrAzSessionManager.get_sessions|&<= = kkm -FT]]* 4f==   m , - /w||J/G.HI E4 #89 ++&&(C ;A$lln - UC5%+, ---':499Q<.%IJ J @&t'8'8'=>>Ejj C  @--':1#%>? ? @sOBE>E7AE>E.E> EE>EE> E;E66E;;E>c+rKt|j||syg}|jD]2}||jvrt d||j |d4ddj |d}|jj|t|j|gzyw)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: rCzUPDATE session SET rEz WHERE session_id=?) r rArGrHrFrIrJrrrKrL)rrrPsetsrRrSs rupdate_sessionzSessionManager.update_sessions 4++z+BCC kkm 'FT]]*"26* =>> KK6(" & '&diio%66IJ E4 #8J<#GHsB5B7c||jvS)zRChecks if the kernel is still considered alive and returns true if its not found. )r4)rrs r kernel_culledzSessionManager.kernel_culleds 3 333rc#Kt|j|d}|ry|jjd|dfdj |d|d}|r3|j j |dztjdt|t|jj|d}|d|d|d |d |d }|d d k(r|d|d d |d <tj|w)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...Nrrr)idrrrkernelnotebook)rr) r r\rrformatlogwarningrr*rMr4 kernel_model)rr,r'r\msgrer-s rr)zSessionManager.row_to_models!+4+=+=c+>N+OPP  KK   H!$\!2 4 6W[!1S=NO   '8!89jj&&3- )$*=*=*J*J3{K[*\]] l#KKK"   v;* $),Vc&k JE* jjsDDc#K|jjd}g}|jD]0} t|j |}|j |2tj|#t $rYSwxYww)z_Returns a list of dictionaries containing all the information from the session databasezSELECT * FROM session) rrfetchallr r)rIrMrr*)rcr8r,r-s r list_sessionszSessionManager.list_sessions s KK   7 8::< C *4+<+>wx?PQU?VWXX DzmTsA(A*)NNNNN)NNNN)F)__name__ __module__ __qualname__rr4r r;rrrHpropertyrrr#r%r coroutiner.r2r9r5r6rArZr\r)rjrnrrr r sO\]N* @ G  GKBH     ]]!! * ]] ! ! ]]$$ ]]!!> ]]/ / b ]]II:4 ]]    D ]] ! ! ]]UUrr )__doc__r0r ImportError pysqlite2rtornadorrtraitlets.config.configurableripython_genutils.py3compatr traitletsrnotebook.utilsr notebook.traittypesr r rtrrr~sN#  , =3'3IU(IU,+,s> A  A