import ssl
import os
from cassandra.cluster import Cluster
from cassandra.auth import PlainTextAuthProvider
from cassandra.query import SimpleStatement
from cassandra.encoder import Encoderdef
e_var(name):
try:
    return os.environ[name]
except KeyError:
    print ("Please set the environment variable " + name)
    sys.exit(1)

ssl_options = {
    'ssl_version': ssl.PROTOCOL_TLSv1}

user = e_var('CAS_USER')
pwd = e_var('CAS_PWD')
ip = e_var('CAS_IPS').split(',')
keyspace = e_var('CAS_KEYSPACE')
ap = PlainTextAuthProvider(username=user, password=pwd)

cluster = Cluster(ip, auth_provider=ap, ssl_options=ssl_options)
session = cluster.connect(keyspace)
session.execute('CREATE TABLE IF NOT EXISTS  sample (key text,data blob, PRIMARY KEY (key))')
print('***table done')
def insert(bytes_length, message ):
    e = Encoder()
    data = ('a' * bytes_length).encode()
    d = {    'key' : 'bucket_1', 'data' :bytearray(data)  }
    session.execute("""INSERT INTO sample (key,data) VALUES (%(key)s, %(data)s)""",d)
    print('***' + message)

insert(3000, 'done first insert')
insert(32727, 'done second insert')
insert(32728, 'done third insert')