Connect with the Python driver

Because Astra DB is based on Apache Cassandra®, you can use Cassandra drivers to connect to your Astra Managed Clusters databases.

To use the Python driver, you need to install the driver and its dependencies, and then connect the driver to your database. Once connected, you can write scripts that use the driver to run commands against your database.

This quickstart explains how to use the Python driver to connect to an Astra Managed Clusters database and send some Cassandra Query Language (CQL) statements to the database. It also explains how to upgrade from an earlier version of the Python driver to a version that supports Astra DB.

Prerequisites

Install Python 3.10 or later.

CPython (the standard Python implementation) and PyPy are supported and tested.

For legacy applications that require an earlier version of the Python driver, see the documentation for that driver version for supported Python versions.
Install pip version 23.0 or later.
Create a database.
Download your database’s Secure Connect Bundle (SCB).

For multi-region databases, download the SCB for a region that is geographically close to your application to reduce latency.

If you need to connect to multiple regions in the same application, you need the SCB for each region, and your driver code must instantiate one root object (session) for each region. For more information, see Best practices for Cassandra drivers.
Set the following environment variables:
- DATABASE_ID: The database ID.
- APPLICATION_TOKEN: An application token with the Database Administrator role.

Driver authentication methods

There are two driver authentication methods:

token authentication

The token authentication method is supported and recommended for most recent driver versions.

In your driver authentication code, pass the literal string token as the username and your application token value (AstraCS:…) as the password. For example:

("token", "AstraCS:...")

clientId and secret authentication

If you are on an older driver version that doesn’t support token authentication, then you might need to use clientId and secret.

When you generate an application token, download or copy the token.json that contains the following values:

{
  "clientId": "CLIENT_ID",
  "secret": "CLIENT_SECRET",
  "token": "APPLICATION_TOKEN"
}

Then, in your driver authentication code, pass clientId as the username and secret as the password. For example:

("CLIENT_ID", "SECRET")

For more information, see Token details.

Install the Python driver

Install the Python driver:

pip install cassandra-driver

If you install an earlier version of the driver, make sure your version is compatible with Astra DB. If you need to query vector data in Astra DB Serverless (vector) databases, make sure your version also supports vector data. For more information, see Cassandra drivers supported by DataStax.

Optional: Verify the installation:
```
pip show cassandra-driver
```
Make sure the returned Version is the latest version or the specific version that you installed.

Connect the Python driver

In the root of your Python project, create a connect_database.py file:
```
cd python_project
touch connect_database.py
```
Copy one of the following connection code examples into connect_database.py, and then replace PATH/TO/SCB.zip with the absolute path to your database’s Secure Connect Bundle (SCB) zip file (secure-connect-DATABASE_NAME.zip):
Production configuration (recommended)
When using the Python driver in production environments or with simulated production workloads, DataStax recommends robust session configuration with profile and cluster details to help optimize driver performance.

The following code initializes a session to connect to your database with the cassandra-driver. It uses an SCB and authentication credentials stored in environment variables. Additionally, it includes options for connection timeout, request timeout, and protocol version.
connect_database.py

import os from cassandra.cluster import Cluster, ExecutionProfile, EXEC_PROFILE_DEFAULT, ProtocolVersion from cassandra.auth import PlainTextAuthProvider import json cloud_config= { 'secure_connect_bundle': "PATH/TO/SCB.zip", 'connect_timeout': 30 } auth_provider=PlainTextAuthProvider("token", os.environ["APPLICATION_TOKEN"]) profile = ExecutionProfile(request_timeout=30) cluster = Cluster( cloud=cloud_config, auth_provider=auth_provider, execution_profiles={EXEC_PROFILE_DEFAULT: profile}, protocol_version=ProtocolVersion.V4 ) session = cluster.connect()
Minimal configuration
You can use a minimal session configuration for testing or lower environments where you don’t need to optimize the cluster details for production workloads.

The following code initializes a session to connect to your database with the cassandra-driver. It uses an SCB and authentication credentials stored in environment variables.
connect_database.py

import os from cassandra.cluster import Cluster from cassandra.auth import PlainTextAuthProvider import json session = Cluster( cloud={"secure_connect_bundle": "PATH/TO/SCB.zip"}, auth_provider=PlainTextAuthProvider("token", os.environ["APPLICATION_TOKEN"]), ).connect()
The connection code creates a Cluster instance to connect to your Astra DB database. You typically have one Cluster for each Astra DB database, and only one Session for the entire application. For more information, see Best practices for Cassandra drivers.
To test the connection, add a simple query to the script.

The following example queries the system.local table. You can replace the example SELECT statement with any CQL statement that you want to run against a keyspace and table in your database.
connect_database.py
```
row = session.execute("select release_version from system.local").one()
if row:
    print(row[0])
else:
    print("An error occurred.")
```
Save and run your Python script:
```
python ./connect_database.py
```
If you ran the example SELECT statement on the system.local table, then the cluster_name value from the system.local table is printed to the console if the script runs successfully.

Upgrade the Python driver

Use these steps if you need to upgrade from an earlier version of the Python driver to a version that supports Astra DB:

Complete the prerequisites.
Install the latest Python driver.

In your existing Python driver code, modify the connection code to use the SCB and token authentication.

import os
from cassandra.cluster import Cluster
from cassandra.auth import PlainTextAuthProvider
import json

cloud_config= {
        'secure_connect_bundle': 'PATH/TO/SCB.zip'
        }
auth_provider = PlainTextAuthProvider("token", os.environ["APPLICATION_TOKEN"])
cluster = Cluster(cloud=cloud_config, auth_provider=auth_provider)
session = cluster.connect()

For more information, see Connect the Python driver.

Run your Python script.

Next steps

You can extend or modify the example script used in this guide to run other commands against your database, or connect to other databases. For more information, see the following: