Source code for autobean_refactor.models.generated.tolerance

# DO NOT EDIT
# This file is automatically generated by autobean_refactor.modelgen.

import decimal
from typing import Iterator, final
from typing_extensions import Self
from .. import base, internal
from ..number_expr import NumberExpr
from ..spacing import Whitespace


@internal.token_model
class Tilde(internal.SimpleDefaultRawTokenModel):
    """Contains literal `~`."""
    RULE = 'TILDE'
    DEFAULT = '~'


@internal.tree_model
class Tolerance(base.RawTreeModel, internal.SpacingAccessorsMixin):
    """Tolerance (e.g. `~ 0.01`)."""
    RULE = 'tolerance'
    INLINE = True

    _tilde = internal.required_field[Tilde]()
    _number = internal.required_field[NumberExpr]()

    raw_number = internal.required_node_property(_number)

    number = internal.required_value_property(raw_number)

    @final
    def __init__(
            self,
            token_store: base.TokenStore,
            tilde: Tilde,
            number: NumberExpr,
    ):
        super().__init__(token_store)
        self._tilde = tilde
        self._number = number

    @property
    def first_token(self) -> base.RawTokenModel:
        return self._tilde.first_token

    @property
    def last_token(self) -> base.RawTokenModel:
        return self._number.last_token

    def clone(self, token_store: base.TokenStore, token_transformer: base.TokenTransformer) -> Self:
        return type(self)(
            token_store,
            type(self)._tilde.clone(self._tilde, token_store, token_transformer),
            type(self)._number.clone(self._number, token_store, token_transformer),
        )

    def _reattach(self, token_store: base.TokenStore, token_transformer: base.TokenTransformer) -> None:
        self._token_store = token_store
        self._tilde = type(self)._tilde.reattach(self._tilde, token_store, token_transformer)
        self._number = type(self)._number.reattach(self._number, token_store, token_transformer)

    def _eq(self, other: base.RawTreeModel) -> bool:
        return (
            isinstance(other, Tolerance)
            and self._tilde == other._tilde
            and self._number == other._number
        )

    @classmethod
    def from_children(
            cls,
            number: NumberExpr,
    ) -> Self:
        tilde = Tilde.from_default()
        tokens = [
            *tilde.detach(),
            Whitespace.from_default(),
            *number.detach(),
        ]
        token_store = base.TokenStore.from_tokens(tokens)
        cls._tilde.reattach(tilde, token_store)
        cls._number.reattach(number, token_store)
        return cls(token_store, tilde, number)

    @classmethod
    def from_value(
            cls,
            number: decimal.Decimal,
    ) -> Self:
        return cls.from_children(
            number=NumberExpr.from_value(number),
        )

    def auto_claim_comments(self) -> None:
        type(self)._number.auto_claim_comments(self._number)

    def iter_children_formatted(self) -> Iterator[tuple[base.RawModel, bool]]:
        yield from type(self)._tilde.iter_children_formatted(self._tilde, False)
        yield Whitespace.from_default(), False
        yield from type(self)._number.iter_children_formatted(self._number, False)