def _migrate_up(self, config, engine, dest, curr, with_data=False): if with_data: data = None pre_upgrade = getattr( self, "_pre_upgrade_%s" % dest, None) if pre_upgrade: data = pre_upgrade(engine) cli.do_alembic_command(config, 'upgrade', dest) if with_data: check = getattr(self, "_check_%s" % dest, None) if check and data: check(engine, data)
def _migrate_down(self, config, engine, dest, curr, with_data=False): if dest: cli.do_alembic_command(config, 'downgrade', dest) else: meta = sqlalchemy.MetaData(bind=engine) meta.drop_all() if with_data: post_downgrade = getattr( self, "_post_downgrade_%s" % curr, None) if post_downgrade: post_downgrade(engine)