Courts registry¶

def get_court(code: str) -> Court | None:
    """Look up a court by its code (e.g. 'delhi', 'bombay-nagpur', 'sci')."""
    return _BY_CODE.get(code.lower())

def get_court_by_name(name: str) -> Court | None:
    """Look up a court by name (case-insensitive)."""
    return _BY_NAME.get(name.lower())

def get_court_by_state_code(state_code: str) -> Court | None:
    """Look up the principal court for a state code (bench variants excluded).

    The archive bucket partitions HCs by state code; this resolves the
    canonical court for that partition. Returns None for state_code "0"
    (Supreme Court — use ``SUPREME_COURT`` directly).
    """
    return _BY_STATE_CODE.get(str(state_code))

def infer_court_from_cnr(cnr: str | None) -> Court | None:
    """Identify the court that issued a CNR from its 4-letter prefix.

    The CNR's prefix is deterministic per court (verified against the AWS
    archive for all 25 HCs + SCI). Returns ``None`` for unknown / malformed
    inputs — never raises.

    Useful for routing archive queries: passing the CNR alone scans every
    HC partition (slow), but ``infer_court_from_cnr(cnr)`` lets the caller
    narrow to one bucket / partition first.
    """
    if not cnr or len(cnr) < 4:
        return None
    code = _CNR_PREFIX_TO_COURT_CODE.get(cnr[:4].upper())
    return get_court(code) if code else None

def list_all_courts() -> list[Court]:
    """Return all courts including Supreme Court."""
    return list(ALL_COURTS)

def list_high_courts() -> list[Court]:
    """Return all High Courts."""
    return list(HIGH_COURTS)