Skip to content

Commit 91edcb5

Browse files
committed
Fix db override
1 parent 0bbf094 commit 91edcb5

File tree

1 file changed

+6
-0
lines changed
  • sqlit/domains/connections/providers/bigquery

1 file changed

+6
-0
lines changed

sqlit/domains/connections/providers/bigquery/adapter.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,12 @@ def supports_cross_database_queries(self) -> bool:
6161
# BigQuery supports cross-dataset queries
6262
return True
6363

64+
def apply_database_override(self, config: ConnectionConfig, database: str) -> ConnectionConfig:
65+
"""Apply a default dataset for unqualified queries."""
66+
if not database:
67+
return config
68+
return config.with_endpoint(database=database)
69+
6470
@property
6571
def system_databases(self) -> frozenset[str]:
6672
return frozenset({"INFORMATION_SCHEMA"})

0 commit comments

Comments
 (0)