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