from sqlalchemy import Table, MetaData, Column, Float

meta = MetaData()


def _get_table(meta):
    meta.reflect()
    AnalyzeTask = Table("analyze_tasks", meta, autoload=True, autoload_with=meta.bind)
    return AnalyzeTask


def _get_column():
    return Column("clone_duration", Float, nullable=False, default=0, server_default="0")


def upgrade(migrate_engine):
    meta.bind = migrate_engine
    table = _get_table(meta)
    col = _get_column()
    col.create(table, populate_default=True)


def downgrade(migrate_engine):
    meta.bind = migrate_engine
    table = _get_table(meta)
    table.c.clone_duration.drop()