# OBS! Det ble litt endringer fra gruppetimen

class Brusautomat:
    def __init__(self, antall):
        # parameter antall brukes til ? sette antall brus i automaten, og antall brus som fylles p? hver gang automaten skal fylles
        self._ant_brus = antall
        self._p?fyll = antall
    
    def kj?p(self, antall):
        # sjekker at vi har nok brus i automaten
        if self._ant_brus - antall >= 0:
            self._ant_brus -= antall
            # returnerer antall brus kunden har kj?pt
            return antall
        else:
            kj?p = self._ant_brus
            self._ant_brus = 0
            # returnerer antall brus kunden har kj?pt (som er mindre enn det kunden ville ha)
            return kj?p

    def fyll_p?(self):
        # fyller p? et fast antall brus. p?fyll bestemmes av konstrukt?ren
        self._ant_brus += self._p?fyll

    def sjekk_antall(self):
        return self._ant_brus