import sys str = sys.argv[1] fh=open(str,'r') str_mod = str.split('.')[0]+'_mod.str' fh_out = open(str_mod,'w') for line in fh: line=line.strip() line=line.split(' ') if line[0].startswith('525132_allozyme'): fh_out.write(line[0]+' '+'OH'+' ') for i in range(2,len(line)-1): fh_out.write(line[i]+' ') elif line[0].startswith('5251'): fh_out.write('\n'+line[0]+' '+'OH'+' ') for i in range(2,len(line)-1): fh_out.write(line[i]+' ') elif line[0].startswith('5252'): fh_out.write('\n'+line[0]+' '+'VA'+' ') for j in range(2,len(line)-1): fh_out.write(line[j]+' ')