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):
|
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
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user