Skip to content

Commit

Permalink
Add missing migration
Browse files Browse the repository at this point in the history
  • Loading branch information
mvdbeek committed May 23, 2020
1 parent b03e552 commit 749f791
Showing 1 changed file with 39 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
"""
Migration script to add a new job_to_input_dataset_collection_element table to track job inputs.
"""
from __future__ import print_function

import logging

from sqlalchemy import Column, ForeignKey, Integer, MetaData, Table, Unicode

log = logging.getLogger(__name__)
metadata = MetaData()

job_to_input_dataset_collection_element_table = Table(
"job_to_input_dataset_collection_element", metadata,
Column("id", Integer, primary_key=True),
Column("job_id", Integer, ForeignKey("job.id"), index=True),
Column("dataset_collection_element_id", Integer, ForeignKey("dataset_collection_element.id"), index=True),
Column("name", Unicode(255)))


def upgrade(migrate_engine):
print(__doc__)
metadata.bind = migrate_engine
metadata.reflect()

try:
job_to_input_dataset_collection_element_table.create()
except Exception:
log.exception("Creating job_to_input_dataset_collection_element table failed")


def downgrade(migrate_engine):
metadata.bind = migrate_engine
metadata.reflect()

try:
job_to_input_dataset_collection_element_table.drop()
except Exception:
log.exception("Dropping job_to_input_dataset_collection_element table failed")

0 comments on commit 749f791

Please sign in to comment.