diff --git a/cassandra/query.py b/cassandra/query.py index c2a32802..3f178539 100644 --- a/cassandra/query.py +++ b/cassandra/query.py @@ -346,7 +346,7 @@ class PreparedStatement(object): return BoundStatement(self).bind(values) def __str__(self): - consistency = ConsistencyLevel.value_to_name[self.consistency_level] + consistency = ConsistencyLevel.value_to_name.get(self.consistency_level, 'Not Set') return (u'' % (self.query_string, consistency)) __repr__ = __str__ @@ -484,7 +484,7 @@ class BoundStatement(Statement): return None def __str__(self): - consistency = ConsistencyLevel.value_to_name[self.consistency_level] + consistency = ConsistencyLevel.value_to_name.get(self.consistency_level, 'Not Set') return (u'' % (self.prepared_statement.query_string, self.raw_values, consistency)) __repr__ = __str__ @@ -622,7 +622,7 @@ class BatchStatement(Statement): self.add(statement, parameters) def __str__(self): - consistency = ConsistencyLevel.value_to_name[self.consistency_level] + consistency = ConsistencyLevel.value_to_name.get(self.consistency_level, 'Not Set') return (u'' % (self.batch_type, len(self._statements_and_parameters), consistency)) __repr__ = __str__