populate_interpro_accession_dict.py

from collections import defaultdict
import json
import os.path

from orthologue_analysis.species import MansoniCladeFromTool
from reannotation.utils import populate_accession_product_dict

if __name__ == "__main__":
    filepath = os.path.join("data", "acc_product.json")
    mars_data_dir = os.path.join("data", "from_MARS", "")

    species = MansoniCladeFromTool("mansoni", data_dir=mars_data_dir, data_label="Sman_LT", prot_filename_suffix=".fa")
    acc_product = defaultdict(dict)
    if os.path.exists(filepath):
        with open(filepath, "r") as f:
            acc_product = json.loads(f.read())
    acc_product = populate_accession_product_dict(species, acc_product)
    with open(filepath, "w") as f:
        json.dump(acc_product, f, sort_keys=True, indent=4)