21 lines
439 B
Python
21 lines
439 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
|
||
|
|
||
|
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
|