cqle: partition key index on model columns

aholmberg · aholmberg · commit f03e6381e7bb · 2016-03-31T15:20:27.000-05:00
PYTHON-535
diff --git a/cassandra/cqlengine/columns.py b/cassandra/cqlengine/columns.py
@@ -159,6 +159,7 @@ def __init__(self,
 
         # the column name in the model definition
         self.column_name = None
+        self._partition_key_index = None
         self.static = static
 
         self.value = None
diff --git a/cassandra/cqlengine/models.py b/cassandra/cqlengine/models.py
@@ -843,6 +843,7 @@ def _get_polymorphic_base(bases):
 
         has_partition_keys = any(v.partition_key for (k, v) in column_definitions)
 
+        partition_key_index = 0
         # transform column definitions
         for k, v in column_definitions:
             # don't allow a column with the same name as a built-in attribute or method
@@ -858,6 +859,9 @@ def _get_polymorphic_base(bases):
             if not has_partition_keys and v.primary_key:
                 v.partition_key = True
                 has_partition_keys = True
+            if v.partition_key:
+                v._partition_key_index = partition_key_index
+                partition_key_index += 1
             _transform_column(k, v)
 
         partition_keys = OrderedDict(k for k in primary_keys.items() if k[1].partition_key)
diff --git a/cassandra/cqlengine/query.py b/cassandra/cqlengine/query.py
@@ -37,6 +37,7 @@ class QueryException(CQLEngineException):
 class IfNotExistsWithCounterColumn(CQLEngineException):
     pass
 
+
 class IfExistsWithCounterColumn(CQLEngineException):
     pass
 
