Untitled
def get_record_amount( self, table_name: str = "imdbdata", ) -> int: session = self.connect() query = f"SELECT COUNT(*) FROM mydatabase.{table_name}" result = session.execute(query) count = result[0].count return count def list_tables(self) -> tp.List[str]: session = self.connect() query = "SELECT table_name FROM system_schema.tables WHERE keyspace_name = 'mydatabase'" result = session.execute(query) tables = [row.table_name for row in result] return tables