Merge pull request #137 from fredrikaverpil/feat/spanner

Author: Maxteabag

README.md

@@ -25,7 +25,7 @@
 ---
 
 ### Connect
-Supports all major databases: SQL Server, PostgreSQL, MySQL, SQLite, MariaDB, FirebirdSQL, Oracle, DuckDB, CockroachDB, ClickHouse, Snowflake, Supabase, CloudFlare D1, Turso, Athena, BigQuery, RedShift, IBM Db2, SAP HANA, Teradata, Trino, Presto and Apache Flight SQL.
+Supports all major databases: SQL Server, PostgreSQL, MySQL, SQLite, MariaDB, FirebirdSQL, Oracle, DuckDB, CockroachDB, ClickHouse, Snowflake, Supabase, CloudFlare D1, Turso, Athena, BigQuery, Spanner, RedShift, IBM Db2, SAP HANA, Teradata, Trino, Presto and Apache Flight SQL.
 
 ![Database Providers](docs/demos/demo-providers.gif)
 
@@ -281,6 +281,8 @@ Most of the time you can just run `sqlit` and connect. If a Python driver is mis
 | Snowflake                           | `snowflake-connector-python` | `pipx inject sqlit-tui snowflake-connector-python` | `python -m pip install snowflake-connector-python` |
 | Firebird                            | `firebirdsql`                | `pipx inject sqlit-tui firebirdsql`                | `python -m pip install firebirdsql`                |
 | Athena                              | `pyathena`                   | `pipx inject sqlit-tui pyathena`                   | `python -m pip install pyathena`                   |
+| BigQuery                            | `google-cloud-bigquery`      | `pipx inject sqlit-tui google-cloud-bigquery`      | `python -m pip install google-cloud-bigquery`      |
+| Spanner                             | `google-cloud-spanner`       | `pipx inject sqlit-tui google-cloud-spanner`       | `python -m pip install google-cloud-spanner`       |
 | Apache Arrow Flight SQL             | `adbc-driver-flightsql`      | `pipx inject sqlit-tui adbc-driver-flightsql`      | `python -m pip install adbc-driver-flightsql`      |
 
 ### SSH Tunnel Support

pyproject.toml

@@ -48,6 +48,7 @@ all = [
     "trino>=0.329.0",
     "presto-python-client>=0.8.4",
     "google-cloud-bigquery",
+    "google-cloud-spanner>=3.0.0",
     "duckdb>=1.1.0",  # min avoids known CVEs
     "clickhouse-connect>=0.7.0",
     "requests>=2.32.4",  # min avoids known CVEs
@@ -72,6 +73,7 @@ teradata = ["teradatasql>=20.0.0"]
 trino = ["trino>=0.329.0"]
 presto = ["presto-python-client>=0.8.4"]
 bigquery = ["google-cloud-bigquery"]
+spanner = ["google-cloud-spanner>=3.0.0"]
 redshift = ["redshift-connector"]
 duckdb = ["duckdb>=1.1.0"]  # min avoids known CVEs
 clickhouse = ["clickhouse-connect>=0.7.0"]
@@ -187,6 +189,7 @@ markers = [
     "firebird: Firebird database tests",
     "clickhouse: ClickHouse database tests",
     "flight: Apache Arrow Flight SQL database tests",
+    "spanner: Google Cloud Spanner database tests",
     "asyncio: async tests",
     "integration: integration tests (may require external services)",
 ]
@@ -241,7 +244,18 @@ module = [
     "adbc_driver_flightsql",
     "adbc_driver_flightsql.dbapi",
     "adbc_driver_manager",
-    "textual_fastdatatable"
+    "textual_fastdatatable",
+    "google.cloud",
+    "google.cloud.bigquery",
+    "google.cloud.bigquery.dbapi",
+    "google.cloud.spanner",
+    "google.cloud.spanner_dbapi",
+    "google.api_core",
+    "google.api_core.client_options",
+    "google.auth",
+    "google.auth.credentials",
+    "google.oauth2",
+    "google.oauth2.service_account",
 ]
 ignore_missing_imports = true
 

sqlit/domains/connections/domain/config.py

@@ -29,6 +29,7 @@ class DatabaseType(str, Enum):
     PRESTO = "presto"
     REDSHIFT = "redshift"
     SNOWFLAKE = "snowflake"
+    SPANNER = "spanner"
     SQLITE = "sqlite"
     SUPABASE = "supabase"
     TERADATA = "teradata"
@@ -50,6 +51,7 @@ class DatabaseType(str, Enum):
     DatabaseType.TERADATA,
     DatabaseType.SNOWFLAKE,
     DatabaseType.BIGQUERY,
+    DatabaseType.SPANNER,
     DatabaseType.TRINO,
     DatabaseType.PRESTO,
     DatabaseType.DUCKDB,

sqlit/domains/connections/providers/spanner/__init__.py

@@ -0,0 +1 @@
+"""Google Cloud Spanner provider package."""