From cc91d0a52df3fa9870d3bb0902f0ca728c7655fa Mon Sep 17 00:00:00 2001 From: Aarav Borthakur Date: Fri, 7 Jul 2023 19:28:47 -0700 Subject: [PATCH 1/4] chore: use new Rockset SQLAlchemy library --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 5bd67ca0c2a3e..21d94e2920066 100644 --- a/setup.py +++ b/setup.py @@ -178,7 +178,7 @@ def get_git_sha() -> str: "trino": ["trino>=0.324.0"], "prophet": ["prophet>=1.0.1, <1.1", "pystan<3.0"], "redshift": ["sqlalchemy-redshift>=0.8.1, < 0.9"], - "rockset": ["rockset>=0.8.10, <0.9"], + "rockset": ["rockset-sqlalchemy>=0.0.1"], "shillelagh": [ "shillelagh[datasetteapi,gsheetsapi,socrata,weatherapi]>=1.1.1, <2" ], From ec535d18abb1d49e0de423badaaf146581e452a6 Mon Sep 17 00:00:00 2001 From: Aarav Borthakur Date: Fri, 7 Jul 2023 19:29:51 -0700 Subject: [PATCH 2/4] docs: update Rockset usage (include virtual instance configuration) --- docs/docs/databases/rockset.mdx | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/docs/docs/databases/rockset.mdx b/docs/docs/databases/rockset.mdx index cfce7c54dbde6..47b7c566d00fa 100644 --- a/docs/docs/databases/rockset.mdx +++ b/docs/docs/databases/rockset.mdx @@ -10,7 +10,16 @@ version: 1 The connection string for Rockset is: ``` -rockset://apikey:{your-apikey}@api.rs2.usw2.rockset.com/ +rockset://{api key}:@{api server} +``` + +Get your API key from the [Rockset console](https://console.rockset.com/apikeys). +Find your API server from the [API reference](https://rockset.com/docs/rest-api/#introduction). Omit the `https://` portion of the URL. + +To target to a specific virtual instance, use this URI format: + +``` +rockset://:{api key}:@{api server}/{VI ID} ``` For more complete instructions, we recommend the [Rockset documentation](https://docs.rockset.com/apache-superset/). From f9fd42c576509a41471d7b7464d8839f25d157ec Mon Sep 17 00:00:00 2001 From: Aarav Borthakur Date: Tue, 11 Jul 2023 10:08:46 -0700 Subject: [PATCH 3/4] Add upper bound of Rockset version --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 21d94e2920066..b62b9a82b7a0c 100644 --- a/setup.py +++ b/setup.py @@ -178,7 +178,7 @@ def get_git_sha() -> str: "trino": ["trino>=0.324.0"], "prophet": ["prophet>=1.0.1, <1.1", "pystan<3.0"], "redshift": ["sqlalchemy-redshift>=0.8.1, < 0.9"], - "rockset": ["rockset-sqlalchemy>=0.0.1"], + "rockset": ["rockset-sqlalchemy>=0.0.1, <1.0.0"], "shillelagh": [ "shillelagh[datasetteapi,gsheetsapi,socrata,weatherapi]>=1.1.1, <2" ], From b54716c3d7c75308fa45c055465b8a933c1e0217 Mon Sep 17 00:00:00 2001 From: Aarav Borthakur Date: Tue, 11 Jul 2023 13:08:35 -0700 Subject: [PATCH 4/4] docs: add Rockset to "install database drivers" page --- docs/docs/databases/installing-database-drivers.mdx | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docs/docs/databases/installing-database-drivers.mdx b/docs/docs/databases/installing-database-drivers.mdx index e4e972f0648b2..5599b35900aae 100644 --- a/docs/docs/databases/installing-database-drivers.mdx +++ b/docs/docs/databases/installing-database-drivers.mdx @@ -51,8 +51,8 @@ Some of the recommended packages are shown below. Please refer to [setup.py](htt | [MySQL](/docs/databases/mysql) | `pip install mysqlclient` | `mysql://:@/` | | [Oracle](/docs/databases/oracle) | `pip install cx_Oracle` | `oracle://` | | [PostgreSQL](/docs/databases/postgres) | `pip install psycopg2` | `postgresql://:@/` | -| [Trino](/docs/databases/trino) | `pip install trino` | `trino://{username}:{password}@{hostname}:{port}/{catalog}` | | [Presto](/docs/databases/presto) | `pip install pyhive` | `presto://` | +| [Rockset](/docs/databases/rockset) | `pip install rockset-sqlalchemy` | `rockset://:@` | [SAP Hana](/docs/databases/hana) | `pip install hdbcli sqlalchemy-hana or pip install apache-superset[hana]` | `hana://{username}:{password}@{host}:{port}` | | [StarRocks](/docs/databases/starrocks) | `pip install starrocks` | `starrocks://:@:/.` | | [Snowflake](/docs/databases/snowflake) | `pip install snowflake-sqlalchemy` | `snowflake://{user}:{password}@{account}.{region}/{database}?role={role}&warehouse={warehouse}` | @@ -60,6 +60,7 @@ Some of the recommended packages are shown below. Please refer to [setup.py](htt | [SQL Server](/docs/databases/sql-server) | `pip install pymssql` | `mssql+pymssql://` | | [Teradata](/docs/databases/teradata) | `pip install teradatasqlalchemy` | `teradatasql://{user}:{password}@{host}` | | [TimescaleDB](/docs/databases/timescaledb) | `pip install psycopg2` | `postgresql://:@:/` | +| [Trino](/docs/databases/trino) | `pip install trino` | `trino://{username}:{password}@{hostname}:{port}/{catalog}` | | [Vertica](/docs/databases/vertica) | `pip install sqlalchemy-vertica-python` | `vertica+vertica_python://:@/` | | [YugabyteDB](/docs/databases/yugabytedb) | `pip install psycopg2` | `postgresql://:@/` | ---