diff options
| author | Pjotr Prins | 2021-01-01 10:59:56 -0600 |
|---|---|---|
| committer | Pjotr Prins | 2021-01-01 10:59:56 -0600 |
| commit | a029c2329e748874bee88317e44d3f47547f71d8 (patch) | |
| tree | 2ef6d412ff5ba3aae61c74898e7b2738ba4e6bc3 /workflows/pull-data/genbank/update-from-genbank.py | |
| parent | 3f059ebde6fe6888e62f4fc232d05fb3a322b011 (diff) | |
| download | bh20-seq-resource-a029c2329e748874bee88317e44d3f47547f71d8.tar.gz bh20-seq-resource-a029c2329e748874bee88317e44d3f47547f71d8.tar.lz bh20-seq-resource-a029c2329e748874bee88317e44d3f47547f71d8.zip | |
genbank: minor fixes
Diffstat (limited to 'workflows/pull-data/genbank/update-from-genbank.py')
| -rwxr-xr-x | workflows/pull-data/genbank/update-from-genbank.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/workflows/pull-data/genbank/update-from-genbank.py b/workflows/pull-data/genbank/update-from-genbank.py index d92f87a..3faea39 100755 --- a/workflows/pull-data/genbank/update-from-genbank.py +++ b/workflows/pull-data/genbank/update-from-genbank.py @@ -34,11 +34,14 @@ dir = args.out if not os.path.exists(dir): raise Exception(f"Directory {dir} does not exist") -request_num = min(BATCH,args.max) +request_num = BATCH +if args.max: + request_num = min(BATCH,args.max) + for i, idsx in enumerate(chunks(list(ids), request_num)): xmlfn = os.path.join(dir, f"metadata_{i}.xml.gz") print(f"Fetching {xmlfn} ({i*request_num})",file=sys.stderr) with gzip.open(xmlfn, 'w') as f: f.write((Entrez.efetch(db='nuccore', id=idsx, retmode='xml').read()).encode()) - if i*request_num >= args.max: + if args.max and i*request_num >= args.max: break |