Create new variable MAX_IMAGE_FRAMES and the tracking-consumer crashe…

…s if input has more frames. (#166)
vanvalenlab · Apr 2, 2021 · 9b3edd5 · 9b3edd5
1 parent d63d31b
commit 9b3edd5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/redis_consumer/consumers/tracking_consumer.py b/redis_consumer/consumers/tracking_consumer.py
@@ -103,6 +103,11 @@ def _load_data(self, redis_hash, subdir, fname):
         remaining_hashes = set()
         frames = {}
 
+        if num_frames > settings.MAX_IMAGE_FRAMES:
+            raise ValueError('This tiff file has {} frames but the maximum '
+                             'number of allowed frames is {}.'.format(
+                                 num_frames, settings.MAX_IMAGE_FRAMES))
+
         self.logger.debug('Got tiffstack shape %s.', tiff_stack.shape)
 
         uid = uuid.uuid4().hex

diff --git a/redis_consumer/settings.py b/redis_consumer/settings.py
@@ -45,6 +45,7 @@
 MAX_RETRY = config('MAX_RETRY', default=5, cast=int)
 MAX_IMAGE_HEIGHT = config('MAX_IMAGE_HEIGHT', default=2048, cast=int)
 MAX_IMAGE_WIDTH = config('MAX_IMAGE_WIDTH', default=2048, cast=int)
+MAX_IMAGE_FRAMES = config('MAX_IMAGE_FRAMES', default=60, cast=int)
 
 # Redis client connection
 REDIS_HOST = config('REDIS_HOST', default='redis-master')