from enum import Enum class GarbageType(Enum): PAPER = 0, PLASTIC_AND_METAL = 1 GLASS = 3 BIO = 4 MIXED = 5 class Garbage: img: str shape: str flexibility: str does_smell: str weight: str size: str color: str softness: str does_din: str def __init__(self, img: str, shape: str, flexibility: str, does_smell: str, weight: str, size: str, color: str, softness: str, does_din: str) -> None: self.img = img self.shape = shape self.flexibility = flexibility self.does_smell = does_smell self.weight = weight self.size = size self.color = color self.softness = softness self.does_din = does_din class RecognizedGarbage: garbage_type: GarbageType src: Garbage def __init__(self, src: Garbage, garbage_type: GarbageType) -> None: self.garbage_type = garbage_type self.src = src