From 6f257826efdd3271b94b3ab781c1972d1058d66f Mon Sep 17 00:00:00 2001 From: Terry Truong Date: Fri, 17 Jun 2022 20:37:06 +1000 Subject: Fix reduced-tree composite-node-inclusion bug --- backend/data/genReducedTreeData.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/backend/data/genReducedTreeData.py b/backend/data/genReducedTreeData.py index 0991f92..208c937 100755 --- a/backend/data/genReducedTreeData.py +++ b/backend/data/genReducedTreeData.py @@ -120,10 +120,17 @@ for (name, nodeObj) in nodeMap.items(): numChildren = len(nodeObj["children"]) if numChildren < PREF_NUM_CHILDREN: children = [row[0] for row in dbCur.execute("SELECT child FROM edges where node = ?", (name,))] - newChildren = [n for n in children if - not (n in nodeMap or n in namesToRemove) and - (dbCur.execute("SELECT name from node_imgs WHERE name = ?", (n,)).fetchone() != None or - dbCur.execute("SELECT name from linked_imgs WHERE name = ? AND otol_id2 IS NULL", (n,)).fetchone() != None)] + newChildren = [] + for n in children: + if n in nodeMap or n in namesToRemove: + continue + if compNameRegex.fullmatch(name) != None: + continue + if dbCur.execute("SELECT name from node_imgs WHERE name = ?", (n,)).fetchone() == None: + continue + if dbCur.execute("SELECT name from linked_imgs WHERE name = ?", (n,)).fetchone() == None: + continue + newChildren.append(n) newChildNames = newChildren[:max(0, PREF_NUM_CHILDREN - numChildren)] nodeObj["children"].extend(newChildNames) namesToAdd.extend(newChildNames) -- cgit v1.2.3