File size: 677 Bytes
8287b2e 6e18947 345e36f fe70438 6e18947 345e36f 6e18947 345e36f 6e18947 345e36f 6e18947 345e36f 6e18947 345e36f fe70438 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from typing import Any, Dict, List
from .dataclass import Dataclass, OptionalField
from .utils import recursive_copy
class ReusableGenerator(Dataclass):
generator: callable
gen_argv: List[Any] = OptionalField(default_factory=list)
gen_kwargs: Dict[str, Any] = OptionalField(default_factory=dict)
def activate(self):
return self.generator(*self.gen_argv, **self.gen_kwargs)
def __iter__(self):
yield from self.activate()
def __call__(self):
yield from iter(self)
class CopyingReusableGenerator(ReusableGenerator):
def __iter__(self):
for instance in self.activate():
yield recursive_copy(instance)
|