Fix: Tags strings splitting

OWASP · Oct 5, 2023 · 496e8fb · 496e8fb
1 parent f7105bd
commit 496e8fb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/application/database/db.py b/application/database/db.py
@@ -331,7 +331,7 @@ def add_cre(self, dbcre: CRE):
                 "document_id": dbcre.id,
                 "description": dbcre.description,
                 "links": [],  # dbcre.links,
-                "tags": dbcre.tags,
+                "tags": [dbcre.tags] if isinstance(dbcre.tags, str) else dbcre.tags,
             }
         )
 
@@ -344,7 +344,7 @@ def add_dbnode(self, dbnode: Node):
                     "doctype": dbnode.ntype,
                     "document_id": dbnode.id,
                     "description": dbnode.description or "",
-                    "tags": dbnode.tags or [],
+                    "tags": [dbnode.tags] if isinstance(dbnode.tags, str) else dbnode.tags,
                     "hyperlink": "",  # dbnode.hyperlink or "",
                     "version": dbnode.version or "",
                     "section": dbnode.section or "",
@@ -361,7 +361,7 @@ def add_dbnode(self, dbnode: Node):
                     "document_id": dbnode.id,
                     "description": dbnode.description,
                     "links": [],  # dbnode.links,
-                    "tags": dbnode.tags,
+                    "tags": [dbnode.tags] if isinstance(dbnode.tags, str) else dbnode.tags,
                     "metadata": "{}",  # dbnode.metadata,
                     "hyperlink": "",  # dbnode.hyperlink or "",
                     "version": dbnode.version or "",
@@ -380,7 +380,7 @@ def add_dbnode(self, dbnode: Node):
                     "document_id": dbnode.id,
                     "description": dbnode.description,
                     "links": [],  # dbnode.links,
-                    "tags": dbnode.tags,
+                    "tags": [dbnode.tags] if isinstance(dbnode.tags, str) else dbnode.tags,
                     "metadata": "{}",  # dbnode.metadata,
                     "hyperlink": "",  # dbnode.hyperlink or "",
                     "version": dbnode.version or "",