23 lines
833 B
Python
23 lines
833 B
Python
|
#!/usr/bin/env python3
|
||
|
|
import sys
|
||
|
|
if len(sys.argv) != 4:
|
||
|
|
sys.exit(f"python3 {sys.argv[0]} <blast.out> <reference_fam.table> <geneIDs.list> > out_genefam.table")
|
||
|
|
blast_file, ref_file, ids_file = sys.argv[1:]
|
||
|
|
blast = {}
|
||
|
|
with open(blast_file, encoding='utf-8', errors='replace') as fh:
|
||
|
|
for line in fh:
|
||
|
|
f = line.split()
|
||
|
|
if len(f) > 1 and f[0] not in blast: blast[f[0]] = f[1]
|
||
|
|
reffam = {}
|
||
|
|
with open(ref_file, encoding='utf-8', errors='replace') as fh:
|
||
|
|
for line in fh:
|
||
|
|
f = line.split()
|
||
|
|
if len(f) >= 2: reffam[f[0]] = f[1]
|
||
|
|
with open(ids_file, encoding='utf-8', errors='replace') as fh:
|
||
|
|
for line in fh:
|
||
|
|
f = line.split()
|
||
|
|
if not f: continue
|
||
|
|
refid = blast.get(f[0], 'un')
|
||
|
|
famid = reffam.get(refid, 'un') if refid != 'un' else 'un'
|
||
|
|
print(f"{f[0]} {famid}")
|