root/galaxy-central/lib/galaxy/model/migrate/versions/0051_imported_col_for_jobs_table.py @ 2

"""
Migration script to add imported column for jobs table.
"""

from sqlalchemy import *
from sqlalchemy.orm import *
from migrate import *
from migrate.changeset import *

import logging
log = logging.getLogger( __name__ )

metadata = MetaData( migrate_engine )
db_session = scoped_session( sessionmaker( bind=migrate_engine, autoflush=False, autocommit=True ) )

def upgrade():
        print __doc__
        metadata.reflect()

        # Create and initialize imported column in job table.
        Jobs_table = Table( "job", metadata, autoload=True )
        c = Column( "imported", Boolean, default=False, index=True )
        try:
            # Create
            c.create( Jobs_table )
            assert c is Jobs_table.c.imported
            
            # Initialize.
            if migrate_engine.name == 'mysql' or migrate_engine.name == 'sqlite': 
                default_false = "0"
            elif migrate_engine.name == 'postgres':
                default_false = "false"
            db_session.execute( "UPDATE job SET imported=%s" % default_false )
            
        except Exception, e:
            print "Adding imported column to job table failed: %s" % str( e )
            log.debug( "Adding imported column to job table failed: %s" % str( e ) )
    
def downgrade():
        metadata.reflect()

        # Drop imported column from job table.
        Jobs_table = Table( "job", metadata, autoload=True )
        try:
            Jobs_table.c.imported.drop()
        except Exception, e:
            print "Dropping column imported from job table failed: %s" % str( e )
            log.debug( "Dropping column imported from job table failed: %s" % str( e ) )