Moved locking behaviour into common superclass. Moved context-managed behaviour into...

[Adafruit_Blinka-hackapet.git] / python / digitalio / __init__.py

diff --git a/python/digitalio/__init__.py b/python/digitalio/__init__.py
index e90513dce15f58572a1a9f1dd07d866b37be4425..db8a1a4637beddecbaa00ea543197ab9985ece3e 100644 (file)
--- a/python/digitalio/__init__.py
+++ b/python/digitalio/__init__.py
@@ -1,6 +1,6 @@
 from machine import Pin
 from agnostic import board as boardId
-from mcp import Enum
+from mcp import Enum,ContextManaged
 
 
 class DriveMode(Enum):
@@ -26,7 +26,7 @@ Pull.DOWN = Pull()
 #Pull.NONE = Pull()
 
 
-class DigitalInOut(object):
+class DigitalInOut(ContextManaged):
     _pin = None
 
     def __init__(self, pin):
@@ -45,12 +45,6 @@ class DigitalInOut(object):
     def deinit(self):
         del self._pin
 
-    def __enter__(self):
-        return self
-
-    def __exit__(self, exc_type, exc_value, traceback):
-        self.deinit()
-
     @property
     def direction(self):
         return self.__direction