divisor.cli_input

  1# SPDX-License-Identifier: MPL-2.0 AND LicenseRef-Commons-Clause-License-Condition-1.0
  2# <!-- // /*  d a r k s h a p e s */ -->d
  3
  4"""CLI helper functions for interactive input handling and state management."""
  5
  6from typing import Callable, Optional
  7
  8from nnll.console import nfo
  9
 10from divisor.controller import ManualTimestepController
 11from divisor.state import MenuState
 12
 13
 14def get_float_input(
 15    prompt: str,
 16    current_value: float,
 17    default_value: float = 0.0,
 18    allow_empty: bool = True,
 19) -> Optional[float]:
 20    """Get float input with optional reset to default.\n
 21    :param prompt: Input prompt string
 22    :param current_value: Current value to display
 23    :param default_value: Value to use if input is empty
 24    :param allow_empty: Whether empty input is allowed
 25    :returns: Parsed float value or None if invalid
 26    """
 27    try:
 28        user_input = input(prompt).strip()
 29        if allow_empty and user_input == "":
 30            return default_value
 31        return float(user_input)
 32    except ValueError:
 33        return None
 34
 35
 36def get_int_input(
 37    prompt: str,
 38    current_value: int,
 39    generate_random: Optional[Callable[[], int]] = None,
 40) -> Optional[int]:
 41    """Get integer input with optional random generation.\n
 42    :param prompt: Input prompt string
 43    :param current_value: Current value to display
 44    :param generate_random: Optional function to generate random value if input is empty
 45    :returns: Parsed integer value or None if invalid
 46    """
 47    try:
 48        user_input = input(prompt).strip()
 49        if user_input == "" and generate_random is not None:
 50            return generate_random()
 51        return int(user_input)
 52    except ValueError:
 53        return None
 54
 55
 56def handle_toggle(
 57    controller: ManualTimestepController,
 58    state: MenuState,
 59    current_value: bool,
 60    setter_func: Callable[[bool], None],
 61    clear_prediction_cache: Optional[Callable[[], None]] = None,
 62    enabled_msg: str = "ENABLED",
 63    disabled_msg: str = "DISABLED",
 64) -> MenuState:
 65    """Generic toggle handler.\n
 66    :param controller: ManualTimestepController instance
 67    :param state: Current input state
 68    :param current_value: Current boolean value
 69    :param setter_func: Controller setter method
 70    :param clear_prediction_cache: Optional function to clear cache
 71    :param enabled_msg: Message when enabled
 72    :param disabled_msg: Message when disabled
 73    :returns: Updated input state
 74    """
 75    new_value = not current_value
 76    setter_func(new_value)
 77    if clear_prediction_cache is not None:
 78        clear_prediction_cache()
 79    state = controller.current_state
 80    status = enabled_msg if new_value else disabled_msg
 81    nfo(f"{status}")
 82    return state
 83
 84
 85def handle_float_setting(
 86    controller: ManualTimestepController,
 87    state: MenuState,
 88    prompt: str,
 89    current_value: float,
 90    setter_func: Callable[[float], None],
 91    clear_prediction_cache: Callable[[], None],
 92    default_value: float = 0.0,
 93    value_name: str = "value",
 94    format_str: str = ".4f",
 95) -> MenuState:
 96    """Handle float setting with reset option.\n
 97    :param controller: ManualTimestepController instance
 98    :param state: Current input state
 99    :param prompt: Input prompt
100    :param current_value: Current float value
101    :param setter_func: Controller setter method
102    :param clear_prediction_cache: Function to clear cache
103    :param default_value: Default value for empty input
104    :param value_name: Name of the value for messages
105    :param format_str: Format string for display
106    :returns: Updated input state
107    """
108    try:
109        user_input = input(prompt).strip()
110        if user_input == "":
111            new_value = default_value
112            reset_msg = f"{value_name} reset to {default_value:{format_str}}"
113        else:
114            new_value = float(user_input)
115            reset_msg = f"{value_name} set to {new_value:{format_str}}"
116
117        setter_func(new_value)
118        clear_prediction_cache()
119        state = controller.current_state
120        nfo(reset_msg)
121    except ValueError:
122        nfo(f"Invalid {value_name}, keeping current value")
123    return state