implement count functionality

Signed-off-by: wiseaidev <[email protected]>
redis · Aug 8, 2022 · 8b6fe19 · 8b6fe19
1 parent dcd84e0
commit 8b6fe19
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/aredis_om/model/model.py b/aredis_om/model/model.py
@@ -760,6 +760,9 @@ async def all(self, batch_size=10):
             return await query.execute()
         return await self.execute()
 
+    async def count(self, batch_size=10):
+        return len(await self.all(batch_size))
+
     def sort_by(self, *fields: str):
         if not fields:
             return self

diff --git a/tests/test_hash_model.py b/tests/test_hash_model.py
@@ -95,6 +95,11 @@ async def members(m):
 
     yield member1, member2, member3
 
+@py_test_mark_asyncio
+async def test_count_query(members, m):
+
+    count = await m.Member.find(m.Member.first_name="Kim").count()
+    assert count == 1
 
 @py_test_mark_asyncio
 async def test_exact_match_queries(members, m):