Function _abstract_atom_init

tables/atom.py:609–631

Return a constructor for an abstract `Atom` class.

(
    deftype: str, defvalue: Any
)

def _abstract_atom_init(
    deftype: str, defvalue: Any
) -> Callable[[Atom, int | None, Shape, Any], None]:
    """Return a constructor for an abstract `Atom` class."""
    defitemsize = split_type(deftype)[1]

    def __init__(  # noqa: N807
        self: Atom,
        itemsize: int | None = defitemsize,
        shape: Shape = (),
        dflt: Any = defvalue,
    ) -> None:
        assert self.kind in atom_map
        try:
            atomclass = atom_map[self.kind][itemsize]
        except KeyError:
            raise _invalid_itemsize_error(
                self.kind, itemsize, atom_map[self.kind]
            )
        self.__class__ = atomclass
        atomclass.__init__(self, shape, dflt)

    return __init__


class StringAtom(Atom):  # type: ignore[misc]