def protein(self, query, mode):
"""Return the Protein object"""
o = ''
p=None
if mode == "GI":
p = Protein.select(Protein.q.gi.contains(query))
elif mode == "Accession":
p = Protein.select(Protein.q.accesion.contains(query))
if p[0].organism == 1:
o = "Drosophila melanogaster"
else:
o = "Saccaromyces cervisiae"
title = "Protein: " + mode + query
return dict(title=title, protein=p[0], organism=o)
def protein_list(self):
title = "Search Page"
proteins = Protein.select()
count = proteins.count()
return dict(title=title, proteins=proteins, count=count)
