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

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