Akcja.py

@@ -8,32 +8,32 @@ class Akcja:
     def __init__(self, typ):
         self.typ = typ
         if self.typ == "nawodnienie":
-            self.srodki.append(Srodek(1, "woda", "woda"))
-            self.srodki.append(Srodek(1.5, "powerade", "woda")) #nawadnia lepiej niż woda
-            self.benefit["nawodnienie"] = 100
+            self.srodki.append(Srodek.Srodek(1, "woda", "woda"))
+            self.srodki.append(Srodek.Srodek(1.5, "powerade", "woda")) #nawadnia lepiej niż woda
+            self.benefits["nawodnienie"] = 100
         if self.typ == "zyznosc":
-            self.srodki.append(Srodek(2, "obornik", "nawoz"))
-            self.srodki.append(Srodek(3, "azotan", "nawoz"))
-            self.srodki.append(Srodek(4, "wapno", "nawoz"))
-            self.srodki.append(Srodek(5, "superfosfat", "nawoz"))
+            self.srodki.append(Srodek.Srodek(2, "obornik", "nawoz"))
+            self.srodki.append(Srodek.Srodek(3, "azotan", "nawoz"))
+            self.srodki.append(Srodek.Srodek(4, "wapno", "nawoz"))
+            self.srodki.append(Srodek.Srodek(5, "superfosfat", "nawoz"))
             self.benefits["zyznosc"] = 100
         if self.typ == "wzrost":
-            self.srodki.append(Srodek(6, "witaminy", "odzywka"))
-            self.srodki.append(Srodek(7, "aminokwasy", "odzywka"))
-            self.srodki.append(Srodek(8, "algi morskie", "odzywka"))
+            self.srodki.append(Srodek.Srodek(6, "witaminy", "odzywka"))
+            self.srodki.append(Srodek.Srodek(7, "aminokwasy", "odzywka"))
+            self.srodki.append(Srodek.Srodek(8, "algi morskie", "odzywka"))
             self.benefits["wzrost"] = 20
         if self.typ == "grzyb":
-            self.srodki.append(Srodek(9, "mankozeb", "ochrona"))
-            self.srodki.append(Srodek(10, "czosnek", "ochrona")) #tak czosnek zabija grzyby
-            self.benefit["grzyb"] = "brak"
+            self.srodki.append(Srodek.Srodek(9, "mankozeb", "ochrona"))
+            self.srodki.append(Srodek.Srodek(10, "czosnek", "ochrona")) #tak czosnek zabija grzyby
+            self.benefits["grzyb"] = "brak"
         if self.typ == "bakterie":
-            self.srodki.append(Srodek(11, "miedź", "ochrona"))
-            self.srodki.append(Srodek(12, "streptomycyna ", "ochrona"))
-            self.benefit["grzyb"] = "brak"
+            self.srodki.append(Srodek.Srodek(11, "miedź", "ochrona"))
+            self.srodki.append(Srodek.Srodek(12, "streptomycyna ", "ochrona"))
+            self.benefits["grzyb"] = "brak"
         if self.typ == "pasożyt":
-            self.srodki.append(Srodek(13, "Cyjantraniliprol", "ochrona"))
-            self.srodki.append(Srodek(14, "Permetryna", "ochrona"))
-            self.srodki.append(Srodek(15, "Abamektyna", "ochrona"))
+            self.srodki.append(Srodek.Srodek(13, "Cyjantraniliprol", "ochrona"))
+            self.srodki.append(Srodek.Srodek(14, "Permetryna", "ochrona"))
+            self.srodki.append(Srodek.Srodek(15, "Abamektyna", "ochrona"))
             self.benefits["grzyb"] = "brak"
     def getBenefit(self):
         return self.benefits

Osprzet.py

@@ -1,14 +1,19 @@
+import Akcja
 class Osprzet:
-    akcja = None
-    def __init__(self, id, marka, model):
+    def __init__(self, id, marka, model, akcje = None):
         self.id = id
         self.marka = marka
         self.model = model
+        if akcje is None:
+            self.akcje = []
+        else:
+            self.akcje = akcje
 
 
 plug = Osprzet(1,'Bomet', 'U031')
 siewnik = Osprzet(2, "Amazone", "12001-C")
-rozsiewacz = Osprzet(3, 'John Deere', 'TF 1500')
-opryskiwacz = Osprzet(4, 'John Deere', 'M720')
+rozsiewacz = Osprzet(3, 'John Deere', 'TF 1500', [Akcja.Akcja("zyznosc")])
+opryskiwacz = Osprzet(4, 'John Deere', 'M720', [Akcja.Akcja("grzyb"), Akcja.Akcja("bakterie"), Akcja.Akcja("nawodnienie"), Akcja.Akcja("wzrost")])
 header = Osprzet(5, 'John Deere', 'X350R')
-#jak istnieją jakieś bardziej profesjonalne nazwy czy lepsze to śmiało zmieńcie
+#jak istnieją jakieś bardziej profesjonalne nazwy czy lepsze to śmiało zmieńcie
+

Tractor.py

@@ -42,7 +42,12 @@ class Tractor:
         self.osprzet = new_osprzet
 
     def print_osprzet_info(self):
-        if self.osprzet is not None:
-            print("Wyposazenie:", self.osprzet.marka, self.osprzet.model)
+        print("ID:", self.osprzet.id)
+        print("Marka:", self.osprzet.marka)
+        print("Model:", self.osprzet.model)
+        if self.osprzet.akcje:
+            print("Akcje:")
+            for akcja in self.osprzet.akcje:
+                print("- Typ:", akcja.typ)
         else:
-            print("Brak")
+            print("Brak akcji przypisanych do tego sprzętu.")