From 20daadc112a08f9c4564e5be8af371a8420ea325 Mon Sep 17 00:00:00 2001
From: mohamad <mohamad>
Date: Sun, 1 Jun 2025 17:50:02 +0200
Subject: [PATCH] refactor: Update migration function to access revision
 strings from RevisionStep objects for improved clarity

---
 be/alembic/migrations.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/be/alembic/migrations.py b/be/alembic/migrations.py
index b535a37..ae3b924 100644
--- a/be/alembic/migrations.py
+++ b/be/alembic/migrations.py
@@ -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