rahix/openflexure-microscope-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixed TestCaptureMethods

Browse source
This commit is contained in:
Joel Collins 2019-01-08 13:23:36 +00:00
parent af76442149
commit 3c34b43e32
2 changed files with 14 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
openflexure_microscope/camera/capture.py
View file

@ -40,14 +40,16 @@ class StreamObject(object):
# Create file name. Default to UUID
if not filename:
filename = self.id
self.build_file_path(filename, folder, self.format)
self.filename = filename
self.folder = folder
self.build_file_path(self.filename, self.folder, self.format)
# Byte stream properties
self.stream = io.BytesIO() # Byte stream that data will be written to
# Set default write target
self.write_to_file = write_to_file
if self.write_to_file is False:
if not self.write_to_file:
logging.debug("Target for {} set to 'stream'".format(self.id))
self.target = self.stream
else:
@ -69,6 +71,11 @@ class StreamObject(object):
Stored files will be cleaned up automatically.".format(self.id))
self.keep_on_disk = False # Flag file to be removed on close.
self.context_manager = True # Used in metadata
# Re-generate file name for temp file
logging.info("Re-generating file name for temp file")
self.build_file_path(self.filename, self.folder, self.format)
return self
def __exit__(self, *args):