diff --git a/garbage.py b/garbage.py new file mode 100644 index 0000000..5769b4a --- /dev/null +++ b/garbage.py @@ -0,0 +1,21 @@ +from enum import Enum + +class GarbageType (Enum): + PAPER = 0, + PLASTIC_AND_METAL = 1 + GLASS = 3 + BIO = 4 + MIXED = 5 + +class Garbage: + img: str + + def __init__(self, img: str) -> None: + self.img = img + +class RecognizedGarbage (Garbage): + garbage_type: GarbageType + + def __init__(self, src: Garbage, garbage_type: GarbageType) -> None: + super().__init__(src.img) + self.garbage_type = garbage_type \ No newline at end of file