Merge pull request 'refactor: Update migration function to access revision strings from RevisionStep objects for improved clarity' (#33) from ph4 into prod

Reviewed-on: #33
This commit is contained in:
mo 2025-06-01 17:50:18 +02:00
commit 8a98aee6c1

View File

@ -22,8 +22,9 @@ def _get_migration_fn(script_directory, current_rev):
def migration_fn(rev, context): def migration_fn(rev, context):
# Get all upgrade steps from current revision to head # Get all upgrade steps from current revision to head
revisions = script_directory._upgrade_revs("head", current_rev) revisions = script_directory._upgrade_revs("head", current_rev)
for revision in revisions: for revision_step in revisions:
script = script_directory.get_revision(revision) # Access the revision string from the RevisionStep object
script = script_directory.get_revision(revision_step.revision)
script.module.upgrade(context) script.module.upgrade(context)
return migration_fn return migration_fn