Collections
Anytime your results return multiple values then an instance of
Collection is returned. This allows you to iterate over your values and has a lot of shorthand methods.When using collections as a query result you can iterate over it as if the collection with a normal list:
1
users = User.get() #== <masoniteorm.collections.Collection>
2
users.count() #== 50
3
users.pluck('email') #== <masoniteorm.collections.Collection> of emails
4
5
for user in users:
6
user.email #== '[email protected]'
Copied!
Available Methods
all
Returns the underlying list or dict represented by the collection:
1
users = User.get().all() #== [<app.User.User>, <app.User.User>]
2
3
Collection([1, 2, 3]).all() #== [1, 2, 3]
Copied!
avg
Returns the average of all items in the collection:
1
Collection([1, 2, 3, 4, 5]).avg() #== 3
Copied!
If the collection contains nested objects or dictionaries (e.g. for a collection of models), you must pass a key to use for determining which values to calculate the average:
1
average_price = Product.get().avg('price')
Copied!
chunk
Chunks a collection into multiple, smaller collections of a given size. Uses a generator to keep each chunk small. Useful for chunking large data sets where pulling too many results in memory will overload the application.
1
collection = Collection([1, 2, 3, 4, 5, 6, 7])
2
chunks = collection.chunk(2).serialize() #== [[1, 2], [3, 4], [5, 6], [7]]
Copied!
collapse
Collapses a collection of lists into a flat collection:
1
collection = Collection([[1, 2, 3], [4, 5, 6])
2
collection.collapse().serialize() #== [1, 2, 3, 4, 5, 6]
Copied!
contains
Determines whether the collection contains a given item:
1
collection = Collection(['foo', 'bar'])
2
collection.contains('foo') #== True
Copied!
You can also pass a key / value pair to the contains method, which will determine if the given pair exists in the collection.
Finally, you may also pass a callback to the contains method to perform your own truth test:
1
collection = Collection([1, 2, 3, 4, 5])
2
collection.contains(lambda item: item > 5) #== False
Copied!
count
Returns the total number of items in the collection.
len() standard python method can also be used.diff
Returns the difference as a collection against another collection
1
collection = Collection([1, 2, 3, 4, 5])
2
diff = collection.diff([2, 4, 6, 8])
3
diff.all() #== [1, 3, 5]
Copied!
each
Iterates over the items in the collection and passes each item to a given callback:
1
posts.each(lambda post: post.author().save(author))
Copied!
every
Creates a new collection by applying a given callback on every element:
1
collection = Collection([1, 2, 3])
2
collection.every(lambda x: x*2 ).all() #== [2, 4, 6]
Copied!
filter
Filters the collection by a given callback, keeping only those items that pass a given truth test:
1
collection = Collection([1, 2, 3, 4])
2
filtered = collection.filter(lambda item: item > 2)
3
filtered.all() #== [3, 4]
Copied!
first
Returns the first item of the collection, if no arguments are given.
When given a truth test as callback, it returns the first element in the collection that passes the test:
1
collection = Collection([1, 2, 3, 4])
2
collection.first(lambda item: item > 2)
Copied!
flatten
Flattens a multi-dimensional collection into a single dimension:
1
collection = Collection([1, 2, [3, 4, 5, {'foo': 'bar'}]])
2
flattened = collection.flatten().all() #== [1, 2, 3, 4, 5, 'bar']
Copied!
forget
Removes an item from the collection by its key:
1
collection = Collection([1, 2, 3, 4, 5])
2
collection.forget(1).all() #== [1,3,4,5]
3
collection.forget(0,2).all() #== [3,5]
Copied!
Unlike most other collection methods,
forget does not return a new modified collection; it modifies the collection it is called on.for_page
Paginates the collection by returning a new collection containing the items that would be present on a given page number:
1
collection = Collection([1, 2, 3, 4, 5, 6, 7, 8, 9])
2
chunk = collection.for_page(2, 4).all() #== 4, 5, 6, 7
Copied!
for_page(page, count) takes the page number and the number of items to show per page.get
Returns the item at a given key or index. If the key does not exist, None is returned. An optional default value can be passed as the second argument:
1
collection = Collection([1, 2, 3])
2
collection.get(0) #== 1
3
collection.get(4) #== None
4
collection.get(4, 'default') #== 'default'
5
6
collection = Collection({"apples": 1, "cherries": 2})
7
collection.get("apples") #== 1
Copied!
group_by
Returns a collection where items are grouped by the given key:
1
collection = Collection([
2
{"id": 1, "type": "a"},
3
{"id": 2, "type": "b"},
4
{"id": 3, "type": "a"}
5
])
6
collection.implode("type").all()
7
#== {'a': [{'id': 1, 'type': 'a'}, {'id': 4, 'type': 'a'}],
8
# 'b': [{'id': 2, 'type': 'b'}]}
Copied!
implode
Joins the items in a collection with
, or the given glue string.1
collection = Collection(['foo', 'bar', 'baz'])
2
collection.implode() #== foo,bar,baz
3
collection.implode('-') #== foo-bar-baz
Copied!
If the collection contains dictionaries or objects, you must pass the key of the attributes you wish to join:
1
collection = Collection([
2
{'account_id': 1, 'product': 'Desk'},
3
{'account_id': 2, 'product': 'Chair'}
4
])
5
collection.implode(key='product') #== Desk,Chair
6
collection.implode(" - ", key='product') #== Desk - Chair
Copied!
is_empty
Returns
True if the collection is empty; otherwise, False is returned:1
Collection([]).is_empty() #== True
Copied!
last
Returns the last element in the collection if no arguments are given.
Returns the last element in the collection that passes the given truth test:
1
collection = Collection([1, 2, 3, 4])
2
last = collection.last(lambda item: item < 3) #== 2
Copied!
map
Iterates through the collection and passes each value to the given callback. The callback is free to modify the item and return it, thus forming a new collection of modified items:
1
collection = Collection([1, 2, 3, 4])
2
multiplied = collection.map(lambda item: item * 2).all() #== [2, 4, 6, 8]
Copied!
map_into
Iterates through the collection and cast each value into the given class:
1
collection = Collection([1,2])
2
collection.map_into(str).all() #== ["1", "2"]
Copied!
A class method can also be specified. Some additional keywords arguments can be passed to this method:
1
class Point:
2
@classmethod
3
def as_dict(cls, coords, one_dim=False):
4
if one_dim:
5
return {"X": coords[0]}
6
return {"X": coords[0], "Y": coords[1]}
7
8
collection = Collection([(1,2), (3,4)])
9
collection.map_into(Point, "as_dict") #== [{'X': 1, 'Y': 2}, {'X': 3, 'Y': 4}]
10
collection.map_into(Point, "as_dict", one_dim=True) #== [{'X': 1}, {'X': 3}]
Copied!
max
Retrieves max value of the collection:
1
collection = Collection([1,2,3])
2
collection.max() #== 3
Copied!
If the collection contains dictionaries or objects, you must pass the key on which to compute max value:
1
collection = Collection([
2
{'product_id': 1, 'product': 'Desk'},
3
{'product_id': 2, 'product': 'Chair'}
4
{'product_id': 3, 'product': 'Table'}
5
])
6
collection.max("product_id") #== 3
Copied!
merge
Merges the given list into the collection:
1
collection = Collection(['Desk', 'Chair'])
2
collection.merge(['Bookcase', 'Door'])
3
collection.all() #== ['Desk', 'Chair', 'Bookcase', 'Door']
Copied!
Unlike most other collection methods,
merge does not return a new modified collection; it modifies the collection it is called on.pluck
Retrieves all of the collection values for a given key:
1
collection = Collection([
2
{'product_id': 1, 'product': 'Desk'},
3
{'product_id': 2, 'product': 'Chair'}
4
{'product_id': 3, 'product': None}
5
])
6
7
plucked = collection.pluck('product').all() #== ['Desk', 'Chair', None]
Copied!
A key can be given to pluck the collection into a dictionary with the given key
1
collection.pluck("product", "product_id") #== {1: 'Desk', 2: 'Chair', 3: None}
Copied!
You can pass
keep_nulls=False to remove None value in the collection.1
collection.pluck("product", keep_nulls=False) #== ['Desk', 'Chair']
Copied!
pop
Removes and returns the last item from the collection:
1
collection = Collection([1, 2, 3, 4, 5])
2
collection.pop() #== 5
3
collection.all() #== [1, 2, 3, 4]
Copied!
prepend
Adds an item to the beginning of the collection:
1
collection = Collection([1, 2, 3, 4])
2
collection.prepend(0)
3
collection.all() #== [0, 1, 2, 3, 4]
Copied!
pull
Removes and returns an item from the collection by its key:
1
collection = Collection([1, 2, 3, 4])
2
collection.pull(1) #== 2
3
collection.all() #== [1, 3, 4]
4
5
collection = Collection({'apple': 1, 'cherry': 3, 'lemon': 2})
6
collection.pull('cherry') #== 3
7
collection.all() #== {'apple': 1, 'lemon': 2}
Copied!
push
Appends an item to the end of the collection:
1
collection = Collection([1, 2, 3, 4])
2
collection.push(5)
3
collection.all() #== [1, 2, 3, 4, 5]
Copied!
put
Sets the given key and value in the collection:
1
collection = Collection([1, 2, 3, 4])
2
collection.put(1, 5)
3
collection.all() #== [1, 5, 3, 4]
4
5
collection = Collection({'apple': 1, 'cherry': 3, 'lemon': 2})
6
collection.put('cherry', 0)
7
collection.all() #== {'apple': 1, 'cherry': 0, 'lemon': 2}
Copied!
random
Returns a random item from the collection
1
user = User.all().random() #== returns a random User instance
Copied!
An integer count can be given to
random method to specify how many items you would like to randomly retrieve from the collection. A collection will always be returned when the items count is specified1
users = User.all().random(3) #== returns a Collection of 3 users
2
users.count() #== 3
3
users.all() #== returns a list of 3 users
Copied!
If the collection length is smaller than specified count a
ValueError will be raised.reduce
Reduces the collection to a single value, passing the result of each iteration into the subsequent iteration.
1
collection = Collection([1, 2, 3])
2
collection.reduce(lambda result, item: (result or 0) + item) #== 6
Copied!
Initial value is
0 by default but can be overridden:1
collection.reduce(lambda result, item: (result or 0) + item, 4) #== 10
Copied!
reject
It's the inverse of filter method. It filters the collection using the given callback. The callback should return
True for any items to remove from the resulting collection:1
collection = Collection([1, 2, 3, 4])
2
filtered = collection.reject(lambda item: item > 2)
3
filtered.all() #== [1, 2]
Copied!
Unlike most other collection methods,
reject does not return a new modified collection; it modifies the collection it is called on.reverse
Reverses the order of the items in the collection:
1
collection = Collection([1, 2, 3])
2
collection.reverse().all() #== [3, 2, 1]
Copied!
Unlike most other collection methods,
reverse does not return a new modified collection; it modifies the collection it is called on.serialize
Converts the collection into a list. If the collection’s values are ORM models, the models will also be converted to dictionaries:
1
collection = Collection([1, 2, 3])
2
collection.serialize() #== [1, 2, 3]
3
4
collection = Collection([User.find(1)])
5
collection.serialize() #== [{'id': 1, 'name': 'John', 'email': '[email protected]'}]
Copied!
Be careful,
serialize also converts all of its nested objects. If you want to get the underlying items as is, use the all method instead.shift
Removes and returns the first item from the collection:
1
collection = Collection([1, 2, 3, 4, 5])
2
collection.shift() #== 1
3
collection.all() #== [2, 3, 4, 5]
Copied!
sort
Sorts the collection:
1
collection = Collection([5, 3, 1, 2, 4])
2
sorted = collection.sort()
3
sorted.all() #== [1, 2, 3, 4, 5]
Copied!
sum
Returns the sum of all items in the collection:
1
Collection([1, 2, 3, 4, 5]).sum() #== 15
Copied!
If the collection contains dictionaries or objects, you must pass a key to use for determining which values to sum:
1
collection = Collection([
2
{'name': 'JavaScript: The Good Parts', 'pages': 176},
3
{'name': 'JavaScript: The Defnitive Guide', 'pages': 1096}
4
])
5
collection.sum('pages') #== 1272
Copied!
take
Returns a new collection with the specified number of items:
1
collection = Collection([0, 1, 2, 3, 4, 5])
2
chunk = collection.take(3)
3
chunk.all() #== [0, 1, 2]
Copied!
You can also pass a negative integer to take the specified amount of items from the end of the collection:
1
chunk = collection.chunk(-2)
2
chunk.all() #== [4, 5]
Copied!
to_json
Converts the collection into JSON:
1
collection = Collection([{'name': 'Desk', 'price': 200}])
2
collection.to_json() #== '[{"name": "Desk", "price": 200}]'
Copied!
transform
Iterates over the collection and calls the given callback with each item in the collection. The items in the collection will be replaced by the values returned by the callback:
1
collection = Collection([1, 2, 3, 4, 5])
2
collection.transform(lambda item: item * 2)
3
collection.all() #== [2, 4, 6, 8, 10]
Copied!
unique
Returns all of the unique items in the collection:
1
collection = Collection([1, 1, 2, 2, 3, 4, 2])
2
unique = collection.unique()
3
unique.all() #== [1, 2, 3, 4]
Copied!
When dealing with dictionaries or objects, you can specify the key used to determine uniqueness:
1
collection = Collection([
2
{'name': 'Sam', 'role': 'admin'},
3
{'name': 'Joe', 'role': 'basic'},
4
{'name': 'Joe', 'role': 'admin'},
5
])
6
unique = collection.unique('name')
7
unique.all()
8
# [
9
# {'name': 'Sam', 'role': 'admin'},
10
# {'name': 'Joe', 'role': 'basic'}
11
# ]
Copied!
where
Filters the collection by a given key / value pair:
1
collection = Collection([
2
{'name': 'Desk', 'price': 200},
3
{'name': 'Chair', 'price': 100},
4
{'name': 'Bookcase', 'price': 150},
5
{'name': 'Door', 'price': 100},
6
])
7
filtered = collection.where('price', 100)
8
filtered.all()
9
# [
10
# {'name': 'Chair', 'price': 100},
11
# {'name': 'Door', 'price': 100}
12
# ]
Copied!
zip
Merges together the values of the given list with the values of the collection at the corresponding index:
1
collection = Collection(['Chair', 'Desk'])
2
zipped = collection.zip([100, 200])
3
zipped.all() #== [('Chair', 100), ('Desk', 200)]
Copied!
Last modified 2mo ago