refactor: Update migration function to access revision strings from RevisionStep objects for improved clarity #33

Merged
mo merged 1 commits from ph4 into prod 2025-06-01 17:50:18 +02:00
Showing only changes of commit 20daadc112 - Show all commits

View File

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