Diff two migration folders

diif_migrate.py

from unipath import Path
import re

MIGRATION_NAME_PATTERN = r"^\d+_(.*).rb$"


def get_migration_names(path):
    return set(migration_name(migration_filename)
               for migration_filename in
               Path(path).listdir(names_only=True))


def migration_name(filename):
    return re.match(MIGRATION_NAME_PATTERN, filename).group(1)


def compare_migrations():
    old_migrations = get_migration_names('./db/migrated')
    new_migrations = get_migration_names('./db/migrate')

    new_migrations_not_in_old = new_migrations.difference(old_migrations)

    print('\n'.join(new_migrations_not_in_old))

if __name__ == '__main__':
    compare_migrations()