Merge pull request 'add garbage class' (#8) from garbage into master

Reviewed-on: #8
Reviewed-by: Mikołaj Gawor <mikgaw@st.amu.edu.pl>
This commit is contained in:
Paweł Felcyn 2023-03-25 17:50:08 +01:00
commit 94e65feaa9
1 changed files with 21 additions and 0 deletions

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