# coding=utf-8 # # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2023 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT """Utility stuff which is not directly related to the plugin domain itself.""""" import functools import itertools from typing import Any, Callable, Generator, Iterator, Iterable, TypeVar from ._logs import logger _T = TypeVar('_T') # NOTE(vlebedev): Taken from https://docs.python.org/3/library/itertools.html#itertools.batched # TODO(vlebedev): Remove once codebase is updated to Python >= 3.12 def batched(iterable: Iterable[_T], n: int) -> Iterator[tuple[_T, ...]]: """batched('ABCDEFG', 3) --> ABC DEF G Batch data into tuples of length n. The last batch may be shorter.""" if n < 1: raise ValueError('n must be at least one') it = iter(iterable) while batch := tuple(itertools.islice(it, n)): yield batch def repeast_last(upstream: Iterable[_T]) -> Iterator[_T]: for item in upstream: last = item yield item while True: try: yield last except NameError: return def bootstrap_gen(func: Callable[..., Generator[Any, Any, Any]]): @functools.wraps(func) def wrapper(*args, **kwargs): g = func(*args, **kwargs) g.send(None) return g return wrapper def log_exceptions(func): @functools.wraps(func) def wrapper(*args, **kwargs): try: return func(*args, **kwargs) except Exception: logger.exception('Function %s failed!', func.__name__) raise return wrapper