diff --git a/setup.py b/setup.py index 260f1657c87b4..2454074d2b92e 100644 --- a/setup.py +++ b/setup.py @@ -121,10 +121,11 @@ def get_git_sha(): "pybigquery>=0.4.10", "google-cloud-bigquery>=2.4.0", ], - "clickhouse": ["clickhouse-sqlalchemy>= 0.1.4, <0.2"], + "clickhouse": ["clickhouse-sqlalchemy>=0.1.4, <0.2"], "cockroachdb": ["cockroachdb>=0.3.5, <0.4"], "cors": ["flask-cors>=2.0.0"], "crate": ["crate[sqlalchemy]>=0.26.0, <0.27"], + "databricks": ["databricks-dbapi[sqlalchemy]>=0.5.0, <0.6"], "db2": ["ibm-db-sa>=0.3.5, <0.4"], "dremio": ["sqlalchemy-dremio>=1.1.5, <1.2"], "drill": ["sqlalchemy-drill==0.1.dev"], diff --git a/superset/db_engine_specs/databricks.py b/superset/db_engine_specs/databricks.py new file mode 100644 index 0000000000000..d67370be7e331 --- /dev/null +++ b/superset/db_engine_specs/databricks.py @@ -0,0 +1,23 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you under the Apache License, Version 2.0 (the +# "License"); you may not use this file except in compliance +# with the License. You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, +# software distributed under the License is distributed on an +# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +# KIND, either express or implied. See the License for the +# specific language governing permissions and limitations +# under the License.o +from superset.db_engine_specs.hive import HiveEngineSpec + + +class DatabricksHiveEngineSpec(HiveEngineSpec): + engine = "databricks" + engine_name = "Databricks Hive" + driver = "pyhive"