fix: Dremio alias (#28222)

This commit is contained in:
Beto Dealmeida 2024-04-26 07:49:20 -04:00 committed by GitHub
parent c5e7d870f0
commit 173d5d09bf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 1 deletions

View File

@ -154,7 +154,7 @@ def get_available_engine_specs() -> dict[type[BaseEngineSpec], set[str]]:
try:
dialect = ep.load()
except Exception as ex: # pylint: disable=broad-except
logger.warning("Unable to load SQLAlchemy dialect %s: %s", dialect, ex)
logger.warning("Unable to load SQLAlchemy dialect %s: %s", ep.name, ex)
else:
backend = dialect.name
if isinstance(backend, bytes):

View File

@ -37,6 +37,13 @@ FIXED_ALIAS_IN_SELECT_VERSION = Version("24.1.0")
class DremioEngineSpec(BaseEngineSpec):
engine = "dremio"
engine_name = "Dremio"
engine_aliases = {"dremio+flight"}
sqlalchemy_uri_placeholder = (
"dremio+flight://data.dremio.cloud:443/?"
"Token=<TOKEN>&"
"UseEncryption=true&"
"disableCertificateVerification=true"
)
_time_grain_expressions = {
None: "{col}",