Compare commits

...

2 Commits

Author SHA1 Message Date
94e65feaa9 Merge pull request 'add garbage class' (#8) from garbage into master
Reviewed-on: #8
Reviewed-by: Mikołaj Gawor <mikgaw@st.amu.edu.pl>
2023-03-25 17:50:08 +01:00
Pawel Felcyn
9b3902a163 add garbage class 2023-03-25 10:53:19 +01:00

21
garbage.py Normal file
View File

@ -0,0 +1,21 @@
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