diff --git a/schema/versions/d3abd475fbb4_correcting_restore_condor.py b/schema/versions/d3abd475fbb4_correcting_restore_condor.py new file mode 100644 index 0000000000000000000000000000000000000000..2b047f9063d733b02a8404ae15f806fef8c478af --- /dev/null +++ b/schema/versions/d3abd475fbb4_correcting_restore_condor.py @@ -0,0 +1,77 @@ +"""correnting restore condor + +Revision ID: d3abd475fbb4 +Revises: 6508afd4da68 +Create Date: 2021-07-16 16:51:30.483952 + +""" +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision = "d3abd475fbb4" +down_revision = "6508afd4da68" +branch_labels = None +depends_on = None + + +condor_content = """executable = std_restore_imaging.sh +arguments = {{product_locator}} {{cal_locator}} {{request_id}} metadata.json PPR.xml + +output = restore.out +error = restore.err +log = condor.log + +SBIN_PATH = /lustre/aoc/cluster/pipeline/$ENV(CAPO_PROFILE)/workspaces/sbin +should_transfer_files = yes +transfer_input_files = $(SBIN_PATH)/.matplotlib, $(SBIN_PATH)/pycapo, $(SBIN_PATH)/framework.sh, $(SBIN_PATH)/productfetcher, $(SBIN_PATH)/casa_envoy, $(SBIN_PATH)/vela, $(SBIN_PATH)/deliver, ./PPR.xml, ./metadata.json +transfer_output_files = working, rawdata, products + +request_memory = {{ramInGb}} +getenv = True +environment = "CAPO_PATH=/home/casa/capo" + +queue + + +""" + +old_content = """executable = std_restore_imaging.sh +arguments = {{product_locator}} {{cal_product_locator}} {{request_id}} metadata.json PPR.xml + +output = restore.out +error = restore.err +log = condor.log + +SBIN_PATH = /lustre/aoc/cluster/pipeline/$ENV(CAPO_PROFILE)/workspaces/sbin +should_transfer_files = yes +transfer_input_files = $(SBIN_PATH)/.matplotlib, $(SBIN_PATH)/pycapo, $(SBIN_PATH)/framework.sh, $(SBIN_PATH)/productfetcher, $(SBIN_PATH)/casa_envoy, $(SBIN_PATH)/vela, $(SBIN_PATH)/deliver, ./PPR.xml, ./metadata.json +transfer_output_files = working, rawdata, products + +request_memory = {{ramInGb}} +getenv = True +environment = "CAPO_PATH=/home/casa/capo" + +queue + + +""" + + +def upgrade(): + op.execute( + f""" + UPDATE workflow_templates + SET content = E'{condor_content}' where filename = 'restore_cms.condor' + """ + ) + + +def downgrade(): + op.execute( + f""" + UPDATE workflow_templates + SET content = E'{old_content}' where filename = 'restore_cms.condor' + """ + )