Relocatable Null

schema/versions/257537f99abc_make_null_relocatable_for_testing.py

+"""make null relocatable for testing
+
+Revision ID: 257537f99abc
+Revises: 42a9e35d0809
+Create Date: 2021-09-14 09:52:41.666489
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = "257537f99abc"
+down_revision = "42a9e35d0809"
+branch_labels = None
+depends_on = None
+
+condor_content = """executable = null.sh
+arguments = {{arguments}}
+
+output = null.out
+error = null.err
+log = condor.log
+should_transfer_files = yes
+transfer_input_files = /lustre/aoc/cluster/pipeline/$ENV(CAPO_PROFILE)/workspaces/sbin/null
+
+#run on a condor node in CV-NAASC
+{{#remote-cv}}
+requirements = (CVPOST == True)
++partition = "CVPOST"
+{{/remote-cv}}
+
+#run on a NMT condor node. (naming consistent with vlass-seci)
+{{#remote}}
+requirements = (HasLustre =!= True)
+{{/remote}}
+
+queue
+
+"""
+
+
+old_content = """executable = null.sh
+arguments = {{arguments}}
+
+output = null.out
+error = null.err
+log = null.log
+should_transfer_files = yes
+transfer_input_files = /lustre/aoc/cluster/pipeline/$ENV(CAPO_PROFILE)/workspaces/sbin/null
+
+
+queue
+
+"""
+
+
+def upgrade():
+    op.execute(
+        f"""
+        UPDATE workflow_templates
+        SET content=E'{condor_content}' WHERE filename='null.condor'
+        """
+    )
+
+
+def downgrade():
+    op.execute(
+        f"""
+        UPDATE workflow_templates
+        SET content=E'{old_content}' WHERE filename='null.condor'
+        """
+    )