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

Update get_or_create method and its tests

Browse Source
This commit is contained in:
Mojix Coder
2022-02-06 19:14:17 +03:30
parent 2517572a14
commit 4431a643d5
3 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ormar/queryset/queryset.py
View File

@ -998,10 +998,8 @@ class QuerySet(Generic[T]):
try:
return await self.get(*args, **kwargs), False
except NoMatch:
if _defaults is None:
return await self.create(**kwargs), True
else:
return await self.create(**kwargs, **_defaults), True
_defaults = _defaults or {}
return await self.create(**{**kwargs, **_defaults}), True
async def update_or_create(self, **kwargs: Any) -> "T":
"""

6
ormar/relations/querysetproxy.py
View File

@ -507,10 +507,8 @@ class QuerysetProxy(Generic[T]):
try:
return await self.get(*args, **kwargs), False
except NoMatch:
if _defaults is None:
return await self.create(**kwargs), True
else:
return await self.create(**kwargs, **_defaults), True
_defaults = _defaults or {}
return await self.create(**{**kwargs, **_defaults}), True
async def update_or_create(self, **kwargs: Any) -> "T":
"""