diff --git a/classes/rachunek.py b/classes/rachunek.py new file mode 100644 index 0000000..ff115e8 --- /dev/null +++ b/classes/rachunek.py @@ -0,0 +1,16 @@ +import pygame +import prefs +class Rachunek: + def __init__(self,numer_zamowienia): + self.numer_zamowienia = numer_zamowienia + self.zamowienia = [] + + def dodaj_zamowienie(self,zamowienie): + if zamowienie.numer_zamowienia == self.numer_zamowienia: + self.zamowienia.append(zamowienie) + + def suma(self): + suma_rachunek = 0 + for zamowienie in self.zamowienia: + suma_rachunek += zamowienie.posilek.cena + return suma_rachunek diff --git a/classes/zamowenie.py b/classes/zamowenie.py new file mode 100644 index 0000000..a292975 --- /dev/null +++ b/classes/zamowenie.py @@ -0,0 +1,8 @@ +import pygame +import prefs +class Zamowienie: + def __init__(self,numer_zamowienia,stolik,kelner,posilk): + self.numer_zamowienia = numer_zamowienia + self.stolik = stolik + self.kelner = kelner + self.posilk = posilk \ No newline at end of file