Validar a Inscrição Estadual da Bahia

// IE da Bahia: confere tamanho, prefixo e dígito verificador.
// Não diz se a inscrição está ativa: isso só a consulta ao cadastro responde.
function validarIeBA(ie) {
  const n = String(ie).replace(/[\s./-]/g, "");
  if (!/^\d{8,9}$/.test(n) || /^(\d)\1+$/.test(n)) return false;
  const d = [...n].map(Number);
  const k = n.length - 2; // dígitos antes dos DVs (6 ou 7)
  const chave = n.length === 8 ? d[0] : d[1]; // decide o módulo
  const mod10 = [0, 1, 2, 3, 4, 5, 8].includes(chave);
  const calc = (s) => (mod10 ? (10 - (s % 10)) % 10 : s % 11 < 2 ? 0 : 11 - (s % 11));
  // O 2º DV vem primeiro: pesos (k + 1) até 2
  let s2 = 0;
  for (let i = 0; i < k; i++) s2 += d[i] * (k + 1 - i);
  const dv2 = calc(s2);
  // O 1º DV usa os mesmos dígitos mais o 2º DV: pesos (k + 2) até 2
  let s1 = dv2 * 2;
  for (let i = 0; i < k; i++) s1 += d[i] * (k + 2 - i);
  return calc(s1) === d[k] && dv2 === d[k + 1];
}

import re


def validar_ie_ba(ie: str) -> bool:
    """IE da Bahia: confere tamanho, prefixo e dígito verificador.

    Não diz se a inscrição está ativa: isso só a consulta ao cadastro responde.
    """
    n = re.sub(r"[\s./-]", "", ie)
    if not re.fullmatch(r"\d{8,9}", n) or len(set(n)) == 1:
        return False
    d = [int(c) for c in n]
    k = len(n) - 2  # dígitos antes dos DVs (6 ou 7)
    chave = d[0] if len(n) == 8 else d[1]  # decide o módulo
    mod10 = chave in (0, 1, 2, 3, 4, 5, 8)

    def calc(s):
        if mod10:
            return (10 - s % 10) % 10
        return 0 if s % 11 < 2 else 11 - s % 11

    # O 2º DV vem primeiro: pesos (k + 1) até 2
    dv2 = calc(sum(d[i] * (k + 1 - i) for i in range(k)))
    # O 1º DV usa os mesmos dígitos mais o 2º DV: pesos (k + 2) até 2
    dv1 = calc(sum(d[i] * (k + 2 - i) for i in range(k)) + dv2 * 2)
    return dv1 == d[k] and dv2 == d[k + 1]