the one that is working for cds headers

import os

import pickle

path_to_cds_pickle_files = '/home/rijandhakal/Desktop/map-multifile-cds-to-protein-alignment/CDS_pickles' # path to folder with cds pickle files, for worksation:/home/opensourceferns/Desktop/map-multifile-cds-to-protein-alignment/CDS_pickles  for the laptop:/home/rijandhakal/Desktop/map-multifile-cds-to-protein-alignment/CDS_pickles
path_to_result = '/home/rijandhakal/Desktop/map-multifile-cds-to-protein-alignment/Test_folder/Test_results' # path to folder where I want to eventually dump my results. For the laptop: /home/rijandhakal/Desktop/map-multifile-cds-to-protein-alignment/Test_folder/Test_results, for the workstation: /home/opensourceferns/Desktop/map-multifile-cds-to-protein-alignment/Test_folder/Test_results
path_to_algined_files = '/home/rijandhakal/Desktop/map-multifile-cds-to-protein-alignment/Test_folder/Aligned_peptide' # this one is the lest folder and will have to be changed later. For the workstation: /home/opensourceferns/Desktop/map-multifile-cds-to-protein-alignment/Test_folder/Aligned_peptide, for the laptop:'/home/rijandhakal/Desktop/map-multifile-cds-to-protein-alignment/Test_folder/Aligned_peptide'
path_to_script_accessories ='/home/rijandhakal/Desktop/map-multifile-cds-to-protein-alignment/Script_accesories' # path to the accessory files. For the laptop: '/home/rijandhakal/Desktop/map-multifile-cds-to-protein-alignment/Script_accesories'. For the workstation: /home/opensourceferns/Desktop/map-multifile-cds-to-protein-alignment/Script_accesories

def main():

    hop_to_species_shorthand = os.chdir(path_to_script_accessories) # this is the folder that has the dict in pickle format

    with open('cds_identifiers.pkl','rb') as cds_shorthand: # this opens the pickle file
        cds_shorthand_dict = pickle.load(cds_shorthand) # open the dict

    hop_to_aligned_files = os.chdir(path_to_algined_files) # hop to the peptide files

    list_of_aligned_files = os.listdir(path_to_algined_files) # make a list of the files in the peptide folder

    for file in list_of_aligned_files: # go through the files
        with open(file,'rt') as current_peptide_file: # using with to not have to bother closing the files manaully
            current_peptide_headers = current_peptide_file.readlines() # turning it into an iterable object

    def test_species(current_peptide_header): # this is a prototype, this function is actually called inside current_peptide_header
        for key,value in cds_shorthand_dict.items():
            if key in current_peptide_header:
                return value

    for current_peptide_header in current_peptide_headers: # opeing the peptide header line
        if current_peptide_header[0] == '>': # '>' makes sure that the header is actually a header
            this_file = test_species(current_peptide_header)
            map_sequence = key_vs_key(current_peptide_header,this_file,file)


def key_vs_key(current_peptide_header,this_file,file):

    file = os.path.splitext(file)[0]
    file = f'{path_to_result}/{file}.{"acds"}'

    hop_to_cds_pickles = os.chdir(path_to_cds_pickle_files)
    with open(this_file,'rb') as current_cds_pickle:
        current_pickle = pickle.load(current_cds_pickle)

    for Universal_header,cds_sequence in current_pickle.items():
        if current_peptide_header == Universal_header:
            with open(file,'a') as current_acds_file:
                current_acds_file.write(current_peptide_header)
                current_acds_file.write(cds_sequence)

main()