11 KiB
Making Queries
cqlengine.query
Retrieving objects
Once you've populated Cassandra with data, you'll probably want to retrieve some of it. This is accomplished with QuerySet objects. This section will describe how to use QuerySet objects to retrieve the data you're looking for.
Retrieving all objects
The simplest query you can make is to return all objects from a table.
This is accomplished with the
.all()method, which returns a QuerySet of all objects in a tableUsing the Person example model, we would get all Person objects like this:
all_objects = Person.objects.all()
Retrieving objects with filters
Typically, you'll want to query only a subset of the records in your database.
That can be accomplished with the QuerySet's
.filter(\*\*)method.For example, given the model definition:
class Automobile(Model): manufacturer = columns.Text(primary_key=True) year = columns.Integer(primary_key=True) model = columns.Text() price = columns.Decimal()...and assuming the Automobile table contains a record of every car model manufactured in the last 20 years or so, we can retrieve only the cars made by a single manufacturer like this:
q = Automobile.objects.filter(manufacturer='Tesla')We can then further filter our query with another call to .filter
q = q.filter(year=2012)Note: all queries involving any filtering MUST define either an '=' or an 'in' relation to either a primary key column, or an indexed column.
Accessing objects in a QuerySet
There are several methods for getting objects out of a queryset
- iterating over the queryset
for car in Automobile.objects.all(): #...do something to the car instance pass
- list index
q = Automobile.objects.all() q[0] #returns the first result q[1] #returns the second result
- list slicing
q = Automobile.objects.all() q[1:] #returns all results except the first q[1:9] #returns a slice of the resultsNote: CQL does not support specifying a start position in it's queries. Therefore, accessing elements using array indexing / slicing will load every result up to the index value requested
- calling
get() <query.QuerySet.get>on the querysetq = Automobile.objects.filter(manufacturer='Tesla') q = q.filter(year=2012) car = q.get()this returns the object matching the queryset
- calling
first() <query.QuerySet.first>on the querysetq = Automobile.objects.filter(manufacturer='Tesla') q = q.filter(year=2012) car = q.first()this returns the first value in the queryset
Filtering Operators
Equal To <query.QueryOperator.EqualsOperator>The default filtering operator.
q = Automobile.objects.filter(manufacturer='Tesla') q = q.filter(year=2012) #year == 2012In addition to simple equal to queries, cqlengine also supports querying with other operators by appending a
__<op>to the field name on the filtering call
in (__in) <query.QueryOperator.InOperator>q = Automobile.objects.filter(manufacturer='Tesla') q = q.filter(year__in=[2011, 2012])
> (__gt) <query.QueryOperator.GreaterThanOperator>q = Automobile.objects.filter(manufacturer='Tesla') q = q.filter(year__gt=2010) # year > 2010
>= (__gte) <query.QueryOperator.GreaterThanOrEqualOperator>q = Automobile.objects.filter(manufacturer='Tesla') q = q.filter(year__gte=2010) # year >= 2010
< (__lt) <query.QueryOperator.LessThanOperator>q = Automobile.objects.filter(manufacturer='Tesla') q = q.filter(year__lt=2012) # year < 2012
<= (__lte) <query.QueryOperator.LessThanOrEqualOperator>q = Automobile.objects.filter(manufacturer='Tesla') q = q.filter(year__lte=2012) # year <= 2012
TimeUUID Functions
In addition to querying using regular values, there are two functions you can pass in when querying TimeUUID columns to help make filtering by them easier. Note that these functions don't actually return a value, but instruct the cql interpreter to use the functions in it's query.
returns the minimum time uuid value possible for the given datetime
returns the maximum time uuid value possible for the given datetime
Example
class DataStream(Model): time = cqlengine.TimeUUID(primary_key=True) data = cqlengine.Bytes() min_time = datetime(1982, 1, 1) max_time = datetime(1982, 3, 9) DataStream.filter(time__gt=cqlengine.MinTimeUUID(min_time), time__lt=cqlengine.MaxTimeUUID(max_time))
Token Function
Token functon may be used only on special, virtual column pk__token, representing token of partition key (it also works for composite partition keys). Cassandra orders returned items by value of partition key token, so using cqlengine.Token we can easy paginate through all table rows.
Example
class Items(Model): id = cqlengine.Text(primary_key=True) data = cqlengine.Bytes() query = Items.objects.all().limit(10) first_page = list(query); last = first_page[-1] next_page = list(query.filter(pk__token__gt=cqlengine.Token(last.pk)))
QuerySets are imutable
When calling any method that changes a queryset, the method does not actually change the queryset object it's called on, but returns a new queryset object with the attributes of the original queryset, plus the attributes added in the method call.
Example
#this produces 3 different querysets #q does not change after it's initial definition q = Automobiles.objects.filter(year=2012) tesla2012 = q.filter(manufacturer='Tesla') honda2012 = q.filter(manufacturer='Honda')
Ordering QuerySets
Since Cassandra is essentially a distributed hash table on steroids, the order you get records back in will not be particularly predictable.
However, you can set a column to order on with the
.order_by(column_name)method.Example
#sort ascending q = Automobiles.objects.all().order_by('year') #sort descending q = Automobiles.objects.all().order_by('-year')Note: Cassandra only supports ordering on a clustering key. In other words, to support ordering results, your model must have more than one primary key, and you must order on a primary key, excluding the first one.
For instance, given our Automobile model, year is the only column we can order on.
Values Lists
There is a special QuerySet's method
.values_list()- when called, QuerySet returns lists of values instead of model instances. It may significantly speedup things with lower memory footprint for large responses. Each tuple contains the value from the respective field passed into thevalues_list()call — so the first item is the first field, etc. For example:
Batch Queries
cqlengine now supports batch queries using the BatchQuery class. Batch queries can be started and stopped manually, or within a context manager. To add queries to the batch object, you just need to precede the create/save/delete call with a call to batch, and pass in the batch object.
You can only create, update, and delete rows with a batch query, attempting to read rows out of the database with a batch query will fail.
from cqlengine import BatchQuery #using a context manager with BatchQuery() as b: now = datetime.now() em1 = ExampleModel.batch(b).create(example_type=0, description="1", created_at=now) em2 = ExampleModel.batch(b).create(example_type=0, description="2", created_at=now) em3 = ExampleModel.batch(b).create(example_type=0, description="3", created_at=now) # -- or -- #manually b = BatchQuery() now = datetime.now() em1 = ExampleModel.batch(b).create(example_type=0, description="1", created_at=now) em2 = ExampleModel.batch(b).create(example_type=0, description="2", created_at=now) em3 = ExampleModel.batch(b).create(example_type=0, description="3", created_at=now) b.execute()
QuerySet method reference
all()
Returns a queryset matching all rows
count()
Returns the number of matching rows in your QuerySet
filter(**values)
- param values
-
See
retrieving-objects-with-filters
Returns a QuerySet filtered on the keyword arguments
get(**values)
- param values
-
See
retrieving-objects-with-filters
Returns a single object matching the QuerySet. If no objects are
matched, a ~models.Model.DoesNotExist exception is raised. If
more than one object is found, a ~models.Model.MultipleObjectsReturned exception is
raised.
limit(num)
Limits the number of results returned by Cassandra.
Note that CQL's default limit is 10,000, so all queries without a limit set explicitly will have an implicit limit of 10,000
order_by(field_name)
- param field_name
-
the name of the field to order on. Note: the field_name must be a clustering key
- type field_name
-
string
Sets the field to order on.
allow_filtering()
Enables the (usually) unwise practive of querying on a clustering key without also defining a partition key