Merge pull request 'GarbageTruck' (#10) from s464933-patch-1 into master

Reviewed-on: #10
Reviewed-by: Paweł Felcyn <pawfel1@st.amu.edu.pl>
This commit is contained in:
Paweł Felcyn 2023-03-26 21:13:49 +02:00
commit 8501e7e7b4
1 changed files with 34 additions and 0 deletions

34
garbageTruck.py Normal file
View File

@ -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)