Das deutsche Python-Forum

Code: Alles auswählen

#!/usr/bin/env python3
from attr import attrib, attrs


@attrs
class Tier:
    ART = None
    name = attrib()
    einlieferungsdatum = attrib()
    id = attrib()

    @property
    def art(self):
        return self.ART


@attrs
class Tierheim:
    name = attrib()
    _tiere = attrib(factory=list)

    def __len__(self):
        return len(self._tiere)

    def __iter__(self):
        return iter(self._tiere)

    def add(self, tier):
        self._tiere.append(tier)

    def print(self):
        for tier in self:
            print(f"{tier.name} ({tier.art})")
        print(len(self))


@attrs
class Saeugetier(Tier):
    ART = "Säugetier"
    saeugerklasse = attrib()


@attrs
class Vogel(Tier):
    ART = "Vogel"
    ist_flugfaehig = attrib()
    schnabelfarbe = attrib()


@attrs
class Fisch(Tier):
    ART = "Fisch"
    lebensraum = attrib()
    fischart = attrib()


def main():
    affe = Saeugetier("Koko", "10.12.2014", 18874, "Pflanzentiere")
    fisch = Fisch("Nemo", "20.05.2010", 17896, "Süßwasser", "Raubfisch")
    vogel = Vogel("Sky", "02.12.2018", 147896, True, "blau")

    tierheim = Tierheim("AAAA")
    tierheim.add(affe)
    tierheim.add(fisch)
    tierheim.add(vogel)

    tierheim.print()
    
    print(tierheim)


if __name__ == "__main__":
    main()