making sure we're 1.2 compatible

This commit is contained in:
Jon Haddad
2014-06-18 13:39:40 -07:00
parent f709878577
commit 4cab5851f0
4 changed files with 11 additions and 7 deletions

View File

@@ -268,8 +268,12 @@ def connection_manager():
def execute_native(query, params=None, consistency_level=None):
# TODO use consistency level
prepared = session.prepare(query)
result = session.execute(prepared, params)
if isinstance(query, BaseCQLStatement):
params = query.get_context()
query = str(query)
params = params or {}
result = session.execute(query, params)
return result
def get_session():

View File

@@ -223,7 +223,7 @@ def get_fields(model):
ks_name = model._get_keyspace()
col_family = model.column_family_name(include_keyspace=False)
query = "select * from system.schema_columns where keyspace_name = ? and columnfamily_name = ?"
query = "select * from system.schema_columns where keyspace_name = %s and columnfamily_name = %s"
tmp = execute_native(query, [ks_name, col_family])
# Tables containing only primary keys do not appear to create

View File

@@ -4,7 +4,7 @@ from datetime import datetime, timedelta
from cqlengine import BaseContainerColumn, Map, columns
from cqlengine.columns import Counter, List, Set
from cqlengine.connection import execute, RowResult
from cqlengine.connection import execute, RowResult, execute_native
from cqlengine.exceptions import CQLEngineException, ValidationError
from cqlengine.functions import Token, BaseQueryFunction, QueryValue
@@ -775,7 +775,7 @@ class DMLQuery(object):
if self._batch:
return self._batch.add_query(q)
else:
return execute(q, consistency_level=self._consistency)
return execute_native(q, consistency_level=self._consistency)
def batch(self, batch_obj):
if batch_obj is not None and not isinstance(batch_obj, BatchQuery):

View File

@@ -8,9 +8,9 @@ class BaseClauseTests(TestCase):
ss = BaseClause('a', 'b')
assert ss.get_context_size() == 1
ctx = {}
ctx = []
ss.set_context_id(10)
ss.update_context(ctx)
assert ctx == {'10': 'b'}
assert ctx == ['10': 'b']