add explicit SCHEMA and CATALOG

Author: leo-altertable

examples/client_usage.py

@@ -14,8 +14,18 @@
 ALTERTABLE_PORT = int(os.getenv("ALTERTABLE_PORT", "443"))
 ALTERTABLE_USERNAME = os.getenv("ALTERTABLE_USERNAME")
 ALTERTABLE_PASSWORD = os.getenv("ALTERTABLE_PASSWORD")
+ALTERTABLE_CATALOG = os.getenv("ALTERTABLE_CATALOG")
+ALTERTABLE_SCHEMA = os.getenv("ALTERTABLE_SCHEMA")
 ALTERTABLE_TLS = os.getenv("ALTERTABLE_TLS", "true").lower() == "true"
 
+CONNECTION_SETTINGS = {
+    "host": ALTERTABLE_HOST,
+    "port": ALTERTABLE_PORT,
+    "catalog": ALTERTABLE_CATALOG,
+    "schema": ALTERTABLE_SCHEMA,
+    "tls": ALTERTABLE_TLS,
+}
+
 
 def example_basic_query():
     """Execute a simple SQL query."""
@@ -27,12 +37,8 @@ def example_basic_query():
     with Client(
         username=ALTERTABLE_USERNAME,
         password=ALTERTABLE_PASSWORD,
-        host=ALTERTABLE_HOST,
-        port=ALTERTABLE_PORT,
-        tls=ALTERTABLE_TLS,
-        # Uncomment to use a specific catalog and schema
-        # catalog = "my_catalog",
-        # schema = "my_schema"
+        # Optional settings (default should work for most cases)
+        **CONNECTION_SETTINGS,
     ) as client:
         # Execute query
         reader = client.execute("SELECT * FROM users WHERE age > 18")
@@ -54,9 +60,7 @@ def example_updates():
     with Client(
         username=ALTERTABLE_USERNAME,
         password=ALTERTABLE_PASSWORD,
-        host=ALTERTABLE_HOST,
-        port=ALTERTABLE_PORT,
-        tls=ALTERTABLE_TLS,
+        **CONNECTION_SETTINGS,
     ) as client:
         # Insert
         rows = client.execute_update("INSERT INTO users (name, age) VALUES ('Alice', 25)")
@@ -82,9 +86,7 @@ def example_prepared_statement():
     with Client(
         username=ALTERTABLE_USERNAME,
         password=ALTERTABLE_PASSWORD,
-        host=ALTERTABLE_HOST,
-        port=ALTERTABLE_PORT,
-        tls=ALTERTABLE_TLS,
+        **CONNECTION_SETTINGS,
     ) as client:
         # Prepare statement
         with client.prepare("SELECT * FROM users WHERE id = ?") as stmt:
@@ -107,9 +109,7 @@ def example_transactions():
     with Client(
         username=ALTERTABLE_USERNAME,
         password=ALTERTABLE_PASSWORD,
-        host=ALTERTABLE_HOST,
-        port=ALTERTABLE_PORT,
-        tls=ALTERTABLE_TLS,
+        **CONNECTION_SETTINGS,
     ) as client:
         # Begin transaction
         with client.begin_transaction():
@@ -137,9 +137,7 @@ def example_metadata():
     with Client(
         username=ALTERTABLE_USERNAME,
         password=ALTERTABLE_PASSWORD,
-        host=ALTERTABLE_HOST,
-        port=ALTERTABLE_PORT,
-        tls=ALTERTABLE_TLS,
+        **CONNECTION_SETTINGS,
     ) as client:
         # Get catalogs
         print("Catalogs:")