openstack/deb-python-cassandra-driver
Code Issues Proposed changes

refactoring count query

Browse Source
This commit is contained in:
Blake Eggleston
2013-10-27 16:22:31 -07:00
parent ec2337858c
commit 3556c99724
1 changed files with 4 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
cqlengine/query.py
View File

@@ -539,18 +539,12 @@ class AbstractQuerySet(object):
""" Returns the number of rows matched by this query """ """ Returns the number of rows matched by this query """
if self._batch: if self._batch:
raise CQLEngineException("Only inserts, updates, and deletes are available in batch mode") raise CQLEngineException("Only inserts, updates, and deletes are available in batch mode")
#TODO: check for previous query execution and return row count if it exists #TODO: check for previous query execution and return row count if it exists
if self._result_cache is None: if self._result_cache is None:
qs = ['SELECT COUNT(*)'] query = self._select_query()
qs += ['FROM {}'.format(self.column_family_name)] query.count = True
if self._where: _, result = execute(str(query), query.get_context())
qs += ['WHERE {}'.format(self._where_clause())]
if self._allow_filtering:
qs += ['ALLOW FILTERING']
qs = ' '.join(qs)
_, result = execute(qs, self._where_values())
return result[0][0] return result[0][0]
else: else:
return len(self._result_cache) return len(self._result_cache)