Fix another issue with SharedPostgreSQL undo migration

Author: DavidMStraub

gramps_webapi/__main__.py

@@ -351,7 +351,7 @@ def migrate_gramps_undodb(ctx):
     dbmgr = ctx.obj["db_manager"]
     db_handle = dbmgr.get_db().db
     try:
-        migrate_undodb(db_handle.undodb, tree_id=dbmgr.dirname)
+        migrate_undodb(db_handle.undodb)
     finally:
         close_db(db_handle)
 

gramps_webapi/api/tasks.py

@@ -408,7 +408,7 @@ def upgrade_database_schema(self, tree: str, user_id: str):
         tree=tree, view_private=True, readonly=False, user_id=user_id
     )
     try:
-        migrate_undodb(db_handle.undodb, tree)
+        migrate_undodb(db_handle.undodb)
     finally:
         close_db(db_handle)
 
@@ -420,7 +420,7 @@ def upgrade_undodb_schema(self, tree: str, user_id: str):
         tree=tree, view_private=True, readonly=False, user_id=user_id
     )
     try:
-        migrate_undodb(db_handle.undodb, tree)
+        migrate_undodb(db_handle.undodb)
     finally:
         close_db(db_handle)
 

gramps_webapi/undodb.py

@@ -602,14 +602,14 @@ def get_transaction(
             return transaction._to_dict(old_data=old_data, new_data=new_data)
 
 
-def migrate(undodb: DbUndoSQL, tree_id: str) -> None:
+def migrate(undodb: DbUndoSQL) -> None:
     """Migrate the undo db to a new schema if needed."""
     with undodb.session_scope() as session:
         # return all rows where old_json AND new_json are NULL
         rows = (
             session.query(Change)
             .join(Connection)
-            .filter(Connection.tree_id == tree_id)
+            .filter(Connection.tree_id == undodb.tree_id)
             .filter(Change.old_json.is_(None), Change.new_json.is_(None))
             .all()
         )