Fix bad error

OWASP · Nov 9, 2023 · 99e08c3 · 99e08c3
1 parent d52c4f3
commit 99e08c3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/application/database/db.py b/application/database/db.py
@@ -13,6 +13,7 @@
     StructuredRel,
     db,
 )
+import neo4j
 from sqlalchemy.orm import aliased
 import os
 import logging
@@ -547,7 +548,11 @@ def standards(self) -> List[str]:
 
     @classmethod
     def everything(self) -> List[str]:
-        return [NEO_DB.parse_node(rec) for rec in NeoDocument.nodes.all()]
+        try:
+            return [NEO_DB.parse_node(rec) for rec in NeoDocument.nodes.all()]
+        except neo4j.exceptions.ServiceUnavailable:
+            logger.error("Neo4j DB offline")
+            return None
 
     @staticmethod
     def parse_node(node: NeoDocument) -> cre_defs.Document: