rfid_test now indicates the everything

indication_driver.py

@@ -54,3 +54,4 @@ class IndicationDriver:
                         self.pixels[idx] = (0, 0, 0)
                 self.pixels.show()
                 await asyncio.sleep(.1)
+        self.clear()

rfid_driver.py

@@ -0,0 +1,17 @@
+import RPi._GPIO as GPIO
+from mfrc522 import SimpleMFRC522
+import asyncio
+
+class RFIDDriver:
+    def __init__(self):
+        self.reader = SimpleMFRC522()
+
+    def __del__(self):
+        GPIO.cleanup()
+    
+    async def get_next_id(self) -> int:
+        id = self.reader.read_id_no_block()
+        while not id:
+            id = self.reader.read_id_no_block()
+            await asyncio.sleep(0.01)
+        return id

test_scripts/rfid_test.py

@@ -1,23 +1,41 @@
 from time import sleep
 import RPi.GPIO as GPIO
 from mfrc522 import SimpleMFRC522
+import asyncio
 
 import sys
 sys.path.append('..')
 import TREx.config as config
 from TREx.indication_driver import IndicationDriver
+from TREx.rfid_driver import RFIDDriver
 
-reader = SimpleMFRC522()
 
-indication_driver = IndicationDriver()
+async def __main__():
-
+    rfid_driver = RFIDDriver()
-while True:
+    indication_driver = IndicationDriver()
-    id, _ = reader.read()
+    indication_task = None
-    if id in config.RFID_LOOKUP_TABLE.keys():
+    while True:
+        id = await rfid_driver.get_next_id()
+        trash_category = None
+        try:
+            if indication_task is not None:
+                indication_driver.stop_loop()
+                await indication_task
             trash_category = config.RFID_LOOKUP_TABLE[id]
-        print(f"{id}: {trash_category}") 
+            indication_task = asyncio.create_task(indication_driver.indicate_async(trash_category))
-        indication_driver.indicate(trash_category)
+            await asyncio.sleep(1)
-    else:
+        except KeyError:
-        print(id)
+            indication_driver.stop_loop()
-    sleep(1)
+
-GPIO.cleanup()
+asyncio.run(__main__())
+
+# while True:
+#     id, _ = reader.read()
+#     if id in config.RFID_LOOKUP_TABLE.keys():
+#         trash_category = config.RFID_LOOKUP_TABLE[id]
+#         print(f"{id}: {trash_category}") 
+#         indication_driver.indicate(trash_category)
+#     else:
+#         print(id)
+#     sleep(1)
+# GPIO.cleanup()