1#@!#!123s

: / opt / cloudlinux / venv / lib / python3.11 / site-packages / lvestats / plugins / generic / burster /
Filename : utils.py
back
# 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