python类get_context()的实例源码

def upgrade():
    '''
    upgrade method
    '''
    if context.get_context().dialect.name not in ('sqlite', 'mysql'):
        create_check_constraint(name='BAD_REPLICAS_STATE_CHK', source='bad_replicas',
                                condition="state in ('B', 'D', 'L', 'R', 'S')")

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name not in ('sqlite', 'mysql'):
        drop_constraint('BAD_REPLICAS_STATE_CHK', 'bad_replicas', type_='check')

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name != 'sqlite':
        drop_column('requests', 'activity')

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name != 'sqlite':

        drop_constraint('heartbeats_pk', 'configs', type_='primary')
        drop_index('heartbeats_updated_at', 'heartbeats')

        if context.get_context().dialect.name != 'mysql':
            drop_constraint('heartbeats_created_nn', 'heartbeats', type_='check')
            drop_constraint('heartbeats_updated_nn', 'heartbeats', type_='check')

    drop_table('heartbeats')

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name != 'sqlite':
        drop_constraint('tokens_account_fk', 'tokens', type_='foreignkey')
        drop_index('TOKENS_ACCOUNT_EXPIRED_AT_IDX', 'tokens')
        create_foreign_key('tokens_account_fk', 'tokens', 'accounts', ['account'], ['account'])

def upgrade():
    '''
    upgrade method
    '''
    if context.get_context().dialect.name not in ('sqlite', 'mysql'):
        drop_constraint('REQUESTS_STATE_CHK', 'requests', type_='check')
        create_check_constraint(name='REQUESTS_STATE_CHK', source='requests', condition="state in ('Q', 'G', 'S', 'D', 'F', 'L', 'N', 'O', 'A', 'U', 'W', 'M')")

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name not in ('sqlite', 'mysql'):
        drop_constraint('REQUESTS_STATE_CHK', 'requests', type_='check')
        create_check_constraint(name='REQUESTS_STATE_CHK', source='requests', condition="state in ('Q', 'G', 'S', 'D', 'F', 'L', 'N', 'O', 'A', 'U', 'W')")

def upgrade():
    '''
    upgrade method
    '''
    if context.get_context().dialect.name not in ('sqlite'):
        add_column('requests', sa.Column('transferred_at', sa.DateTime()))

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name not in ('sqlite'):
        drop_column('requests', 'transferred_at')

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name != 'sqlite':
        drop_column('dataset_locks', 'length')
        drop_column('dataset_locks', 'bytes')
        drop_column('dataset_locks', 'accessed_at')
        drop_column('dids', 'accessed_at')

def upgrade():
    '''
    upgrade method
    '''
    if context.get_context().dialect.name not in ['sqlite']:
        add_column('rules', sa.Column('ignore_availability', sa.Boolean(name='RULES_IGNORE_AVAILABILITY_CHK'), default=False))

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name not in ['sqlite']:
        drop_column('rules', 'ignore_availability')

def upgrade():
    '''
    upgrade method
    '''
    if context.get_context().dialect.name != 'sqlite':
        add_column('dids', sa.Column('transient', sa.Boolean(name='DID_TRANSIENT_CHK'), server_default='0'))

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name != 'sqlite':
        drop_column('dids', 'transient')

def upgrade():
    '''
    upgrade method
    '''
    if context.get_context().dialect.name != 'sqlite':
        add_column('requests', sa.Column('started_at', sa.DateTime))
        add_column('requests_history', sa.Column('started_at', sa.DateTime))

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name != 'sqlite':
        drop_column('requests', 'started_at')
        drop_column('requests_history', 'started_at')

def downgrade():
    '''
    downgrade method
    '''
    drop_table('archive_contents')
    drop_table('archive_contents_history')
    if context.get_context().dialect.name != 'sqlite':
        drop_column('dids', 'is_archive')
        drop_column('dids', 'constituent')
        drop_column('deleted_dids', 'is_archive')
        drop_column('deleted_dids', 'constituent')

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name != 'sqlite':
        drop_column('dids', 'provenance')
        drop_column('dids', 'phys_group')

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name == 'postgresql':
        drop_constraint('COLLECTION_REPLICAS_PK', 'collection_replicas', type_='primary')
        drop_constraint('COLLECTION_REPLICAS_TYPE_CHK', 'collection_replicas')
        drop_constraint('COLLECTION_REPLICAS_STATE_CHK', 'collection_replicas')
        drop_constraint('COLLECTION_REPLICAS_LFN_FK', 'collection_replicas')
        drop_constraint('COLLECTION_REPLICAS_RSE_ID_FK', 'collection_replicas')
        drop_constraint('COLLECTION_REPLICAS_SIZE_NN', 'collection_replicas')
        drop_constraint('COLLECTION_REPLICAS_STATE_NN', 'collection_replicas')
        drop_index('COLLECTION_REPLICAS_RSE_ID_IDX', 'collection_replicas')
    drop_table('collection_replicas')

def downgrade():
    '''
    downgrade method
    '''
    if context.get_context().dialect.name != 'sqlite':
        drop_column('distances', 'packet_loss')
        drop_column('distances', 'latency')
        drop_column('distances', 'mbps_file')
        drop_column('distances', 'mbps_link')
        drop_column('distances', 'queued_total')
        drop_column('distances', 'done_1h')
        drop_column('distances', 'done_6h')