|
@@ -32,14 +32,12 @@ LOG = logging.getLogger(__name__)
|
|
|
class SessionExpired(Exception):
|
|
class SessionExpired(Exception):
|
|
|
pass
|
|
pass
|
|
|
|
|
|
|
|
-
|
|
|
|
|
class QueryExpired(Exception):
|
|
class QueryExpired(Exception):
|
|
|
pass
|
|
pass
|
|
|
|
|
|
|
|
class AuthenticationRequired(Exception):
|
|
class AuthenticationRequired(Exception):
|
|
|
pass
|
|
pass
|
|
|
|
|
|
|
|
-
|
|
|
|
|
class QueryError(Exception):
|
|
class QueryError(Exception):
|
|
|
def __init__(self, message):
|
|
def __init__(self, message):
|
|
|
self.message = message
|
|
self.message = message
|
|
@@ -115,7 +113,11 @@ def get_api(user, snippet, fs, jt):
|
|
|
|
|
|
|
|
|
|
|
|
|
def _get_snippet_session(notebook, snippet):
|
|
def _get_snippet_session(notebook, snippet):
|
|
|
- return [session for session in notebook['sessions'] if session['type'] == snippet['type']][0]
|
|
|
|
|
|
|
+ session = [session for session in notebook['sessions'] if session['type'] == snippet['type']]
|
|
|
|
|
+ if not session:
|
|
|
|
|
+ raise SessionExpired()
|
|
|
|
|
+ else:
|
|
|
|
|
+ return session[0]
|
|
|
|
|
|
|
|
|
|
|
|
|
# Base API
|
|
# Base API
|