sztuczna_inteligencja_2023_smieciarka/garbage.py

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