Add CAS WriteType, avoiding KeyError on CAS write timeout

thobbs · thobbs · commit 9bd8146f4021 · 2014-07-11T14:31:22.000-05:00
Fixes PYTHON-91
diff --git a/cassandra/policies.py b/cassandra/policies.py
@@ -572,12 +572,18 @@ class WriteType(object):
     internally before a BATCH write.
     """
 
+    CAS = 5
+    """
+    A lighweight-transaction write, such as "DELETE ... IF EXISTS".
+    """
+
 WriteType.name_to_value = {
     'SIMPLE': WriteType.SIMPLE,
     'BATCH': WriteType.BATCH,
     'UNLOGGED_BATCH': WriteType.UNLOGGED_BATCH,
     'COUNTER': WriteType.COUNTER,
     'BATCH_LOG': WriteType.BATCH_LOG,
+    'CAS': WriteType.CAS
 }
 
 
