Files
galaxy/scripts/microbes/create_bacteria_table.py
T

80 lines
2.7 KiB
Python

#!/usr/bin/env python
#Dan Blankenberg
import sys, os
assert sys.version_info[:2] >= ( 2, 4 )
def __main__():
base_dir = os.path.join( os.getcwd(), "bacteria" )
try:
base_dir = sys.argv[1]
except:
pass
#print "using default base_dir:", base_dir
organisms = {}
for result in os.walk(base_dir):
this_base_dir,sub_dirs,files = result
for file in files:
if file[-5:] == ".info":
dict = {}
info_file = open(os.path.join(this_base_dir,file),'r')
info = info_file.readlines()
info_file.close()
for line in info:
fields = line.replace("\n","").split("=")
dict[fields[0]]="=".join(fields[1:])
if 'genome project id' in dict.keys():
name = dict['genome project id']
if 'build' in dict.keys():
name = dict['build']
if name not in organisms.keys():
organisms[name] = {'chrs':{},'base_dir':this_base_dir}
for key in dict.keys():
organisms[name][key]=dict[key]
else:
if dict['organism'] not in organisms.keys():
organisms[dict['organism']] = {'chrs':{},'base_dir':this_base_dir}
organisms[dict['organism']]['chrs'][dict['chromosome']]=dict
orgs = organisms.keys()
for org in orgs:
if 'name' not in organisms[org]:del organisms[org]
orgs = organisms.keys()
#need to sort by name
swap_test = False
for i in range(0, len(orgs) - 1):
for j in range(0, len(orgs) - i - 1):
if organisms[orgs[j]]['name'] > organisms[orgs[j + 1]]['name']:
orgs[j], orgs[j + 1] = orgs[j + 1], orgs[j]
swap_test = True
if swap_test == False:
break
print "||'''Organism'''||'''Kingdom'''||'''Group'''||'''Links to UCSC Archaea Browser'''||"
for org in orgs:
org = organisms[org]
at_ucsc = False
#if no gpi, then must be a ncbi chr which corresponds to a UCSC org, w/o matching UCSC designation
try:
build = org['genome project id']
except: continue
if 'build' in org:
build = org['build']
at_ucsc = True
out_str = "||"+org['name']+"||"+org['kingdom']+"||"+org['group']+"||"
if at_ucsc:
out_str = out_str + "Yes"
out_str = out_str + "||"
print out_str
if __name__ == "__main__": __main__()