cQtTnAep

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