16 lines
486 B
Python
16 lines
486 B
Python
|
from garbage import RecognizedGarbage
|
||
|
|
from typing import List, Tuple
|
||
|
|
|
||
|
|
class GarbageCan:
|
||
|
|
position: Tuple[float, float]
|
||
|
|
garbage: List[RecognizedGarbage]
|
||
|
|
|
||
|
|
def __init__(self, position: Tuple[float, float]) -> None:
|
||
|
|
self.position = position
|
||
|
|
self.garbage = []
|
||
|
|
|
||
|
|
def add_garbage(self, garbage: RecognizedGarbage) -> None:
|
||
|
|
self.garbage.append(garbage)
|
||
|
|
|
||
|
|
def remove_garbage(self, garbage: RecognizedGarbage) -> None:
|
||
|
|
self.garbage.remove(garbage)
|