put exception gaurd on file opens

vitiral · Mar 20, 2020 · 4ef2b99 · 4ef2b99
1 parent 26bcdc9
commit 4ef2b99
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/gpio.py b/gpio.py
@@ -63,9 +63,16 @@ def wrapped(pin, *args, **kwargs):
                 with _export_lock:
                     with open(pjoin(gpio_root, 'export'), 'w') as f:
                         _write(f, pin)
-            value = open(pjoin(ppath, 'value'), FMODE)
-            direction = open(pjoin(ppath, 'direction'), FMODE)
-            active_low = open(pjoin(ppath, 'active_low'), FMODE)
+            value, direction, active_low = None, None, None
+            try:
+                value = open(pjoin(ppath, 'value'), FMODE)
+                direction = open(pjoin(ppath, 'direction'), FMODE)
+                active_low = open(pjoin(ppath, 'active_low'), FMODE)
+            except Exception as e:
+                if value: value.close()
+                if direction: direction.close()
+                if active_low: active_low.close()
+                throw e
             _open[pin] = PinState(value=value, direction=direction, active_low=active_low)
         return function(pin, *args, **kwargs)
     return wrapped