save feed requested flag to DB

FNNDSC · Mar 8, 2024 · 98c847b · 98c847b
1 parent 0c2738c
commit 98c847b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/app/config.py b/app/config.py
@@ -3,7 +3,7 @@
 
 class Settings(BaseSettings):
     pflink_mongodb: MongoDsn = 'mongodb://localhost:27017'
-    version: str = "3.9.0"
+    version: str = "3.9.1"
     mongo_username: str = "admin"
     mongo_password: str = "admin"
     log_level: str = "DEBUG"

diff --git a/app/controllers/subprocesses/status.py b/app/controllers/subprocesses/status.py
@@ -62,6 +62,7 @@ def update_workflow_status(key: str, test: bool):
     workflow.response = update_workflow_progress(updated_status)
     pretty_response = pprint.pformat(workflow.response.__dict__)
     logger.debug(f"Updated response: {pretty_response}.", extra=d)
+    logger.info(f"Current status is {workflow.response.workflow_state}.", extra=d)
     update_status_flag(key, workflow, True, test)
     logger.info(f"Finished writing updated status to the DB, releasing lock.", extra=d)
 
@@ -341,7 +342,6 @@ def _parse_response(
             status.status = False
             status.error = Error.feed_deleted.value
             status.state_progress = "0%"
-    logger.info(f"Current status is {status.workflow_state}.", extra=d)
     return status
 
 

diff --git a/app/controllers/subprocesses/utils.py b/app/controllers/subprocesses/utils.py
@@ -48,6 +48,7 @@ def workflow_retrieve_helper(workflow: dict) -> WorkflowDBSchema:
         service_retry=workflow["service_retry"],
         stale=workflow["stale"],
         started=workflow["started"],
+        feed_requested=True if not workflow.get("feed_requested") else workflow["feed_requested"],
     )
 
 
@@ -68,6 +69,7 @@ def workflow_add_helper(workflow: WorkflowDBSchema) -> dict:
         "service_retry": workflow.service_retry,
         "stale": workflow.stale,
         "started": workflow.started,
+        "feed_requested": workflow.feed_requested,
     }