From 992853eeac9ace9f6eb4332516a8ba6ff7545deb Mon Sep 17 00:00:00 2001 From: Terry Truong Date: Tue, 17 May 2022 12:39:16 +1000 Subject: Make reduced tree generation check non-alt-names --- backend/data/genReducedTreeData.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'backend/data') diff --git a/backend/data/genReducedTreeData.py b/backend/data/genReducedTreeData.py index 508e751..9cdf1d6 100755 --- a/backend/data/genReducedTreeData.py +++ b/backend/data/genReducedTreeData.py @@ -28,7 +28,9 @@ with open(nodeNamesFile) as file: if iterNum % 100 == 0: print("Iteration {}".format(iterNum)) # - row = dbCur.execute("SELECT name, alt_name from names WHERE alt_name = ?", (line.rstrip(),)).fetchone() + row = dbCur.execute("SELECT name from nodes WHERE name = ?", (line.rstrip(),)).fetchone() + if row == None: + row = dbCur.execute("SELECT name from names WHERE alt_name = ?", (line.rstrip(),)).fetchone() if row != None: minimalNames.add(row[0]) if len(minimalNames) == 0: -- cgit v1.2.3