close if error in parsing header

lens-biophotonics · Feb 15, 2017 · b616959 · b616959
1 parent 025c258
commit b616959
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/dcimg.py b/dcimg.py
@@ -60,11 +60,13 @@ def open(self, file_name=None):
 
         with open(file_name, 'r') as f:
             mm = mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_COPY)
-            if mm[:5] != b"DCIMG":
-                mm.close()
 
         self.mm = mm
-        self._parse_header()
+
+        try:
+            self._parse_header()
+        except RuntimeError:
+            self.close()
 
     @property
     def nfrms(self):