refactor: Update migration function to access revision strings from RevisionStep objects for improved clarity #33
@ -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
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user