utils: Cancel the conversion tasks if the are older then 24 hours

Signed-off-by: Sayan Chowdhury <[email protected]>

Author: sayanchowdhury

fedimg/consumers.py

@@ -37,6 +37,7 @@
 
 from fedimg.config import PROCESS_COUNT, STATUS_FILTER
 from fedimg.utils import get_rawxz_urls, get_value_from_dict
+from fedimg.utils import cancel_stale_conversion_tasks
 
 _log = logging.getLogger(__name__)
 
@@ -80,6 +81,8 @@ def consume(self, msg):
             _log.debug('%s is not valid status' % msg_info['status'])
             return
 
+        cancel_stale_conversion_tasks()
+
         location = msg_info['location']
         compose_id = msg_info['compose_id']
         try:

fedimg/utils.py

@@ -26,6 +26,7 @@
 import logging
 _log = logging.getLogger(__name__)
 
+import datetime
 import functools
 import os
 import re
@@ -199,3 +200,39 @@ def get_image_name_from_ami_name_for_fedmsg(image_name):
     image_name = name_vt_region.rsplit('-', 4)[:-4][0]
 
     return image_name
+
+
+def cancel_stale_conversion_tasks():
+    _log.info("Preparing to cancel conversion tasks")
+    output, error, retcode = external_run_command([
+        'euca-describe-conversion-tasks --region us-east-1 | grep active',
+    ])
+
+    if retcode != 0:
+        return False
+
+    tasks = output.split('\n')[1:-1]
+
+    for task in tasks:
+        expiration_datetime = datetime.datetime.strptime(
+            task.split()[5],
+            '%Y-%m-%dT%H:%M:%Sz'
+        )
+        import_vol_task_id = task.split()[3]
+
+        start_datetime = expiration_datetime - datetime.timedelta(days=7)
+        now_datetime = datetime.datetime.now()
+
+        delta_time = now_datetime - start_datetime
+
+        if delta_time.total_seconds() >= 86400:
+            _log.info('This is a stale task. Canceling %r'% import_vol_task_id) 
+            output, error, retcode = external_run_command([
+                'euca-cancel-conversion-task --region us-east-1',
+                import_vol_task_id
+            ])
+
+            if retcode != 0:
+                _log.info('Could not cancel the task %r' % import_vol_task_id)
+
+    return True