Fixed broken context manager __exit__

openflexure_microscope/camera/base.py

@@ -226,6 +226,7 @@ class StreamObject(object):
 
     def close(self):
         """Both clear the stream, and delete any associated on-disk data"""
+        log("Closing {}".format(self.id))
         self.delete_stream()
         if not self.keep_on_disk:
             self.delete_file()

openflexure_microscope/camera/pi.py

@@ -100,20 +100,12 @@ class StreamingCamera(BaseCamera):
     # HANDLE CONTEXT MANAGER AND FILE CLOSING
 
     def close(self):
+        """Close method called by BaseCamera __exit__"""
         # Close all StreamObjects
-        #TODO: Figure out why this is so slow.
         for capture_list in [self.images, self.videos]:
             for stream_object in capture_list:
                 stream_object.close()
-
+        self.stop_worker()
-    def __enter__(self):
-        log("Entering context for {}.\
-            Stored files will be cleaned up automatically.".format(self))
-        return self
-
-    def __exit__(self, *args):
-        log("Cleaning up {}".format(self))
-        self.close()
 
     # RETURNING CAPTURES