fix cleanup fcn

[hackapet/Adafruit_Blinka.git] / src / digitalio.py

diff --git a/src/digitalio.py b/src/digitalio.py
index ab9b761881f59c9e39eb8722d312ac9cb2bee02e..c017a0a6341746b5cda0e67fce47085414aff1b9 100755 (executable)
--- a/src/digitalio.py
+++ b/src/digitalio.py
@@ -17,6 +17,10 @@ elif detector.chip.AM33XX:
     from adafruit_blinka.microcontroller.am335x.pin import Pin
 elif detector.chip.SUN8I:
     from adafruit_blinka.microcontroller.allwinner_h3.pin import Pin
+elif detector.chip.SAMA5:
+    from adafruit_blinka.microcontroller.sama5.pin import Pin
+elif detector.chip.TEGRAXXX:
+    from adafruit_blinka.microcontroller.tegra.pin import Pin
 elif detector.chip.STM32:
     from machine import Pin
 from adafruit_blinka import Enum, ContextManaged
@@ -50,7 +54,6 @@ Pull.DOWN = Pull()
 
 #Pull.NONE = Pull()
 
-
 class DigitalInOut(ContextManaged):
     _pin = None
 
@@ -70,6 +73,12 @@ class DigitalInOut(ContextManaged):
     def deinit(self):
         del self._pin
 
+    def cleanup(self, channel=None):
+        if channel == self:
+            Pin.cleanup(self._pin, self._pin)
+        else:
+            Pin.cleanup(self, channel)
+
     @property
     def direction(self):
         return self.__direction