Prodream/ormar
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix collerk/ormar#588 bug in querysetproxy count() method

Browse Source
This commit is contained in:
haydeec1
2022-03-17 14:07:48 -04:00
parent a7e78bbf8c
commit 606a592038
3 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ormar/protocols/queryset_protocol.py
View File

@ -26,7 +26,7 @@ class QuerySetProtocol(Protocol): # pragma: nocover
async def exists(self) -> bool:
...
async def count(self) -> int:
async def count(self, distinct: bool = True) -> int:
...
async def clear(self) -> int:

6
ormar/queryset/queryset.py
View File

@ -682,6 +682,12 @@ class QuerySet(Generic[T]):
"""
Returns number of rows matching the given criteria
(applied with `filter` and `exclude` if set before).
If `distinct` is `True` (the default), this will return the number of primary rows selected. If `False`,
the count will be the total number of rows returned
(including extra rows for `one-to-many` or `many-to-many` left `select_related` table joins).
`False` is the legacy (buggy) behavior for workflows that depend on it.
:param distinct: flag if the primary table rows should be distinct or not
:return: number of rows
:rtype: int

9
ormar/relations/querysetproxy.py
View File

@ -193,17 +193,22 @@ class QuerysetProxy(Generic[T]):
"""
return await self.queryset.exists()
async def count(self) -> int:
async def count(self, distinct: bool = True) -> int:
"""
Returns number of rows matching the given criteria
(applied with `filter` and `exclude` if set before).
If `distinct` is `True` (the default), this will return the number of primary rows selected. If `False`,
the count will be the total number of rows returned
(including extra rows for `one-to-many` or `many-to-many` left `select_related` table joins).
`False` is the legacy (buggy) behavior for workflows that depend on it.
Actual call delegated to QuerySet.
:param distinct: flag if the primary table rows should be distinct or not
:return: number of rows
:rtype: int
"""
return await self.queryset.count()
return await self.queryset.count(distinct=distinct)
async def max(self, columns: Union[str, List[str]]) -> Any: # noqa: A003
"""