Dynamic Manual Execution Time

Author: kcagran

src/backend/apps/clusters/parser.py

@@ -1,5 +1,7 @@
+import decimal
 import logging
 
+from django.conf import settings
 from django.db import transaction
 
 from backend.apps.clusters.models import (
@@ -60,12 +62,26 @@ def organization(self) -> Organization | None:
     @property
     def job_template(self) -> JobTemplate:
         external_job_template = self.data.summary_fields.job_template
-        return JobTemplate.create_or_update(
+        job_template = JobTemplate.create_or_update(
             cluster=self.cluster,
             external_id=external_job_template.id if external_job_template else -1,
             name=external_job_template.name if external_job_template else self.data.name,
             description=external_job_template.description if external_job_template else self.data.description,
         )
+        job_count = Job.objects.filter(job_template=job_template).count()
+        # If no jobs exist for this template, it's safe to calculate the manual execution time
+        if job_count == 0:
+            logger.info("No jobs exist for this template. Calculating manual execution time.")
+            elapsed = self.data.elapsed
+            if elapsed is not None:
+                manual_execution_time = int(decimal.Decimal(elapsed / 60 * 2).quantize(decimal.Decimal(1), rounding=decimal.ROUND_UP))
+                if manual_execution_time < settings.DEFAULT_TIME_TAKEN_TO_MANUALLY_EXECUTE_MINUTES:
+                    manual_execution_time = settings.DEFAULT_TIME_TAKEN_TO_MANUALLY_EXECUTE_MINUTES
+                if manual_execution_time > 1000000:
+                    manual_execution_time = 1000000
+                job_template.time_taken_manually_execute_minutes = manual_execution_time
+                job_template.save(update_fields=['time_taken_manually_execute_minutes'])
+        return job_template
 
     @property
     def launched_by(self) -> AAPUser | None:

src/backend/tests/conftest.py

@@ -532,6 +532,20 @@ def cluster_sync_data(cluster, api_jobs, api_host_summaries):
     )
 
 
+@pytest.fixture
+def cluster_sync_data_elapsed(cluster, api_jobs, api_host_summaries):
+    data = api_jobs[1]
+    data["elapsed"] = 8965.58
+    data["host_summaries"] = [
+        api_host_summaries[1],
+        api_host_summaries[2],
+    ]
+
+    return ClusterSyncData.objects.create(
+        cluster=cluster,
+        data=data,
+    )
+
 @pytest.fixture
 def superuser():
     return User.objects.create(

src/backend/tests/unit/test_parser.py

@@ -253,6 +253,12 @@ def test_update_job_template(self, cluster, cluster_sync_data, job_templates, ex
         for key, value in expected.items():
             assert getattr(job_template, key) == value
 
+    def test_job_template_calculate_manual_execution_time(self, cluster, cluster_sync_data_elapsed):
+        parser = DataParser(cluster_sync_data_elapsed.id)
+        job_template = parser.job_template
+        assert JobTemplate.objects.count() == 1
+        assert job_template.time_taken_manually_execute_minutes == 299
+
     @pytest.mark.parametrize('expected', [
         aap_user_expected_data,
     ])