Post job action database setup.

lib/galaxy/model/migrate/versions/0046_post_job_actions.py

@@ -0,0 +1,49 @@
+"""
+Migration script to create tables for handling post-job actions.
+"""
+
+from sqlalchemy import *
+from sqlalchemy.orm import *
+from migrate import *
+from migrate.changeset import *
+
+import logging
+logging.basicConfig( level=logging.DEBUG )
+log = logging.getLogger( __name__ )
+
+# Need our custom types, but don't import anything else from model
+from galaxy.model.custom_types import *
+
+import datetime
+now = datetime.datetime.utcnow
+
+metadata = MetaData( migrate_engine )
+db_session = scoped_session( sessionmaker( bind=migrate_engine, autoflush=False, autocommit=True ) )
+
+PostJobAction_table = Table("post_job_action", metadata,
+    Column("id", Integer, primary_key=True),
+    Column("workflow_step_id", Integer, ForeignKey( "workflow_step.id" ), index=True, nullable=False),
+    Column("action_type", String(255), nullable=False),
+    Column("output_name", String(255), nullable=True),
+    Column("action_arguments", JSONType, nullable=True))
+
+# PostJobActionAssociation_table = Table("post_job_action_association", metadata,
+#     Column("id", Integer, primary_key=True),
+#     Column("post_job_action_id", Integer, ForeignKey("post_job_action.id"), index=True, nullable=False),
+#     Column("job_id", Integer, ForeignKey("job.id"), index=True, nullable=False))
+
+tables = [PostJobAction_table]#, PostJobActionAssociation_table]
+
+def upgrade():
+    print __doc__
+    metadata.reflect()
+    for table in tables:
+        try:
+            table.create()
+        except:
+            log.warn( "Failed to create table '%s', ignoring (might result in wrong schema)" % table.name )
+
+def downgrade():
+    metadata.reflect()
+    for table in tables:
+        table.drop()