50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
"""Util functions to help filter out similar results."""
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Callable
|
|
from datetime import datetime, timedelta
|
|
import functools
|
|
from typing import Any, TypeVar, overload
|
|
|
|
T = TypeVar("T", int, float, datetime)
|
|
|
|
|
|
@overload
|
|
def ignore_variance(
|
|
func: Callable[..., int], ignored_variance: int
|
|
) -> Callable[..., int]:
|
|
...
|
|
|
|
|
|
@overload
|
|
def ignore_variance(
|
|
func: Callable[..., float], ignored_variance: float
|
|
) -> Callable[..., float]:
|
|
...
|
|
|
|
|
|
@overload
|
|
def ignore_variance(
|
|
func: Callable[..., datetime], ignored_variance: timedelta
|
|
) -> Callable[..., datetime]:
|
|
...
|
|
|
|
|
|
def ignore_variance(func: Callable[..., T], ignored_variance: Any) -> Callable[..., T]:
|
|
"""Wrap a function that returns old result if new result does not vary enough."""
|
|
last_value: T | None = None
|
|
|
|
@functools.wraps(func)
|
|
def wrapper(*args: Any, **kwargs: Any) -> T:
|
|
nonlocal last_value
|
|
|
|
value = func(*args, **kwargs)
|
|
|
|
if last_value is not None and abs(value - last_value) < ignored_variance:
|
|
return last_value
|
|
|
|
last_value = value
|
|
return value
|
|
|
|
return wrapper
|