find_prot.py

def find_prot(fasta_dict, protein_name):
    '''Function to find protein in fasta_dictionary'''

    # Check if fasta_dict contains the protein name key
    if protein_name in fasta_dict:

        # Return biological sequence
        return fasta_dict[protein_name]

    else:

        # Print error message
        print "Error: %s not found" % protein_name

        return None   # We'll see how to deal with this properly next week