def handle(self): """ Executes the command. """ confirm = self.confirm( '<question>Are you sure you want to reset all of the migrations?</question> ', False ) if not confirm: return database = self.option('database') repository = DatabaseMigrationRepository(self.resolver, 'migrations') migrator = Migrator(repository, self.resolver) self._prepare_database(migrator, database) pretend = bool(self.option('pretend')) path = self.option('path') if path is None: path = self._get_migration_path() migrator.reset(path, pretend) for note in migrator.get_notes(): self.line(note)
def handle(self): """ Executes the command. """ if not self.confirm_to_proceed( "<question>Are you sure you want to reset all of the migrations?:</question> " ): return database = self.option("database") repository = DatabaseMigrationRepository(self.resolver, "migrations") migrator = Migrator(repository, self.resolver) self._prepare_database(migrator, database) pretend = bool(self.option("pretend")) path = self.option("path") if path is None: path = self._get_migration_path() migrator.reset(path, pretend) for note in migrator.get_notes(): self.line(note)
def handle(self): sys.path.append(os.getcwd()) try: add_venv_site_packages() from wsgi import container except ImportError: self.comment( 'This command must be ran inside of the root of a Masonite project directory' ) # Get any migration files from the Service Container migration_directory = ['databases/migrations'] for key, value in container.providers.items(): if 'MigrationDirectory' in key: migration_directory.append(value) # Load in the Orator migration system from orator.migrations import Migrator, DatabaseMigrationRepository from config import database repository = DatabaseMigrationRepository(database.DB, 'migrations') migrator = Migrator(repository, database.DB) if not migrator.repository_exists(): repository.create_repository() # Create a new list of migrations with the correct file path instead migration_list = [] for migration in migrator.get_repository().get_ran(): for directory in migration_directory: if os.path.exists(os.path.join(directory, migration + '.py')): migration_list.append(os.path.join(os.getcwd(), directory)) break # Rollback the migrations notes = [] for migration in migrator.get_repository().get_ran(): for migration_directory in migration_list: try: migrator.reset(migration_directory) except QueryException as e: raise e except FileNotFoundError: pass if migrator.get_notes(): notes += migrator.get_notes() # Show notes from the migrator self.line('') for note in notes: if not ('Nothing to rollback.' in note): self.line(note) if not notes: self.info('Nothing to rollback')
def setup_database(): DATABASES = {"sqlite": {"driver": "sqlite", "database": "test.db"}} db = DatabaseManager(DATABASES) Schema(db) Model.set_connection_resolver(db) repository = DatabaseMigrationRepository(db, "migrations") migrator = Migrator(repository, db) if not repository.repository_exists(): repository.create_repository() migrator.reset("app/migrations") migrator.run("app/migrations")
def pytest_runtest_teardown(): from personalwebpageapi.models import db migrations_path = f'{os.getcwd()}/migrations' repository = DatabaseMigrationRepository( db, 'migrations', ) migrator = Migrator(repository, db) if not migrator.repository_exists(): repository.create_repository() migrator.set_connection(db.get_default_connection()) migrator.reset(migrations_path)
def __init__(self): self.db_directory = tempfile.TemporaryDirectory() self.db_file = os.path.join(self.db_directory.name, 'test.db') self.db_manager = DatabaseManager(self.database_configuration()) migrations_directory = os.path.join(os.path.dirname(__file__), "..", 'migrations') migration_repository = DatabaseMigrationRepository(self.db_manager, "migrations") migration_repository.create_repository() migrator = Migrator( migration_repository, self.db_manager, ) migrator.reset(migrations_directory) migrator.run(migrations_directory) _Model.set_connection_resolver(self.db_manager)
class Migrations(HasColoredCommands): def __init__(self, connection=None): self._ran = [] self._notes = [] from config import database database_dict = database.DB self.repository = DatabaseMigrationRepository(database.DB, 'migrations') self.migrator = Migrator(self.repository, database.DB) if not connection or connection == 'default': connection = database.DATABASES['default'] self.migrator.set_connection(connection) if not self.repository.repository_exists(): self.repository.create_repository() from wsgi import container self.migration_directories = ['databases/migrations'] for key, value in container.providers.items(): if isinstance(key, str) and 'MigrationDirectory' in key: self.migration_directories.append(value) try: add_venv_site_packages() except ImportError: self.comment( 'This command must be ran inside of the root of a Masonite project directory' ) def run(self): for directory in self.migration_directories: try: if len(self.migration_directories) > 1: self.info('Migrating: {}'.format(directory)) self.migrator.run(directory) self._ran.append(self.repository.get_ran()) self._notes = self.migrator._notes except Exception as e: self.danger(str(e)) return self def rollback(self): for directory in self.migration_directories: try: if len(self.migration_directories) > 1: self.info('Migrating: {}'.format(directory)) self.migrator.rollback(directory) self._ran.append(self.repository.get_ran()) self._notes = self.migrator._notes except Exception as e: self.danger(str(e)) return self def refresh(self): self.run() self.rollback() def reset(self): for directory in self.migration_directories: try: if len(self.migration_directories) > 1: self.info('Migrating: {}'.format(directory)) self.migrator.reset(directory) self._ran.append(self.repository.get_ran()) self._notes = self.migrator._notes except Exception as e: self.danger(str(e)) return self def ran(self): return self._ran