Merge branch 'release_23.2' into dev

ahmedhamidawan · Feb 22, 2024 · 7b57bf4 · 7b57bf4
2 parents 49e0d43 + 036e469
commit 7b57bf4
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/lib/galaxy/model/__init__.py b/lib/galaxy/model/__init__.py
@@ -9647,11 +9647,12 @@ def save(self):
 
     @classmethod
     def store(cls, server_url, association):
-        try:
+        def get_or_create():
             stmt = select(PSAAssociation).filter_by(server_url=server_url, handle=association.handle).limit(1)
             assoc = cls.sa_session.scalars(stmt).first()
-        except IndexError:
-            assoc = cls(server_url=server_url, handle=association.handle)
+            return assoc or cls(server_url=server_url, handle=association.handle)
+
+        assoc = get_or_create()
         assoc.secret = base64.encodebytes(association.secret).decode()
         assoc.issued = association.issued
         assoc.lifetime = association.lifetime
@@ -9663,7 +9664,7 @@ def store(cls, server_url, association):
     @classmethod
     def get(cls, *args, **kwargs):
         stmt = select(PSAAssociation).filter_by(*args, **kwargs)
-        return cls.sa_session.scalars(stmt)
+        return cls.sa_session.scalars(stmt).all()
 
     @classmethod
     def remove(cls, ids_to_delete):