Skip to main content

Apache Drill for SQLAlchemy

Project description

Apache Drill dialect for SQLAlchemy.


The primary purpose of this is to have a working dialect for Apache Drill that can be used with Apache Superset.

https://superset.incubator.apache.org

Obviously, a working, robust dialect for Drill serves other purposes as well, but most of the iterative planning for this REPO will be based on working with Superset. Other changes will gladly be incorporated, as long as it doesn't hurt Superset integration.

Installation

Installing the dialect is straightforward. Simply:

pip install sqlalchemy-drill

Alternatively, you can download the latest release from github and install from here:

python3 -m pip install git+https://github.com/JohnOmernik/sqlalchemy-drill.git

Usage with REST

Drill's REST API can execute queries with results streamed to JSON returned over chunked HTTP for Drill >= 1.19, otherwise with results buffered and then returned in a conventional HTTP response. A SQLAlchemy URL to connect to Drill over REST looks like the following.

drill+sadrill://<username>:<password>@<host>:<port>/<storage_plugin>?use_ssl=True

To connect to Drill running on a local machine running in embedded mode you can use the following connection string.

drill+sadrill://localhost:8047/dfs?use_ssl=False

Supported URL query parameters

Parameter Type Description
use_ssl boolean Whether to connect to Drill using HTTPS
verify_ssl boolean Whether to verify the server's TLS certificate
impersonation_target[1] string Username of a Drill user to be impersonated by this connection

[1] Requires a build of Drill that incorporates the fix for DRILL-8168.

Trailing metadata

Query result metadata returned by the Drill REST API is stored in the result_md field of the DB-API Cursor object. Note that any trailing metadata, i.e. metadata which comes after result row data, will only be populated after you have iterated through all of the returned rows. If you need this trailing metadata you can make the cursor object reachable after it has been completely iterated by obtaining a reference to it beforehand, as follows.

r = engine.execute('select current_timestamp')
r.cursor.result_md  # access metadata, but only leading metadata
cur = r.cursor      # obtain a reference for use later
r.fetchall()        # iterate through all result data
cur.result_md       # access metadata, including trailing metadata
del cur             # optionally delete the reference when done

Drill < 1.19

In versions of Drill earlier than 1.19, all data values are serialised to JSON strings and column type metadata comes after the data itself. As a result, for these versions of Drill, the drill+sadrill dialect returns every data value as a string. To convert non-string data to its native type you need to typecast it yourself.

Drill >= 1.19

In Drill 1.19 the REST API began making use of numeric types in JSON for numbers and times, the latter via a UNIX time representation while column type metadata was moved ahead of the result data. Because of this, the drill+sadrill dialect is able to return appropriate types for numbers and times when used with Drill >= 1.19.

Usage with JDBC

Connecting to Drill via JDBC is a little more complicated than a local installation and complete instructions can be found on the Drill documentation here: https://drill.apache.org/docs/using-the-jdbc-driver/.

In order to configure SQLAlchemy to work with Drill via JDBC you must:

  • Download the latest JDBC Driver available here: http://apache.osuosl.org/drill/
  • Copy this driver to your classpath or other known path
  • Set an environment variable called DRILL_JDBC_DRIVER_PATH to the full path of your driver location
  • Set an environment variable called DRILL_JDBC_JAR_NAME to the name of the .jar file for the Drill driver.

Additionally, you will need to install JayDeBeApi as well as jPype version 0.6.3. These modules are listed as optional dependencies and will not be installed by the default installer.

If the JDBC driver is not available, the dialect will throw errors when trying to connect. In addition, sqlalchemy-drill will not launch a JVM for you so you need to do this yourself with a call to JPype like the following. See the file test-jdbc.py in this repo for a working example.

jpype.startJVM("-ea", classpath="lib/*")
drill+jdbc://<username>:<passsword>@<host>:<port>

For a simple installation, this might look like:

drill+jdbc://admin:password@localhost:31010

Usage with ODBC

In order to configure SQLAlchemy to work with Drill via ODBC you must:

  • Install latest Drill ODBC Driver: https://drill.apache.org/docs/installing-the-driver-on-linux/
  • Ensure that you have ODBC support in your system (unixODBC package for RedHat-based systems).
  • Install pyodbc Python package. This module is listed as an optional dependency and will not be installed by the default installer.

To connect to Drill with SQLAlchemy use the following connection string:

drill+odbc:///?<ODBC connection parameters>

Connection properties are available in the official documentation: https://drill.apache.org/docs/odbc-configuration-reference/

For a simple installation, this might look like:

drill+odbc:///?Driver=/opt/mapr/drill/lib/64/libdrillodbc_sb64.so&ConnectionType=Direct&HOST=localhost&PORT=31010&AuthenticationType=Plain&UID=admin&PWD=password

or for the case when you have DSN configured in odbc.ini:

drill+odbc:///?DSN=drill_dsn_name

Note: it's better to avoid using connection string with hostname:port or username/password, like 'drill+odbc://admin:password@localhost:31010/' but use only ODBC properties instead to avoid any misinterpretation between these parameters.

Usage with Superset

For a complete tutorial on how to use Superset with Drill, read the tutorial on @cgivre's blog available here: http://thedataist.com/visualize-anything-with-superset-and-drill/.

Current Status/Development Approach

Currently we can connect to drill, and issue queries for most visualizations and get results. We also enumerate table columns for some times of tables. Here are things that are working as some larger issues to work out. (Individual issues are tracked under issues)

  • Connection to Drill via the databases tab in Superset succeeds
  • You can do basic queries for most types of viz/tables
  • There may be issues with advanced queries/joins. As you learn about new ones, please track in issues

Many thanks

to drillpy and pydrill for code used in creating the original drilldbapi.py code for connecting!

Docker

It is recommended to extend the official Docker image to include this Apache Drill driver:

FROM apache/superset
# Switching to root to install the required packages
USER root
RUN pip install git+https://github.com/JohnOmernik/sqlalchemy-drill.git
# Switching back to using the `superset` user
USER superset

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

sqlalchemy_drill-1.1.5.tar.gz (25.4 kB view details)

Uploaded Source

Built Distribution

sqlalchemy_drill-1.1.5-py3-none-any.whl (30.8 kB view details)

Uploaded Python 3

File details

Details for the file sqlalchemy_drill-1.1.5.tar.gz.

File metadata

  • Download URL: sqlalchemy_drill-1.1.5.tar.gz
  • Upload date:
  • Size: 25.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.11.7

File hashes

Hashes for sqlalchemy_drill-1.1.5.tar.gz
Algorithm Hash digest
SHA256 8a167af388ef0530fe79adae1c547fb4b651615d300c106ff659c12e2688aa57
MD5 a741e29671ab13ea85c2d42759539c18
BLAKE2b-256 00a228ed460d308f3fa6e893f5e86b5e09db7617651d7dbfab1f6af1a2cbe13e

See more details on using hashes here.

File details

Details for the file sqlalchemy_drill-1.1.5-py3-none-any.whl.

File metadata

File hashes

Hashes for sqlalchemy_drill-1.1.5-py3-none-any.whl
Algorithm Hash digest
SHA256 ff6e1d38bde109d5a885d6425bd58e6f09927d9a51804199d0cc4b30128cd094
MD5 f21c3f0ab957d4189d12951bd2393bee
BLAKE2b-256 8fcb4f68997046938d28994d898024fb6e38a3c56a6905dd30108c0701fbf834

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page