41 lines
941 B
Python
41 lines
941 B
Python
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):
|
|
garbage_type: GarbageType
|
|
|
|
def __init__(self, src: Garbage, garbage_type: GarbageType) -> None:
|
|
super().__init__(src.img)
|
|
self.garbage_type = garbage_type
|