Skip to content

Commit

Permalink
Save downloaded ontology to path with acronym and submission ID
Browse files Browse the repository at this point in the history
  • Loading branch information
caufieldjh committed Sep 5, 2024
1 parent d44c4ff commit c5e4278
Showing 1 changed file with 8 additions and 3 deletions.
11 changes: 8 additions & 3 deletions src/kg_bioportal/downloader.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,6 @@ def __init__(

return None

# TODO: save NCBO ID and version for each ontology, then pass to transformer
def download(self, onto_list: list = []) -> None:
"""Downloads data files from list of ontologies into data directory.
Expand Down Expand Up @@ -84,8 +83,9 @@ def download(self, onto_list: list = []) -> None:
metadata = requests.get(metadata_url, headers=headers).json()
logging.info(f"Name: {metadata['name']}")
latest_sub_metadata = requests.get(latest_sub_url, headers=headers).json()
submission_id = latest_sub_metadata["submissionId"]
logging.info(
f"Latest submission: {latest_sub_metadata['version']} - released {latest_sub_metadata['released']}"
f"Latest submission: {latest_sub_metadata['version']} - submission ID {submission_id} - released {latest_sub_metadata['released']}"
)

download_onto = requests.get(
Expand All @@ -96,7 +96,12 @@ def download(self, onto_list: list = []) -> None:
.split("filename=")[1]
.replace('"', "")
)
with open(f"{self.output_dir}/{onto_filename}", "wb") as outfile:

outpath = f"{self.output_dir}/{ontology}/{submission_id}/{onto_filename}"
outdir = f"{self.output_dir}/{ontology}/{submission_id}"
if not os.path.exists(outdir):
os.makedirs(outdir)
with open(outpath, "wb") as outfile:
outfile.write(download_onto.content)

return None
Expand Down

0 comments on commit c5e4278

Please sign in to comment.