From a6a774b0fbdaadb48f9985f1ff586ed085519ba0 Mon Sep 17 00:00:00 2001
From: Tyler Hobbs <tyler@datastax.com>
Date: Wed, 21 May 2014 16:31:29 -0500
Subject: [PATCH] Handle unset CL on statements when printing/logging

---
 cassandra/query.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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'<PreparedStatement query="%s", consistency=%s>' %
                 (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'<BoundStatement query="%s", values=%s, consistency=%s>' %
                 (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'<BatchStatement type=%s, statements=%d, consistency=%s>' %
                 (self.batch_type, len(self._statements_and_parameters), consistency))
     __repr__ = __str__