Add annotation for fasta from a dictionary

gistfile1.py

with open ("anotation.txt", "r") as annotation:
anotation_dict = {}
for line in annotation:
line = line.split()
if line: #test whether it is an empty line
anotation_dict[line[0]]=line[1:]
else:
continue
 
# really should not parse the fasta file by myself. there are
# many parsers already there. you can use module from Biopython
ofile = open ("output.txt", "w")
 
with open ("data.fasta", "r") as fasta:
for line in fasta:
if line.startswith (">"):
line = line[1:] # skip the ">" character
line = line.split()
if line[0] in anotation_dict:
new_line = ">" + str(line[0]) + " " + " ".join(anotation_dict[line[0]])
ofile.write ( new_line + "\n")
else:
ofile.write ( ">"+ "".join(line) + "\n")
else:
ofile.write(line +"\n")
 
 
ofile.close()