garbageTruck.py

@@ -0,0 +1,34 @@
+from typing import List
+
+class GarbageTruck:
+    position: Tuple[int, int]
+    paper: List[RecognizedGarbage]
+    plastic_and_metal: List[RecognizedGarbage]
+    glass: List[RecognizedGarbage]
+    bio: List[RecognizedGarbage]
+    mixed: List[RecognizedGarbage]
+    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)