reflex_copy/reflex/components/chakra/forms/radio.py

"""A radio component."""
from typing import Any, Dict, List, Optional, Union

from reflex.components.chakra import ChakraComponent
from reflex.components.chakra.typography.text import Text
from reflex.components.component import Component
from reflex.components.core.foreach import Foreach
from reflex.constants import EventTriggers
from reflex.utils.types import _issubclass
from reflex.vars import Var


class RadioGroup(ChakraComponent):
    """A grouping of individual radio options."""

    tag = "RadioGroup"

    # State var to bind the the input.
    value: Optional[Var[Any]] = None

    # The default value.
    default_value: Optional[Var[Any]] = None

    # The name of the form field
    name: Optional[Var[str]] = None

    def get_event_triggers(self) -> Dict[str, Union[Var, Any]]:
        """Get the event triggers that pass the component's value to the handler.

        Returns:
            A dict mapping the event trigger to the var that is passed to the handler.
        """
        return {
            **super().get_event_triggers(),
            EventTriggers.ON_CHANGE: lambda e0: [e0],
        }

    @classmethod
    def create(cls, *children, **props) -> Component:
        """Create a radio group component.

        Args:
            *children: The children of the component.
            **props: The props of the component.

        Returns:
            The component.
        """
        if len(children) == 1 and isinstance(children[0], list):
            children = [Radio.create(child) for child in children[0]]
        if (
            len(children) == 1
            and isinstance(children[0], Var)
            and _issubclass(children[0]._var_type, List)
        ):
            children = [Foreach.create(children[0], lambda item: Radio.create(item))]
        return super().create(*children, **props)


class Radio(Text):
    """Radios are used when only one choice may be selected in a series of options."""

    tag = "Radio"

    # Value of radio.
    value: Optional[Var[Any]] = None

    # The default value.
    default_value: Optional[Var[Any]] = None

    # The color scheme.
    color_scheme: Optional[Var[str]] = None

    # If true, the radio will be initially checked.
    default_checked: Optional[Var[bool]] = None

    # If true, the radio will be checked. You'll need to pass onChange to update its value (since it is now controlled)
    is_checked: Optional[Var[bool]] = None

    # If true, the radio will be disabled.
    is_disabled: Optional[Var[bool]] = None

    # If true, the radio button will be invalid. This also sets `aria-invalid` to true.
    is_invalid: Optional[Var[bool]] = None

    # If true, the radio will be read-only
    is_read_only: Optional[Var[bool]] = None

    # If true, the radio button will be required. This also sets `aria-required` to true.
    is_required: Optional[Var[bool]] = None

    @classmethod
    def create(cls, *children, **props) -> Component:
        """Create a radio component.

        By default, the value is bound to the first child.

        Args:
            *children: The children of the component.
            **props: The props of the component.

        Returns:
            The radio component.
        """
        if "value" not in props:
            assert len(children) == 1
            props["value"] = children[0]
        return super().create(*children, **props)