diff options
Diffstat (limited to 'backend/cgi-bin/data.py')
| -rwxr-xr-x | backend/cgi-bin/data.py | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/backend/cgi-bin/data.py b/backend/cgi-bin/data.py index b152c14..9b4ca40 100755 --- a/backend/cgi-bin/data.py +++ b/backend/cgi-bin/data.py @@ -76,9 +76,9 @@ class NodeInfo: self.imgInfo = imgInfo # null | ImgInfo class InfoResponse: " Sent as responses to 'info' requests " - def __init__(self, nodeInfo, subNodesInfo = None): + def __init__(self, nodeInfo, subNodesInfo): self.nodeInfo = nodeInfo # NodeInfo - self.subNodesInfo = subNodesInfo # [] | [NodeInfo, NodeInfo] + self.subNodesInfo = subNodesInfo # [] | [NodeInfo | null, NodeInfo | null] # For data lookup def lookupNodes(names, useReducedTree, dbCur): @@ -191,11 +191,10 @@ def lookupInfo(name, useReducedTree, dbCur): subNames = [match.group(1), match.group(2)] if match != None else [] if len(subNames) > 0: nameToSubNodes = lookupNodes(subNames, useReducedTree, dbCur) - if len(nameToSubNodes) < 2: - print(f"ERROR: Unable to find sub-names entries for {name}", file=sys.stderr) - return None + if len(nameToSubNodes) < 2: # Possible when a subname-denoted node has been trimmed away + subNames = [n if n in nameToSubNodes else None for n in subNames] nameToNodes.update(nameToSubNodes) - namesToLookup = [name] if len(subNames) == 0 else subNames + namesToLookup = [name] if len(subNames) == 0 else [n for n in subNames if n != None] # Get desc info nameToDescInfo = {} query = "SELECT name, desc, wiki_id, redirected, from_dbp FROM" \ @@ -219,7 +218,7 @@ def lookupInfo(name, useReducedTree, dbCur): nameToNodes[n], nameToDescInfo[n] if n in nameToDescInfo else None, nameToImgInfo[n] if n in nameToImgInfo else None - ) for n in [name] + subNames + ) if n != None else None for n in [name] + subNames ] return InfoResponse(nodeInfoObjs[0], nodeInfoObjs[1:]) |