Python MigrationHome.flattenの例

コード例 #1

ファイル: flatten.py プロジェクト: dusktreader/ambix

def flatten(
    alembic_home_dir: pathlib.Path = typer.Option(
        "etc/alembic/versions",
        exists=True,
        help="The directory where the alembic migrations live",
    ),
):
    home = MigrationHome(alembic_home_dir)
    home.flatten()

コード例 #2

ファイル: test_migration_home.py プロジェクト: dusktreader/ambix

    def test_flatten(self, scripts_dir):
        home = MigrationHome(scripts_dir)
        home.flatten()

        assert MigrationHome(scripts_dir).generate_dependency_graph() == {
            "aaaaaa": {None},
            "bbbbbb": {"aaaaaa"},
            "cccccc": {"bbbbbb"},
            "dddddd": {"cccccc"},
            "eeeeee": {"dddddd"},
            "ffffff": {"eeeeee"},
            "gggggg": {"ffffff"},
        }