Coverage for mindsdb / integrations / handlers / cassandra_handler / cassandra_handler.py: 0%

1from mindsdb.integrations.handlers.scylla_handler import Handler as ScyllaHandler 

2from mindsdb.integrations.libs.response import ( 

3 HandlerResponse as Response) 

4 

5 

6class CassandraHandler(ScyllaHandler): 

7 """ 

8 This handler handles connection and execution of the Cassandra statements. 

9 """ 

10 

11 name = 'cassandra' 

12 

13 def __init__(self, name, **kwargs): 

14 super().__init__(name, **kwargs) 

15 

16 def get_tables(self) -> Response: 

17 """ 

18 Get the list of tables in the connected Cassandra database. 

19 

20 :return: List of table names. 

21 """ 

22 sql = "DESCRIBE TABLES" 

23 result = self.native_query(sql) 

24 df = result.data_frame 

25 df = df.rename(columns={'name': 'table_name'}) 

26 result.data_frame = df 

27 return result