Esercizio 14dizionario invertibile e squadra

import io

class Coppia:
    def __init__(self, chiave, attributo):
        self.chiave, self.attributo = chiave, attributo

    def __str__(self):
        return f"{self.chiave}:{self.attributo}"

    __repr__ = __str__

    def __eq__(self, altra):
        return (self.chiave, self.attributo) == (altra.chiave, altra.attributo)


class Archivio:
    """Dizionario invertibile: chiave -> attributo (int)."""

    def __init__(self):
        self._d = {}                                   # chiave -> attributo

    def inserisci(self, chiave, attributo):
        self._d[chiave] = attributo                    # chiave presente: sostituisce
        return True

    def cerca(self, chiave):
        return Coppia(chiave, self._d[chiave]) if chiave in self._d else None

    def cancella(self, chiave):
        return Coppia(chiave, self._d.pop(chiave)) if chiave in self._d else None

    def cerca_per_attributo(self, attributo):          # O(n): nessuna, una o più coppie
        return [Coppia(k, a) for k, a in self._d.items() if a == attributo]

    def taglia(self):
        return len(self._d)

    def __str__(self):
        return "\n".join(str(Coppia(k, a)) for k, a in self._d.items())


class Squadra(Archivio):
    MAX_GIOCATORI = 18
    BUDGET = 300                                       # milioni

    def soldi(self):
        return self.BUDGET - sum(self._d.values())

    def inserisci(self, chiave, attributo):
        vecchio = self._d.get(chiave)                  # None se il giocatore non c'è
        if vecchio is None and self.taglia() >= self.MAX_GIOCATORI:
            return False                               # squadra al completo
        if attributo - (vecchio or 0) > self.soldi():
            return False                               # budget superato
        return super().inserisci(chiave, attributo)    # vincoli rispettati: si riusa Archivio

FILE = """ROSSI A.:19
BIANCHI G.:20
VERDI S.:19
NERI M.:8
GIALLI N.:42
ROSSI A.:21
BLU F.:14
VIOLA F.:27
ARANCI M.:24
GRIGI A.:301
"""

def main(f):
    mio_archivio = Archivio()
    for riga in f:
        riga = riga.strip()
        if riga:
            nome, costo = riga.rsplit(":", 1)          # il costo è dopo l'ultimo ":"
            mio_archivio.inserisci(nome, int(costo))
    print(mio_archivio)
    mia_squadra = Squadra()
    for nome, costo in mio_archivio._d.items():
        if not mia_squadra.inserisci(nome, costo):
            print("rifiutato:", nome, costo)
    print(mia_squadra, "\nsoldi:", mia_squadra.soldi(), "taglia:", mia_squadra.taglia())


main(io.StringIO(FILE))                                # con il file vero: main(open("fantacalcio.txt"))

a = Archivio()
for k, v in [("a", 1), ("b", 2), ("c", 1)]:
    a.inserisci(k, v)
print(a.cerca_per_attributo(1), a.cerca_per_attributo(9))   # [a:1, c:1] []
print(a.cerca("b"), a.cerca("z"), a.cancella("b"), a.cancella("b"), a.taglia())   # b:2 None b:2 None 2

s = Squadra()
for i in range(25):                                         # 25 giocatori da 1 milione: vale il limite di 18
    s.inserisci(f"g{i}", 1)
print(s.taglia(), s.soldi())                                # 18 282

s = Squadra()
print(s.inserisci("x", 290), s.inserisci("y", 11), s.inserisci("y", 10), s.soldi())   # True False True 0
print(s.inserisci("x", 295), s.soldi())                     # False 0: sostituire x costerebbe 5 in più
s.inserisci("x", 280)                                       # sostituzione che risparmia 10
print(s.soldi())                                            # 10