Close fileobj in case of error. (Closes: #7)

Signed-off-by: Chris Lamb <chris@chris-lamb.co.uk>
lamby · Jul 25, 2018 · 16f1069 · 16f1069
1 parent 6777cc7
commit 16f1069
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/fadvise/__init__.py b/fadvise/__init__.py
@@ -105,6 +105,10 @@ def _open_wrapper(advice, *args, **kwargs):
 
     fileobj = open(*args, **kwargs)
 
-    set_advice(fileobj, advice, offset=offset, len=len)
+    try:
+        set_advice(fileobj, advice, offset=offset, len=len)
+    except Exception:
+        fileobj.close()
+        raise
 
     return fileobj