feat(task-cleanup): Deletion of experiment deletes all tasks related to it (#231)

* update identifier of Task from task_id to uuid

* remove Task table rows for tasks related to experiment

* moved task deletion line to 2 steps earlier

* fixed pylint issue by suppressing error

* remove backend.delete function call

* fix pylint error

Author: jatinriverlane

aqueductcore/backend/services/experiment.py

@@ -522,11 +522,16 @@ async def remove_experiment(
         experiments_root_dir=str(settings.experiments_dir_path), experiment_uuid=experiment_uuid
     )
 
+    remove_all_experiment_tasks_statement = delete(orm.Task).where(
+        orm.Task.experiment_id == experiment_uuid
+    )
+    await db_session.execute(remove_all_experiment_tasks_statement)
+
     remove_experiment_tag_links_statement = delete(orm.experiment_tag_association).where(
         orm.experiment_tag_association.c.experiment_uuid == experiment_uuid
     )
-
     await db_session.execute(remove_experiment_tag_links_statement)
+
     remove_experiment_statement = delete(orm.Experiment).where(
         orm.Experiment.uuid == experiment_uuid
     )

aqueductcore/backend/services/task_executor.py

@@ -21,8 +21,7 @@
 from aqueductcore.backend.context import UserInfo
 from aqueductcore.backend.errors import AQDDBTaskNonExisting, AQDPermission
 from aqueductcore.backend.models import orm
-from aqueductcore.backend.models.task import (TaskProcessExecutionResult,
-                                              TaskRead)
+from aqueductcore.backend.models.task import TaskProcessExecutionResult, TaskRead
 from aqueductcore.backend.services.utils import task_orm_to_model
 from aqueductcore.backend.settings import settings
 
@@ -42,8 +41,9 @@
 celery_app.conf.update(result_extended=True)
 extension_process = None  # pylint: disable=invalid-name
 
+
 def worker_signal_handler(signo, _):
-    """ Handle SIGINT signal and propagate it to child and grandchild processes. """
+    """Handle SIGINT signal and propagate it to child and grandchild processes."""
     global extension_process  # pylint: disable=global-statement,global-variable-not-assigned
     if extension_process is not None:
         psutil_child_process = psutil.Process(extension_process.pid)