Skip to content

Commit 5d817c5

Browse files
committed
feat: add additional_catalogs parameter to Client
1 parent 0b46662 commit 5d817c5

1 file changed

Lines changed: 13 additions & 0 deletions

File tree

src/altertable_flightsql/client.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,7 @@ def __init__(
108108
*,
109109
catalog: Optional[str] = "altertable",
110110
schema: Optional[str] = "main",
111+
additional_catalogs: Optional[Sequence[str]] = None,
111112
host: str = "flight.altertable.ai",
112113
port: int = 443,
113114
tls: bool = True,
@@ -121,6 +122,11 @@ def __init__(
121122
password: Altertable password (required).
122123
catalog: Default catalog name (default: "altertable").
123124
schema: Default schema name (default: "main").
125+
additional_catalogs: Server-side catalog names to make visible
126+
in this session in addition to ``catalog`` (default: None).
127+
Each name must be a catalog the authenticated user already
128+
has access to; read-only vs read-write is determined by the
129+
catalog's server-side configuration.
124130
host: Altertable server hostname (default: "flight.altertable.ai").
125131
port: Server port (default: 443).
126132
tls: Whether to use TLS/SSL (default: True).
@@ -156,6 +162,13 @@ def __init__(
156162
if schema:
157163
options["schema"] = sql_pb2.SessionOptionValue(string_value=schema)
158164

165+
if additional_catalogs:
166+
options["additional_catalogs"] = sql_pb2.SessionOptionValue(
167+
string_list_value=sql_pb2.SessionOptionValue.StringListValue(
168+
values=list(additional_catalogs),
169+
),
170+
)
171+
159172
if options:
160173
self._set_options(options)
161174

0 commit comments

Comments
 (0)