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