garbageTruck.py

@@ -0,0 +1,33 @@
+from typing import List
+
+class garbageTruck:
+    position: List[int, int]
+    paper: []
+    plastic_and_metal: []
+    glass: []
+    bio: []
+    mixed: []
+    truckSpritePath = 'imgs/dust_car.png'
+    def __init__(self, position: List[int, int]) -> None:
+        self.position = position
+        self.paper = []
+        self.plastic_and_metal = []
+        self.glass = []
+        self.bio = []
+        self.mixed = []
+
+    def sort_garbage(self, RecognizedGarbage) -> None:
+        if RecognizedGarbage.garbage_type == 0:
+            self.paper.append(RecognizedGarbage)
+
+        elif RecognizedGarbage.garbage_type == 1:
+            self.plastic_and_metal.append(RecognizedGarbage)
+
+        elif RecognizedGarbage.garbage_type == 3:
+            self.glass.append(RecognizedGarbage)
+
+        elif RecognizedGarbage.garbage_type == 4:
+            self.bio.append(RecognizedGarbage)
+
+        elif RecognizedGarbage.garbage_type == 5:
+            self.mixed.append(RecognizedGarbage)