From 9b3edd5bc5e59ff0478d128ad76ea02ca90669b9 Mon Sep 17 00:00:00 2001
From: willgraf <7930703+willgraf@users.noreply.github.com>
Date: Fri, 2 Apr 2021 16:12:48 -0700
Subject: [PATCH] Create new variable MAX_IMAGE_FRAMES and the
 tracking-consumer crashes if input has more frames. (#166)

---
 redis_consumer/consumers/tracking_consumer.py | 5 +++++
 redis_consumer/settings.py                    | 1 +
 2 files changed, 6 insertions(+)

diff --git a/redis_consumer/consumers/tracking_consumer.py b/redis_consumer/consumers/tracking_consumer.py
index a5126f9f..e83dd6fb 100644
--- 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
index 321cd755..b5163f80 100644
--- 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')