Merge pull request #19136 from guerler/fix_saved_viz

Restore access to saved visualizations
galaxyproject · Nov 15, 2024 · 5ab07ff · 5ab07ff
2 parents ef6d1bb + 7780c62
commit 5ab07ff
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/lib/galaxy/webapps/galaxy/controllers/visualization.py b/lib/galaxy/webapps/galaxy/controllers/visualization.py
@@ -41,6 +41,16 @@ class VisualizationController(
     def __init__(self, app: StructuredApp):
         super().__init__(app)
 
+    def get_visualization(self, trans, visualization_id, check_ownership=True, check_accessible=False):
+        """
+        Get a Visualization from the database by id, verifying ownership.
+        """
+        visualization = trans.sa_session.get(model.Visualization, trans.security.decode_id(visualization_id))
+        if not visualization:
+            raise MessageException("Visualization not found")
+        else:
+            return self.security_check(trans, visualization, check_ownership, check_accessible)
+
     #
     # -- Functions for operating on visualizations. --
     #