RGBCube/GitHubWrapper
1
Fork 0
mirror of https://github.com/RGBCube/GitHubWrapper synced 2025-05-18 15:05:08 +00:00
Code Issues Activity Actions

Fixed walrus, for real for real

Browse source
This commit is contained in:
VarMonke 2022-05-02 00:53:30 +05:30
parent 456d2f9916
commit b2091313a9
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
github/client.py
View file

@ -190,14 +190,16 @@ class GHClient:
@functools.wraps(func) @functools.wraps(func)
async def wrapped(self: Self, *args: P.args, **kwargs: P.kwargs) -> Optional[Union[T, User, Repository]]: async def wrapped(self: Self, *args: P.args, **kwargs: P.kwargs) -> Optional[Union[T, User, Repository]]:
if type == 'user': if type == 'user':
if obj := self._user_cache.get(kwargs.get('user')): obj = self._user_cache.get(kwargs.get('user'))
if obj:
return obj return obj
user: User = await func(self, *args, **kwargs) # type: ignore user: User = await func(self, *args, **kwargs) # type: ignore
self._user_cache[kwargs.get("user")] = user self._user_cache[kwargs.get("user")] = user
return user return user
if type == 'repo': if type == 'repo':
if obj := self._repo_cache.get(kwargs.get('repo')): obj = self._repo_cache.get(kwargs.get('repo'))
if obj:
return obj return obj
repo: Repository = await func(self, *args, **kwargs) # type: ignore repo: Repository = await func(self, *args, **kwargs) # type: ignore