from person import Person

class Leilighet:
    def __init__(self, adresse, etasje, leilighet_nr):
        self._adresse = adresse
        self._etasje = etasje
        self._leilighet_nr = leilighet_nr

        self._leietakere = []

    def hent_adresse(self):
        adr = self._adresse +' H'
        if self._etasje < 10:
            adr += '0'
        adr += str(self._etasje)
        if self._leilighet_nr < 10:
            adr += '0'
        adr += str(self._leilighet_nr)

        return adr

    def hent_leietakere(self):
        return self._leietakere

    def flytt_inn(self, person):
        if type(person) == Person:
            self._leietakere.append(person)
            person.flytt_inn(self.hent_adresse())

        self._leietakere = list(set(self._leietakere))

    def flytt_ut(self, person):
        if person in self._leietakere:
            person.flytt_inn(None)
            self._leietakere.remove(person)

    def __str__(self):
        return self.hent_adresse() +'('+ str(len(self._leietakere)) + ')'