Australian Postcodes MCP Server

""" common utilities """ from __future__ import annotations from typing import ( Any, Literal, ) def _mklbl(prefix: str, n: int): return [f"{prefix}{i}" for i in range(n)] def check_indexing_smoketest_or_raises( obj, method: Literal["iloc", "loc"], key: Any, axes: Literal[0, 1] | None = None, fails=None, ) -> None: if axes is None: axes_list = [0, 1] else: assert axes in [0, 1] axes_list = [axes] for ax in axes_list: if ax < obj.ndim: # create a tuple accessor new_axes = [slice(None)] * obj.ndim new_axes[ax] = key axified = tuple(new_axes) try: getattr(obj, method).__getitem__(axified) except (IndexError, TypeError, KeyError) as detail: # if we are in fails, the ok, otherwise raise it if fails is not None: if isinstance(detail, fails): return raise