def test_auto(): from peewee_migrate.auto import diff_one, diff_many, model_to_code from peewee_migrate.cli import get_router router = get_router(path.join(CURDIR, 'migrations'), 'sqlite:///:memory:') router.run() migrator = router.migrator models = migrator.orm.values() Person_ = migrator.orm['person'] Tag_ = migrator.orm['tag'] code = model_to_code(Person_) assert code assert "default=dt.datetime.now" in code assert 'db_table = "person"' in code assert "order_by = ('-dob',)" in code changes = diff_many(models, [], migrator=migrator) assert len(changes) == 2 class Person(pw.Model): first_name = pw.IntegerField() last_name = pw.CharField(max_length=1024, null=True, unique=True) tag = pw.ForeignKeyField(Tag_, on_delete='CASCADE', related_name='persons') changes = diff_one(Person, Person_, migrator=migrator) assert len(changes) == 6 assert "on_delete='CASCADE'" in changes[0] assert "related_name='persons'" in changes[0] assert changes[-3] == "migrator.drop_not_null('person', 'last_name')" assert changes[-2] == "migrator.drop_index('person', 'last_name')" assert changes[ -1] == "migrator.add_index('person', 'last_name', unique=True)"
def test_auto(): from peewee_migrate.auto import diff_one, diff_many, model_to_code from peewee_migrate.cli import get_router router = get_router('tests/migrations', 'sqlite:///:memory:') router.run() migrator = router.migrator models = migrator.orm.values() Person_ = migrator.orm['person'] Tag_ = migrator.orm['tag'] code = model_to_code(Person_) assert code import ipdb; ipdb.set_trace() # XXX BREAKPOINT assert 'db_table = "person"' in code changes = diff_many(models, [], migrator=migrator) assert len(changes) == 2 class Person(pw.Model): first_name = pw.IntegerField() last_name = pw.CharField(max_length=1024, null=True) tag = pw.ForeignKeyField(Tag_, on_delete='CASCADE', related_name='persons') changes = diff_one(Person, Person_, migrator=migrator) assert len(changes) == 4 assert "on_delete='CASCADE'" in changes[0] assert "related_name='persons'" in changes[0] assert changes[-1] == "migrator.drop_not_null('person', 'last_name')"
def test_auto(): from peewee_migrate.auto import diff_one, diff_many, model_to_code from peewee_migrate.cli import get_router router = get_router(path.join(CURDIR, 'migrations'), 'sqlite:///:memory:') router.run() migrator = router.migrator models = migrator.orm.values() Person_ = migrator.orm['person'] Tag_ = migrator.orm['tag'] code = model_to_code(Person_) assert code assert 'table_name = "person"' in code changes = diff_many(models, [], migrator=migrator) assert len(changes) == 2 class Person(pw.Model): first_name = pw.IntegerField() last_name = pw.CharField(max_length=1024, null=True, unique=True) tag = pw.ForeignKeyField(Tag_, on_delete='CASCADE', related_name='persons') email = pw.CharField(index=True, unique=True) changes = diff_one(Person, Person_, migrator=migrator) assert len(changes) == 6 assert "on_delete='CASCADE'" in changes[0] assert "backref='persons'" in changes[0] assert changes[-3] == "migrator.drop_not_null('person', 'last_name')" assert changes[-2] == "migrator.drop_index('person', 'last_name')" assert changes[ -1] == "migrator.add_index('person', 'last_name', unique=True)" migrator.drop_index('person', 'email') migrator.add_index('person', 'email', unique=True) class Person(pw.Model): first_name = pw.CharField(unique=True) last_name = pw.CharField(max_length=255, index=True) dob = pw.DateField(null=True) birthday = pw.DateField(default=dt.datetime.now) email = pw.CharField(index=True, unique=True) changes = diff_one(Person_, Person, migrator=migrator) assert not changes class Color(pw.Model): id = pw.AutoField() name = pw.CharField(default='red') code = model_to_code(Color) assert "DEFAULT 'red'" in code
def compile_migrations(migrator, models, reverse=False): """Compile migrations for given models.""" source = migrator.orm.values() if reverse: source, models = models, source migrations = diff_many(models, source, migrator, reverse=reverse) if not migrations: return False migrations = NEWLINE + NEWLINE.join('\n\n'.join(migrations).split('\n')) return CLEAN_RE.sub('\n', migrations)
def test_auto(): from peewee_migrate.auto import diff_one, diff_many, model_to_code from peewee_migrate.cli import get_router router = get_router(path.join(CURDIR, 'migrations'), 'sqlite:///:memory:') router.run() migrator = router.migrator models = migrator.orm.values() Person_ = migrator.orm['person'] Tag_ = migrator.orm['tag'] code = model_to_code(Person_) assert code assert 'table_name = "person"' in code changes = diff_many(models, [], migrator=migrator) assert len(changes) == 2 class Person(pw.Model): first_name = pw.IntegerField() last_name = pw.CharField(max_length=1024, null=True, unique=True) tag = pw.ForeignKeyField(Tag_, on_delete='CASCADE', backref='persons') email = pw.CharField(index=True, unique=True) changes = diff_one(Person, Person_, migrator=migrator) assert len(changes) == 6 assert "on_delete='CASCADE'" in changes[0] assert "backref='persons'" in changes[0] assert changes[-3] == "migrator.drop_not_null('person', 'last_name')" assert changes[-2] == "migrator.drop_index('person', 'last_name')" assert changes[-1] == "migrator.add_index('person', 'last_name', unique=True)" migrator.drop_index('person', 'email') migrator.add_index('person', 'email', unique=True) class Person(pw.Model): first_name = pw.CharField(unique=True) last_name = pw.CharField(max_length=255, index=True) dob = pw.DateField(null=True) birthday = pw.DateField(default=dt.datetime.now) email = pw.CharField(index=True, unique=True) changes = diff_one(Person_, Person, migrator=migrator) assert not changes class Color(pw.Model): id = pw.AutoField() name = pw.CharField(default='red') code = model_to_code(Color) assert "DEFAULT 'red'" in code
def create(self, name='auto', auto=False): """Create a migration.""" migrate = rollback = '' if auto: if isinstance(auto, str): try: auto = import_module(auto) except ImportError: return self.logger.error('Invalid models module: %s', auto) if isinstance(auto, ModuleType): auto = list( filter(lambda m: isinstance(m, type) and issubclass( m, pw.Model)), (getattr(auto, model) for model in dir(auto))) # noqa for migration in self.diff: self.run_one(migration, self.migrator) models1 = auto models2 = list(self.migrator.orm.values()) migrate = diff_many(models1, models2) if not migrate: return self.logger.warn('No changes has found.') migrate = NEWLINE + NEWLINE.join('\n\n'.join(migrate).split('\n')) migrate = CLEAN_RE.sub('\n', migrate) rollback = diff_many(models2, models1) rollback = NEWLINE + NEWLINE.join( '\n\n'.join(rollback).split('\n')) rollback = CLEAN_RE.sub('\n', rollback) self.logger.info('Create a migration "%s"', name) path = self._create(name, migrate, rollback) self.logger.info('Migration has created %s', path) return path
def create(self, name='auto', auto=False): """Create a migration.""" migrate = rollback = '' if auto: if isinstance(auto, str): try: auto = import_module(auto) except ImportError: return self.logger.error('Invalid models module: %s', auto) if isinstance(auto, ModuleType): auto = list(filter( lambda m: isinstance(m, type) and issubclass(m, pw.Model)), (getattr(auto, model) for model in dir(auto))) # noqa for migration in self.diff: self.run_one(migration, self.migrator) models1 = auto models2 = list(self.migrator.orm.values()) migrate = diff_many(models1, models2) if not migrate: return self.logger.warn('No changes has found.') migrate = NEWLINE + NEWLINE.join('\n\n'.join(migrate).split('\n')) migrate = CLEAN_RE.sub('\n', migrate) rollback = diff_many(models2, models1) rollback = NEWLINE + NEWLINE.join('\n\n'.join(rollback).split('\n')) rollback = CLEAN_RE.sub('\n', rollback) self.logger.info('Create a migration "%s"', name) path = self._create(name, migrate, rollback) self.logger.info('Migration has created %s', path) return path
def test_auto(): from peewee_migrate.auto import diff_one, diff_many, model_to_code from peewee_migrate.cli import get_router router = get_router('tests/migrations', 'sqlite:///:memory:') router.run() migrator = router.migrator models = migrator.orm.values() Person_, _ = models code = model_to_code(Person_) assert code changes = diff_many(models, []) assert len(changes) == 2 class Person(pw.Model): first_name = pw.IntegerField() last_name = pw.CharField(max_length=1024) changes = diff_one(Person_, Person) assert len(changes) == 2
def run_db_migration(database): db_storage_dir = os.path.dirname(pavlov_central.storage.__file__) print(f'db_storage_dir={db_storage_dir}') db_migrations_dir = os.path.join(db_storage_dir, 'migrations') print(f'db_migrations_dir={db_migrations_dir}') router = Router(database, migrate_dir=db_migrations_dir, migrate_table='migration', logger=logging.getLogger(), ignore=[ 'basemodel', 'basedatamodel', 'frs5model', 'virtualmodel', 'baseftsmodel', 'ftsmodel', 'lsmtable' ]) print('Run migrations..') router.run() # check migrations src_models = load_models('pavlov_central.storage.models') print(f'src_models={src_models}') if router.ignore: src_models = [ m for m in src_models if m.get_table_name() not in router.ignore ] db_models = router.migrator.orm.values() print(f'db_models={db_models}') diff_found = diff_many(src_models, db_models, router.migrator, reverse=False) if len(diff_found) > 0: logging.warning('migrations diff_found={}'.format(diff_found)) print('migrations diff_found={}'.format(diff_found)) raise RuntimeError('Check db migrations is failed')