'''
Zaimplementuj klasę PracownikPlus, która będzie klasą pochodną Pracownik.
Klasa ta powinna umożliwiać dodatkowo przyznawanie premii pracownikowi.
>>> pp = PracownikPlus('Jan', 'Nowak', 100.0)
>>> pp.praca(4)
>>> pp.praca(6)
>>> pp.premia(700.0)
>>> pp.wyplata()
1700.0
'''

from pracownik2 import Pracownik

# W tej wersji zakładamy, że klasa Pracownik została napisana w wersji 2,
# gdzie obiekt Pracownik posiada atrybut kasa.
# Jeśli tak jest, to możemy kwotę premii dodawać od razu do tego atrybutu,
# a nie musimy zmieniać treści żadnej metody.
# Uwaga! Takie podejście jest ryzykowne, bo jeśli klasa, z której dziedziczymy, zmieni swoje szczegóły implementacji,
# to klasa dziedzicząca przestanie działać.

class PracownikPlus(Pracownik):
    def premia(self, kwota_premii):
        self.kasa += kwota_premii
