klasy2 - sklep sprawdza pełnoletniość

dzien7/klasy_wprowadzenie/klasy2.py

@@ -26,29 +26,34 @@ class Sklep:
         self.cennik[produkt] = cena
 
     def sprzedaj(self, produkt, klient, sztuk=1):
-        koszt = self.cennik[produkt] * sztuk
-        print(f'Kliencie {klient.imie}, za swoje zakupy płacisz {koszt}')
+        if produkt not in self.cennik:
+            print(f'Nieznany produkt: {produkt}')
+        elif produkt in {'piwo'} and not klient.jest_pelnoletnia():
+            print(f'Osobom niepełnoletnim nie sprzedajemy {produkt}')
+        else:
+            koszt = self.cennik[produkt] * sztuk
+            print(f'Kliencie {klient.imie}, za swoje zakupy płacisz {koszt}')
 
 
 # W tym przykładzie używamy klasy od razu w tym samym pliku,
 # ale w praktyce częściej w jednym pliku definiuje się klasę, a w innym importuje i korzysta.
 
-a = Osoba('Ala', 'Kowalska', 30)
-b = Osoba(imie='Bartek', nazwisko='Malinowski', wiek=15)
-print(a)
-print(b)
+ala = Osoba('Ala', 'Kowalska', 30)
+bartek = Osoba(imie='Bartek', nazwisko='Malinowski', wiek=15)
+print(ala)
+print(bartek)
 
 print('\nOsoby się przedstawiają:')
-a.przedstaw_sie()
-if a.jest_pelnoletnia():
-    print(f'{a.imie} jest osobą pełnoletnią')
+ala.przedstaw_sie()
+if ala.jest_pelnoletnia():
+    print(f'{ala.imie} jest osobą pełnoletnią')
 else:
-    print(f'{a.imie} jest osobą niepełnoletnią')
-b.przedstaw_sie()
-if b.jest_pelnoletnia():
-    print(f'{b.imie} jest osobą pełnoletnią')
+    print(f'{ala.imie} jest osobą niepełnoletnią')
+bartek.przedstaw_sie()
+if bartek.jest_pelnoletnia():
+    print(f'{bartek.imie} jest osobą pełnoletnią')
 else:
-    print(f'{b.imie} jest osobą niepełnoletnią')
+    print(f'{bartek.imie} jest osobą niepełnoletnią')
 print()
 
 zabka = Sklep(nazwa='Żabka')
@@ -56,4 +61,7 @@ zabka.zdefiniuj_produkt('cola', 6)
 zabka.zdefiniuj_produkt('piwo', 7.50)
 print('Produkty dostępne w Żabce:', zabka.cennik)
 
-zabka.sprzedaj('piwo', a)
+zabka.sprzedaj('piwo', ala)
+zabka.sprzedaj('ciasteczka', ala)
+zabka.sprzedaj('piwo', bartek)
+zabka.sprzedaj('cola', bartek, 3)
\ No newline at end of file