diff options
Diffstat (limited to 'apply_migrations.py')
| -rw-r--r-- | apply_migrations.py | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/apply_migrations.py b/apply_migrations.py index 4ade81e..4b00dfe 100644 --- a/apply_migrations.py +++ b/apply_migrations.py @@ -9,6 +9,10 @@ migrations_dir = 'migrations' # Relative directory path # Get all migration files migration_files = sorted(os.listdir(migrations_dir)) +db.ensure_migration_table_exists() + +num_applied = 0 + # Apply migrations in alphabetical order for migration_file in migration_files: migration_path = os.path.join(migrations_dir, migration_file) @@ -18,8 +22,12 @@ for migration_file in migration_files: migration_sql = f.read() print(migration_sql) + number = int(migration_file.split('_')[0]) - db.apply_migration(migration_sql) - print(f"* Migration applied: {migration_path}") + if db.ensure_migration_applied(number, migration_sql): + print(f"* Migration applied: {migration_path}\n") + num_applied += 1 + else: + print(f"* Migration already applied: {migration_path}\n") -print(f"* {len(migration_files)} migrations applied.") +print(f"* {num_applied} migrations applied.") |