handle other possible states, add return

bcgsc · Jul 9, 2024 · ef0ca7b · ef0ca7b
1 parent 3a3e594
commit ef0ca7b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pori_python/ipr/connection.py b/pori_python/ipr/connection.py
@@ -84,12 +84,13 @@ def check_status(interval: int = 5, num_attempts: int = 5):
                     logger.info(f'checking report loading status in {interval} seconds')
                     time.sleep(interval)
                     current_status = self.get(f'reports-async/{report_id}')
-                    if current_status['state'] not in ['active', 'ready']:
+                    if current_status['state'] not in ['active', 'ready', 'waiting', 'completed']:
                         raise Exception(
                             f'async report upload in unexpected state: {current_status}'
                         )
-                    if current_status['state'] == 'ready':
+                    if current_status['state'] in ['ready', 'completed']:
                         return current_status
+                return current_status
 
             current_status = check_status()
             if current_status['state'] == 'active':