Esercizio 9codice cliente che si incrementa

import io
import re

class TourOperator:
    def __init__(self, codice_iniziale):
        if not re.fullmatch(r"[A-Z][0-9]{3}", codice_iniziale):   # formato Lnnn
            raise ValueError("codice non valido: " + codice_iniziale)
        self._prossimo = codice_iniziale            # None quando i codici sono finiti
        self._clienti = {}                          # codice -> (nome, destinazione)

    def add(self, nome, dest):
        if self._prossimo is None:
            raise OverflowError("codici esauriti")
        self._clienti[self._prossimo] = (nome, dest)
        self._incrementa()

    def _incrementa(self):
        lettera, numero = self._prossimo[0], int(self._prossimo[1:])
        if numero < 999:                            # basta incrementare il numero
            self._prossimo = f"{lettera}{numero + 1:03d}"
        elif lettera < "Z":                         # nnn -> 000 e lettera successiva
            self._prossimo = f"{chr(ord(lettera) + 1)}000"
        else:                                       # Z999 era l'ultimo codice
            self._prossimo = None

    def find(self, codice):
        return self._clienti[codice]                # KeyError se assente

    def __str__(self):
        return "\n".join(f"{c} : {n} : {d}" for c, (n, d) in self._clienti.items())


def main(codice, f):
    t = TourOperator(codice)
    for riga in f:
        riga = riga.strip()
        if riga:
            nome, dest = riga.split(":")
            t.add(nome, dest)
    print(t)


viaggi = "Ironman:Malibu\nSpider man:New York\nThor:Asgard\nBatman:Gotham\nSuperman:Metropolis\n"
main("A998", io.StringIO(viaggi))        # con la tastiera: main(sys.argv[1], sys.stdin)

A998 : Ironman : Malibu
A999 : Spider man : New York
B000 : Thor : Asgard
B001 : Batman : Gotham
B002 : Superman : Metropolis

for iniziale, atteso in [("C998", "C999"), ("C999", "D000"), ("A000", "A001"), ("Y999", "Z000")]:
    t = TourOperator(iniziale)
    t.add("x", "y")
    assert t._prossimo == atteso, (iniziale, t._prossimo)

t = TourOperator("Z999")
t.add("a", "b")
print(t._prossimo, t.find("Z999"))               # None ('a', 'b')
try:
    t.add("c", "d")
except OverflowError as e:
    print("OverflowError:", e)                   # OverflowError: codici esauriti

for cattivo in ("a123", "A12", "AB12", "12AB", "A1x3"):
    try:
        TourOperator(cattivo)
    except ValueError:
        pass
    else:
        print("accettato?", cattivo)
print("ok")