root/galaxy-central/lib/galaxy/model/migrate/versions/0012_user_address.py

"""
This script adds a new user_address table that is currently only used with sample requests, where
a user can select from a list of his addresses to associate with the request.  This script also
drops the request.submitted column which was boolean and replaces it with a request.state column
which is a string, allowing for more flexibility with request states.
"""
from sqlalchemy import *
from sqlalchemy.orm import *
from sqlalchemy.exc import *
from migrate import *
from migrate.changeset import *
import datetime
now = datetime.datetime.utcnow
import sys, logging
# Need our custom types, but don't import anything else from model
from galaxy.model.custom_types import *

log = logging.getLogger( __name__ )
log.setLevel(logging.DEBUG)
handler = logging.StreamHandler( sys.stdout )
format = "%(name)s %(levelname)s %(asctime)s %(message)s"
formatter = logging.Formatter( format )
handler.setFormatter( formatter )
log.addHandler( handler )

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

def display_migration_details():
    print "========================================"
    print "This script adds a new user_address table that is currently only used with sample requests, where"
    print "a user can select from a list of his addresses to associate with the request.  This script also"
    print "drops the request.submitted column which was boolean and replaces it with a request.state column"
    print "which is a string, allowing for more flexibility with request states."
    print "========================================"

UserAddress_table = Table( "user_address", metadata,
    Column( "id", Integer, primary_key=True),
    Column( "create_time", DateTime, default=now ),
    Column( "update_time", DateTime, default=now, onupdate=now ),
    Column( "user_id", Integer, ForeignKey( "galaxy_user.id" ), index=True ),
    Column( "desc", TEXT),
    Column( "name", TrimmedString( 255 ), nullable=False),
    Column( "institution", TrimmedString( 255 )),
    Column( "address", TrimmedString( 255 ), nullable=False),
    Column( "city", TrimmedString( 255 ), nullable=False),
    Column( "state", TrimmedString( 255 ), nullable=False),
    Column( "postal_code", TrimmedString( 255 ), nullable=False),
    Column( "country", TrimmedString( 255 ), nullable=False),
    Column( "phone", TrimmedString( 255 )),
    Column( "deleted", Boolean, index=True, default=False ),
    Column( "purged", Boolean, index=True, default=False ) )

def upgrade():
    display_migration_details()
    # Load existing tables
    metadata.reflect()
    # Add all of the new tables above
    try:
        UserAddress_table.create()
    except Exception, e:
        log.debug( "Creating user_address table failed: %s" % str( e ) ) 
    # Add 1 column to the request_type table
    try:
        RequestType_table = Table( "request_type", metadata, autoload=True )
    except NoSuchTableError:
        RequestType_table = None
        log.debug( "Failed loading table request_type" )
    if RequestType_table:
        try:
            col = Column( "deleted", Boolean, index=True, default=False )
            col.create( RequestType_table )
            assert col is RequestType_table.c.deleted
        except Exception, e:
            log.debug( "Adding column 'deleted' to request_type table failed: %s" % ( str( e ) ) )
    # Delete the submitted column
    try:
        Request_table = Table( "request", metadata, autoload=True )
    except NoSuchTableError:
        Request_table = None
        log.debug( "Failed loading table request" )
    if Request_table:
        try:
            Request_table.c.submitted.drop()
        except Exception, e:
            log.debug( "Deleting column 'submitted' to request table failed: %s" % ( str( e ) ) )   
        try:
            col = Column( "state", TrimmedString( 255 ), index=True  )
            col.create( Request_table )
            assert col is Request_table.c.state
        except Exception, e:
            log.debug( "Adding column 'state' to request table failed: %s" % ( str( e ) ) )

def downgrade():
    pass