Update segprint.py for new database

Signed-off-by: Clifford Wolf <clifford@clifford.at>
Signed-off-by: Tim 'mithro' Ansell <mithro@mithis.com>
This commit is contained in:
Clifford Wolf 2017-11-17 02:25:11 +01:00 committed by Tim 'mithro' Ansell
parent b6e9b63dff
commit b6e77bc285
1 changed files with 2 additions and 2 deletions

View File

@ -67,12 +67,12 @@ def get_database(segtype):
segbitsdb[segtype] = list()
with open("%s/%s/seg_%s.segbits" % (os.getenv("XRAY_DATABASE_DIR"), os.getenv("XRAY_DATABASE"), segtype), "r") as f:
with open("%s/%s/segbits_%s.db" % (os.getenv("XRAY_DATABASE_DIR"), os.getenv("XRAY_DATABASE"), segtype), "r") as f:
for line in f:
line = line.split()
segbitsdb[segtype].append(line)
with open("%s/%s/seg_%s.segbits" % (os.getenv("XRAY_DATABASE_DIR"), os.getenv("XRAY_DATABASE"), segtype.replace("_", "_int_")), "r") as f:
with open("%s/%s/segbits_int_%s.db" % (os.getenv("XRAY_DATABASE_DIR"), os.getenv("XRAY_DATABASE"), segtype[-1]), "r") as f:
for line in f:
line = line.split()
segbitsdb[segtype].append(line)