Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 10 additions & 6 deletions tests/smoke.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from rich.console import Console
from rich.table import Table

from wherobots.db import connect, connect_direct
from wherobots.db import connect, connect_direct, errors
from wherobots.db.constants import DEFAULT_ENDPOINT, DEFAULT_SESSION_TYPE
from wherobots.db.connection import Connection
from wherobots.db.region import Region
Expand Down Expand Up @@ -103,8 +103,12 @@ def execute(conn: Connection, sql: str) -> pandas.DataFrame:
cursor.execute(sql)
return cursor.fetchall()

with conn_func() as conn:
with concurrent.futures.ThreadPoolExecutor() as pool:
futures = [pool.submit(execute, conn, s) for s in args.sql]
for future in concurrent.futures.as_completed(futures):
render(future.result())
try:
with conn_func() as conn:
with concurrent.futures.ThreadPoolExecutor() as pool:
futures = [pool.submit(execute, conn, s) for s in args.sql]
for future in concurrent.futures.as_completed(futures):
render(future.result())
except errors.Error as e:
sys.stderr.write(f"\n{e}\n")
sys.exit(1)
10 changes: 9 additions & 1 deletion wherobots/db/driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,15 @@ def connect(
)
resp.raise_for_status()
except requests.HTTPError as e:
raise InterfaceError("Failed to create SQL session!", e)
details = str(e)
try:
info = e.response.json()
errors = info.get("errors", [])
if errors and isinstance(errors, list):
details = f"{errors[0]['message']}: {errors[0]['details']}"
except requests.JSONDecodeError:
pass
raise InterfaceError(f"Failed to create SQL session: {details}") from e

# At this point we've been redirected to /sql/session/{session_id}, which we'll need to keep polling until the
# session is in READY state.
Expand Down