Skip to content

Commit 295ed54

Browse files
author
Robert Hubley
committed
- Support both min famdb *and* current dfam 3.8 famdb
1 parent 00a8318 commit 295ed54

File tree

2 files changed

+12
-4
lines changed

2 files changed

+12
-4
lines changed

famdb.py

+4
Original file line numberDiff line numberDiff line change
@@ -538,6 +538,10 @@ def command_append(args):
538538
"""
539539

540540
lookup = args.db_dir.get_all_taxa_names()
541+
repbase_lookup = {}
542+
with open(REPBASE_FILE) as file:
543+
repbase_lookup = json.load(file)
544+
lookup.update(repbase_lookup)
541545

542546
header = None
543547

famdb_helper_classes.py

+8-4
Original file line numberDiff line numberDiff line change
@@ -632,13 +632,17 @@ def set_family_code(family, code, value):
632632
for spec in matches.group(1).split(","):
633633
name = spec.strip()
634634
if name:
635-
name = name.replace("[", "")
636-
name = name.replace("]", "")
637-
tax_id = lookup.get(name.lower())
635+
tax_id = lookup.get(name)
638636
if tax_id is not None:
639637
family.clades += [tax_id]
640638
else:
641-
LOGGER.warning("Could not find taxon for '%s'", name)
639+
name = name.replace("[", "")
640+
name = name.replace("]", "")
641+
tax_id = lookup.get(name.lower())
642+
if tax_id is not None:
643+
family.clades += [tax_id]
644+
else:
645+
LOGGER.warning("Could not find taxon for '%s' upper or lower: line=%s, and ID=%s", name, value, family.accession)
642646
matches = re.search(r"SearchStages:\s*(\S+)", value)
643647
if matches:
644648
family.search_stages = matches.group(1).strip()

0 commit comments

Comments
 (0)