"""
Migration script to add the api_keys table.
"""

from sqlalchemy import *
from migrate import *
from migrate.changeset import *
from galaxy.model.custom_types import *

import datetime
now = datetime.datetime.utcnow

import logging
log = logging.getLogger( __name__ )

metadata = MetaData( migrate_engine )

APIKeys_table = Table( "api_keys", metadata,
    Column( "id", Integer, primary_key=True ),
    Column( "create_time", DateTime, default=now ),
    Column( "user_id", Integer, ForeignKey( "galaxy_user.id" ), index=True ),
    Column( "key", TrimmedString( 32 ), index=True, unique=True ) )

def upgrade():
    print __doc__
    metadata.reflect()
    try:
        APIKeys_table.create()
    except Exception, e:
        log.debug( "Creating api_keys table failed: %s" % str( e ) )

def downgrade():
    # Load existing tables
    metadata.reflect()
    try:
        APIKeys_table.drop()
    except Exception, e:
        log.debug( "Dropping api_keys table failed: %s" % str( e ) )