Accept callables for scroll_offset, min_brightness and max_brightness.

Author: jonathanslenders

prompt_toolkit/layout/menus.py

@@ -1,6 +1,17 @@
 import math
 from itertools import zip_longest
-from typing import TYPE_CHECKING, Dict, Iterable, List, Optional, Tuple, TypeVar, cast
+from typing import (
+    TYPE_CHECKING,
+    Callable,
+    Dict,
+    Iterable,
+    List,
+    Optional,
+    Tuple,
+    TypeVar,
+    Union,
+    cast,
+)
 
 from prompt_toolkit.application.current import get_app
 from prompt_toolkit.buffer import CompletionState
@@ -261,7 +272,7 @@ class CompletionsMenu(ConditionalContainer):
     def __init__(
         self,
         max_height: Optional[int] = None,
-        scroll_offset: int = 0,
+        scroll_offset: Union[int, Callable[[], int]] = 0,
         extra_filter: FilterOrBool = True,
         display_arrows: FilterOrBool = False,
         z_index: int = 10 ** 8,

prompt_toolkit/styles/style_transformation.py

@@ -15,7 +15,7 @@
 
 from prompt_toolkit.cache import memoized
 from prompt_toolkit.filters import FilterOrBool, to_filter
-from prompt_toolkit.utils import to_float, to_str
+from prompt_toolkit.utils import AnyFloat, to_float, to_str
 
 from .base import ANSI_COLOR_NAMES, Attrs
 from .style import parse_color
@@ -153,7 +153,7 @@ class AdjustBrightnessStyleTransformation(StyleTransformation):
     """
 
     def __init__(
-        self, min_brightness: float = 0.0, max_brightness: float = 1.0
+        self, min_brightness: AnyFloat = 0.0, max_brightness: AnyFloat = 1.0
     ) -> None:
 
         self.min_brightness = min_brightness

prompt_toolkit/utils.py

@@ -28,6 +28,7 @@
     "take_using_weights",
     "to_str",
     "to_int",
+    "AnyFloat",
     "to_float",
 ]
 
@@ -284,7 +285,10 @@ def to_int(value: Union[Callable[[], int], int]) -> int:
         return int(value)
 
 
-def to_float(value: Union[Callable[[], float], float]) -> float:
+AnyFloat = Union[Callable[[], float], float]
+
+
+def to_float(value: AnyFloat) -> float:
     " Turn callable or float into float. "
     if callable(value):
         return to_float(value())