From 9b3902a163ff48f9df07a6aa8f4c5f49c0f58cd7 Mon Sep 17 00:00:00 2001
From: Pawel Felcyn <pawel.felcyn@onet.pl>
Date: Sat, 25 Mar 2023 10:53:19 +0100
Subject: [PATCH] add garbage class

---
 garbage.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 garbage.py

diff --git a/garbage.py b/garbage.py
new file mode 100644
index 0000000..5769b4a
--- /dev/null
+++ b/garbage.py
@@ -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
\ No newline at end of file