diff options
| -rw-r--r-- | workflows/pull-data/genbank/genbank.py | 70 | ||||
| -rwxr-xr-x | workflows/pull-data/genbank/transform-genbank-xml2yamlfa.py | 10 |
2 files changed, 72 insertions, 8 deletions
diff --git a/workflows/pull-data/genbank/genbank.py b/workflows/pull-data/genbank/genbank.py index 7383261..3b5931b 100644 --- a/workflows/pull-data/genbank/genbank.py +++ b/workflows/pull-data/genbank/genbank.py @@ -1,7 +1,13 @@ # Genbank XML parser +from collections import namedtuple +import dateutil +from dateutil.parser import parse as dateparse +from dateutil.tz import gettz import os +import re import sys +import types import xml.etree.ElementTree as ET class GBError(Exception): @@ -47,8 +53,62 @@ Example of an output JSON: } """ -def get_metadata(id, gb): - return True,None +def get_metadata(id, gbseq): + host = types.SimpleNamespace() + sample = types.SimpleNamespace() + submitter = types.SimpleNamespace() + warnings = [] + + def warn(msg): + print(f"WARNING: {msg}",file=sys.stderr) + warnings.append(msg) + + host.host_species = "http://purl.obolibrary.org/obo/NCBITaxon_9606" + sample.sample_id = id + sample.database = "https://www.ncbi.nlm.nih.gov/genbank/" + sample.source_database_accession = f"http://identifiers.org/insdc/{id}#sequence" + # <GBQualifier> + # <GBQualifier_name>country</GBQualifier_name> + # <GBQualifier_value>USA: Cruise_Ship_1, California</GBQualifier_value> + # </GBQualifier> + sample.collection_location = "FIXME" + + # --- Handling dates --- + # <GBSeq_create-date>29-JUL-2020</GBSeq_create-date> + n = gbseq.find("./GBSeq_create-date") + creation_date = dateparse(n.text).date() + + # <GBSeq_update-date>30-JUL-2020</GBSeq_update-date> + n = gbseq.find("./GBSeq_update-date") + update_date = dateparse(n.text).date() + + # <GBQualifier> + # <GBQualifier_name>collection_date</GBQualifier_name> + # <GBQualifier_value>2020-04-01</GBQualifier_value> + # </GBQualifier> + n = gbseq.find(".//GBQualifier/GBQualifier_name/[.='collection_date']/../GBQualifier_value") + try: + date = dateparse(n.text).date() + sample.collection_date = str(date) + except dateutil.parser._parser.ParserError as e: + warn(str(e)) + sample.collection_date = str(creation_date) + except AttributeError: + warn("Missing collection_date - used creation_date instead") + sample.collection_date = str(creation_date) + + info = { + 'id': 'placeholder', + 'update_date': str(update_date), + 'host': host, + 'sample': sample, + #'virus': virus, + #'technology': technology, + 'submitter': submitter, + 'warnings': warnings, + } + print(info) + return True,info def get_sequence(id, gbseq): seq = None @@ -63,6 +123,8 @@ def get_sequence(id, gbseq): raise GBError(f"Sequence too short") return seq +# ---- BELOW IS JUST FOR REFERENCE ---- + min_len_to_count = 15000 num_seq_with_len_ge_X_bp = 0 @@ -78,7 +140,7 @@ if None: accession_version = GBSeq.find('GBSeq_accession-version').text try: - info_for_yaml_dict = { + info = { 'id': 'placeholder', 'host': {}, 'sample': {}, @@ -342,7 +404,7 @@ if None: # fw.write('>{}\n{}'.format(accession_version, GBSeq_sequence.text.upper())) with open(os.path.join(dir_fasta_and_yaml, '{}.yaml'.format(accession_version)), 'w') as fw: - json.dump(info_for_yaml_dict, fw, indent=2) + json.dump(info, fw, indent=2) except: print("Unexpected error for the ID {}: {}".format(accession_version, sys.exc_info()[0])) accession_with_errors_list.append(accession_version) diff --git a/workflows/pull-data/genbank/transform-genbank-xml2yamlfa.py b/workflows/pull-data/genbank/transform-genbank-xml2yamlfa.py index c4e3eba..ebdf17e 100755 --- a/workflows/pull-data/genbank/transform-genbank-xml2yamlfa.py +++ b/workflows/pull-data/genbank/transform-genbank-xml2yamlfa.py @@ -40,6 +40,7 @@ for xmlfn in args.files: for gb in tree.findall('./GBSeq'): valid = None error = None + meta = {} id = gb.find("GBSeq_locus").text basename = dir+"/"+id print(f" parsing {id}") @@ -54,14 +55,15 @@ for xmlfn in args.files: f2.write(seq) # print(seq) except genbank.GBError as e: - print(f"OS error: {e}") + error = f"{e} for {id}" + print(error,file=sys.stderr) valid = False - error = str(e) state = {} - if not valid: - state['valid'] = False + state['valid'] = valid if error: state['error'] = error + if meta['warnings']: + state['warnings'] = meta['warnings'] states[id] = state print(states) |