integration with alembic 0.6+

integration with alembic 0.6+