Restore imaging capability definitions and templates

schema/versions/74c70c6e1f08_add_restore_imaging_capability.py

@@ -44,7 +44,7 @@ set -o errexit
 ./framework.sh -d .
 chmod 770 .
 cd rawdata/
-../productfetcher --product-locator $1 $2
+../productfetcher --product-locator $1 --product-locator $2
 cd ../
 ./casa_envoy --integrated -i $4 $5
 ./deliver -p ./products --prefix $3
@@ -190,6 +190,22 @@ fix_restore_metadata = """{
 
 """
 
+fix_restore_script = """#!/bin/sh
+
+export HOME=$TMPDIR
+
+set -o errexit
+./framework.sh -d .
+chmod 770 .
+cd rawdata/
+../productfetcher --product-locator $1 --product-locator $2
+cd ../
+./casa_envoy --restore -c $4 $5
+cp ./working/{{sdmId}}.ms ./products/
+./deliver -p ./products --prefix $3
+
+"""
+
 
 def upgrade():
     restore_image_steps = """prepare-and-run-workflow std_restore_imaging
@@ -240,6 +256,12 @@ await-qa
         SET content = E'{fix_restore_metadata}' WHERE filename='metadata.json' AND workflow_name='restore_cms'
         """
     )
+    op.execute(
+        f"""
+        UPDATE workflow_templates
+        SET content = E'{fix_restore_script}' WHERE filename='restore_cms.sh'
+        """
+    )
 
 
 def downgrade():
@@ -261,3 +283,52 @@ def downgrade():
         DELETE FROM workflow_templates WHERE workflow_name = 'std_restore_imaging'
         """
     )
+
+    op.execute(
+        f"""
+        UPDATE workflow_templates
+        SET content = E'{old_restore_metadata}' WHERE filename='metadata.json' AND workflow_name='restore_cms'
+        """
+    )
+    op.execute(
+        f"""
+        UPDATE workflow_templates
+        SET content = E'{old_restore_script}' WHERE filename='restore_cms.sh'
+        """
+    )
+
+
+old_restore_script = """#!/bin/sh
+
+export HOME=$TMPDIR
+
+set -o errexit
+./framework.sh -d .
+chmod 770 .
+cd rawdata/
+../productfetcher --product-locator $1 $2
+cd ../
+./casa_envoy --restore -c $4 $5
+cp ./working/{{sdmId}}.ms ./products/
+./deliver -p ./products --prefix $3
+
+"""
+
+
+old_restore_metadata = """{
+  "fileSetIds": ["{{sdmId}}", "{{calSdmId}}"]
+  "workflowName": "std_calibration",
+  "systemId": "{{request_id}}",
+  "creationTime": "{{created_at}}",
+  "productLocator": "{{product_locator}}",
+  "calProductLocator": "{{cal_locator}}",
+  "projectMetadata": {
+    "projectCode": "{{projectCode}}",
+    "title": "{{title}}",
+    "startTime": "{{startTime}}",
+    "observer": "{{observer}}"
+  },
+  "destinationDirectory": "{{root_directory}}/{{relative_path}}"
+}
+    
+"""