KbgGhPbA

1from cassandra.cluster import Cluster
2from cassandra.query import dict_factory
3
4
5def create_keyspace(session, keyspace):
6    session.execute("""
7        CREATE KEYSPACE IF NOT EXISTS """+keyspace+"""
8        WITH replication = { 'class': 'SimpleStrategy', 'replication_factor': '1' }
9        """)
10
11
12def create_table(session, keyspace, table):
13    session.execute("""
14        CREATE TABLE IF NOT EXISTS """+ keyspace+"""."""+table+""" (
15        user_id int ,
16        avg_movie_rating float,
17        PRIMARY KEY(user_id)
18        
19        """)
20
21
22def push_data_table(session, keyspace, table, userId, avgMovieRating):
23    session.execute(
24        """
25        INSERT INTO """+keyspace+"""."""+table+""" (user_id, avg_movie_rating)
26        VALUES (%(user_id)s, %(avg_movie_rating)s)
27        """,
28        'user_id': userId,
29        'avg_movie_rating': avgMovieRating
30
31
32def get_data_table(session, keyspace, table):
33    rows = session.execute("SELECT * FROM "+keyspace+"."+table+";")
34    for row in rows:
35        print(row)
36
37
38def clear_table(session, keyspace, table):
39    session.execute("TRUNCATE "+keyspace+"."+table+";")
40
41
42def delete_table(session, keyspace, table):
43    session.execute("DROP TABLE "+keyspace+"."+table+";")