Python Pandas-Cassandra connector

Usage

Convert pandas DF to cassandra

import pandra as cql
from cassandra.cluster import Cluster

cluster = Cluster([
                   'ip_cluster'
                   ])

cassandra_session = cluster.connect('name_space')

test = [
    {'a': 1, 'b': 2, 'c': 'dfs'},
    {'a': 4, 'b': 7, 'c': 'test'},
]

cql_df = cql.CassandraDataFrame.from_dict(test)

column_types = {
    'a': cql.IntegerType('a', primary_key=True),
    'b': cql.IntegerType('b'),
    'c': cql.TextType('c')
}

cql_df.to_cassandra(cassandra_session=cassandra_session,
                    table_name='test_table',
                    data_types=column_types,
                    debug=False,
                    create_table=True)

Read table from Cassandra

import pandra as cql
cdf = cql.CassandraDataFrame.from_cassandra(cassandrasession, 'SELECT * FROM test_table')

Define your proper DataType and CassandraTable

import pandra as cql

class MyDataType(cql.DataType):
    def __init__(self, name, primary_key=False):
        super(MyDataType, self).__init__(name, 'cassandra_data_type', primary_key)

# Define your own class
class User(cql.CassandraTable):
    id = cql.IntegerType('id', primary_key=True)
    name = cql.TextType('username', primary_key=True)
    email = cql.TextType('email')
    password = cql.TextType('password')
    mydatatype = MyDataType('mydatatype')

# Create a new User object
u = User(id=123456, name='Michael', email='test@orm.org', password='my-pwd', mydatatype = 1)

# Create table User
User.create()
# 'CREATE TABLE User ( id int, username text, email text, password text, mydatatype cassandra_data_type, PRIMARY KEY ( id, username ) );'

# Insert data to table User
u.insert()
# ('INSERT INTO User (id, username, email, password, mydatatype) VALUES (%s, %s, %s, %s, %s)', [123456, 'Michael', 'test@orm.org', 'my-pwd', 1])