digitalio: detect chip instead of board

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

diff --git a/src/digitalio.py b/src/digitalio.py
old mode 100644 (file)
new mode 100755 (executable)
index 635db1e..ab9b761
--- a/src/digitalio.py
+++ b/src/digitalio.py
@@ -1,28 +1,53 @@
-from machine import Pin
-from adafruit_blinka.agnostic import board as boardId
-from adafruit_blinka import Enum,ContextManaged
+"""
+`digitalio` - Digital input and output control
+=================================================
 
+See `CircuitPython:digitalio` in CircuitPython for more details.
+
+* Author(s): cefn
+"""
+
+from adafruit_blinka.agnostic import board_id, detector
+
+# pylint: disable=ungrouped-imports,wrong-import-position
+
+if detector.chip.BCM2XXX:
+    from adafruit_blinka.microcontroller.bcm283x.pin import Pin
+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.STM32:
+    from machine import Pin
+from adafruit_blinka import Enum, ContextManaged
 
 class DriveMode(Enum):
-    PUSH_PULL=None
-    OPEN_DRAIN=None
+    PUSH_PULL = None
+    OPEN_DRAIN = None
+
+
 DriveMode.PUSH_PULL = DriveMode()
 DriveMode.OPEN_DRAIN = DriveMode()
 
 
 class Direction(Enum):
-    INPUT=None
-    OUTPUT=None
+    INPUT = None
+    OUTPUT = None
+
+
 Direction.INPUT = Direction()
 Direction.OUTPUT = Direction()
 
 
 class Pull(Enum):
-    UP=None
-    DOWN=None
+    UP = None
+    DOWN = None
     #NONE=None
+
+
 Pull.UP = Pull()
 Pull.DOWN = Pull()
+
 #Pull.NONE = Pull()
 
 
@@ -90,7 +115,8 @@ class DigitalInOut(ContextManaged):
                 if hasattr(Pin, "PULL_DOWN"):
                     self._pin.init(mode=Pin.IN, pull=Pin.PULL_DOWN)
                 else:
-                    raise NotImplementedError("{} unsupported on {}".format(Pull.DOWN, boardId))
+                    raise NotImplementedError("{} unsupported on {}".format(
+                        Pull.DOWN, board_id))
             elif pul is None:
                 self._pin.init(mode=Pin.IN, pull=None)
             else:
@@ -112,5 +138,3 @@ class DigitalInOut(ContextManaged):
             self._pin.init(mode=Pin.OPEN_DRAIN)
         elif mod is DriveMode.PUSH_PULL:
             self._pin.init(mode=Pin.OUT)
-
-# __all__ = ['DigitalInOut', 'DriveMode', 'Direction','Pull']