#!/usr/bin/env python3

name = input('Protein name: ')

path1 = name.lower() + '.needle'
file1 = open(path1)
seq_ecoli_1 = ''
seq_bacsu_1 = ''

for line in file1:
    if not line.startswith('#'):
        if line.startswith(name.upper() + '_ECOLI'):
            seq_ecoli_1 += line.split()[2]
        elif line.startswith(name.upper() + '_BACSU'):
            seq_bacsu_1 += line.split()[2]
file1.close()



path2 = name.lower() + '.water'
file2 = open(path2)
seq_ecoli_2 = ''
seq_bacsu_2 = ''

for line in file2:
    if not line.startswith('#'):
        if line.startswith(name.upper() + '_ECOLI'):
            seq_ecoli_2 += line.split()[2]
        elif line.startswith(name.upper() + '_BACSU'):
            seq_bacsu_2 += line.split()[2]
file1.close()

coverage1 = (len(seq_ecoli_2) - seq_ecoli_2.count('-')) / (len(seq_ecoli_1) - seq_ecoli_1.count('-')) * 100
coverage2 = (len(seq_bacsu_2) - seq_bacsu_2.count('-')) / (len(seq_bacsu_1) - seq_bacsu_1.count('-')) * 100
print(f'Coverage {name} (E. coli): {coverage1:.01f}')
print(f'Coverage {name} (B. subtilis): {coverage2:.01f}')