steffo/async-chain
steffo/async-chain
1
Fork 0
mirror of https://github.com/Steffo99/async-chain.git synced 2024-11-23 21:44:22 +00:00
Code Activity

🔧 Use __slots__ on Chain classes

Browse source
This commit is contained in:
Steffo 2021-06-14 04:32:00 +02:00
parent 361c310ec6
commit 82fc6a84ec
Signed by: steffo
GPG key ID: 6965406171929D01
1 changed files with 29 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
async_chain/chain.py
View file

@ -1,38 +1,40 @@
from __future__ import annotations
import inspect
import typing as t
import abc
class Chain(metaclass=abc.ABCMeta):
def __init__(self):
pass
__slots__ = []
def __getattr__(self, item):
def __getattr__(self, item) -> ChainGetAttr:
return ChainGetAttr(previous=self, item=item)
def __getitem__(self, item):
def __getitem__(self, item) -> ChainGetItem:
return ChainGetItem(previous=self, item=item)
def __call__(self, *args, **kwargs):
def __call__(self, *args, **kwargs) -> ChainCall:
return ChainCall(previous=self, args=args, kwargs=kwargs)
def __await__(self):
return self.__evaluate__().__await__()
@abc.abstractmethod
def __repr__(self):
def __repr__(self) -> str:
raise NotImplementedError()
@abc.abstractmethod
def __display__(self):
def __display__(self) -> str:
raise NotImplementedError()
@abc.abstractmethod
async def __evaluate__(self):
async def __evaluate__(self) -> t.Any:
raise NotImplementedError()
class ChainStart(Chain):
__slots__ = ("__start__",)
def __init__(self, start: t.Any):
super().__init__()
self.__start__: t.Any = start
@ -48,6 +50,8 @@ class ChainStart(Chain):
class ChainNode(Chain, metaclass=abc.ABCMeta):
__slots__ = ("__previous__",)
def __init__(self, previous: Chain):
super().__init__()
self.__previous__: Chain = previous
@ -57,17 +61,19 @@ class ChainNode(Chain, metaclass=abc.ABCMeta):
class ChainGetAttr(ChainNode):
__slots__ = ("__item__",)
def __init__(self, previous: Chain, item: str):
super().__init__(previous=previous)
self._item: str = item
self.__item__: str = item
def __display__(self):
return f"{self.__previous__.__display__()}.{self._item!s}"
return f"{self.__previous__.__display__()}.{self.__item__!s}"
async def __evaluate__(self):
previous = await self.__previous__.__evaluate__()
current = getattr(previous, self._item)
current = getattr(previous, self.__item__)
if inspect.isawaitable(current):
return await current
@ -76,17 +82,19 @@ class ChainGetAttr(ChainNode):
class ChainGetItem(ChainNode):
__slots__ = ("__item__",)
def __init__(self, previous: Chain, item: t.Any):
super().__init__(previous=previous)
self._item: t.Any = item
self.__item__: t.Any = item
def __display__(self):
return f"{self.__previous__.__display__()}[{self._item!r}]"
return f"{self.__previous__.__display__()}[{self.__item__!r}]"
async def __evaluate__(self):
previous = await self.__previous__.__evaluate__()
current = previous[self._item]
current = previous[self.__item__]
if inspect.isawaitable(current):
return await current
@ -95,21 +103,23 @@ class ChainGetItem(ChainNode):
class ChainCall(ChainNode):
__slots__ = ("__args__", "__kwargs__",)
def __init__(self, previous: Chain, args, kwargs):
super().__init__(previous=previous)
self._args = args
self._kwargs = kwargs
self.__args__ = args
self.__kwargs__ = kwargs
def __display__(self):
args = map(lambda a: f"{a!r}", self._args)
kwargs = map(lambda k, v: f"{k!s}={v!r}", self._kwargs)
args = map(lambda a: f"{a!r}", self.__args__)
kwargs = map(lambda k, v: f"{k!s}={v!r}", self.__kwargs__)
allargs = ", ".join([*args, *kwargs])
return f"{self.__previous__.__display__()}({allargs})"
async def __evaluate__(self):
previous = await self.__previous__.__evaluate__()
current = previous(*self._args, **self._kwargs)
current = previous(*self.__args__, **self.__kwargs__)
if inspect.isawaitable(current):
return await current