Remove try statement from middleware

Signed-off-by: jamshale <[email protected]>
jamshale · Apr 24, 2024 · 9eaf786 · 9eaf786
1 parent 6418dfd
commit 9eaf786
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/aries_cloudagent/admin/server.py b/aries_cloudagent/admin/server.py
@@ -211,12 +211,13 @@ async def upgrade_middleware(request: web.BaseRequest, handler: Coroutine):
     """Blocking middleware for upgrades."""
     context: AdminRequestContext = request["context"]
 
+    is_upgrading = []
     async with context.profile.session() as session:
         storage = session.inject(BaseStorage)
-        try:
-            await storage.find_record(RECORD_TYPE_ACAPY_UPGRADING, tag_query={})
-        except StorageNotFoundError:
-            return await handler(request)
+        is_upgrading = await storage.find_all_records(RECORD_TYPE_ACAPY_UPGRADING)
+
+    if not is_upgrading:
+        return await handler(request)
 
     raise web.HTTPServiceUnavailable(reason="Upgrade in progress")